ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more


Initialize the TPU devices.

Used in the notebooks

Used in the guide Used in the tutorials

cluster_resolver A tf.distribute.cluster_resolver.TPUClusterResolver, which provides information about the TPU cluster.

The tf.tpu.Topology object for the topology of the TPU cluster. If called inside tf.function, it returns the serialized topology object instead.

RuntimeError If running inside a tf.function.
NotFoundError If no TPU devices found in eager mode.