tf.tpu.experimental.initialize_tpu_system

Initialize the TPU devices.

Compat aliases for migration

tf.compat.v1.tpu.experimental.initialize_tpu_system

tf.tpu.experimental.initialize_tpu_system(
    cluster_resolver=None
)

Used in the guide	Used in the tutorials
Migrate from TPU embedding_columns to TPUEmbedding layer Migrate from TPUEstimator to TPUStrategy Use TPUs	TensorFlow 2 TPUEmbeddingLayer: Quick Start Training with Orbit Image classification with Model Garden Instance Segmentation with Model Garden Object detection with Model Garden

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.

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