W3cubDocs

/TensorFlow 2.4

tf.tpu.experimental.initialize_tpu_system

Initialize the TPU devices.

Args
cluster_resolver A tf.distribute.cluster_resolver.TPUClusterResolver, which provides information about the TPU cluster.
Returns
The tf.tpu.Topology object for the topology of the TPU cluster. If called inside tf.function, it returns the serialized topology object instead.
Raises
RuntimeError If running inside a tf.function.
NotFoundError If no TPU devices found in eager mode.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/tpu/experimental/initialize_tpu_system