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. |