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