tf.tpu.experimental.initialize_tpu_system
Инициализировать устройства TPU.
tf.tpu.experimental.initialize_tpu_system(
cluster_resolver=None
)
| Аргументы |
cluster_resolver | tf.distribute.cluster_resolver.TPUClusterResolver, предоставляющий информацию о кластере TPU. |
| Возвращает |
| Объект tf.tpu.Topology для топологии кластера TPU. Если вызывается внутри tf.function, возвращается сериализованный объект топологии. |
| Возбуждает |
RuntimeError | Если выполняется внутри tf.function. |
NotFoundError | Если в режиме eager не найдено устройств TPU. |