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