tf.tpu.experimental.initialize_tpu_system

Initialize the TPU devices.

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.