TensorFlow

API

 tf.tpu / experimental / experimental.Overview


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.

此页内容是否对您有帮助