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


Shuts down the TPU devices.

This will clear all caches, even those that are maintained through sequential calls to tf.tpu.experimental.initialize_tpu_system, such as the compilation cache.

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

RuntimeError If no TPU devices found for eager execution or if run in a tf.function.