Spec-Zone.ru › TensorFlow 2.9

tf.tpu.experimental.initialize_tpu_system

Просмотреть исходный код на GitHub

Инициализировать устройства TPU.

Просмотр псевдонимов

Псевдонимы для миграции

См. Руководство по миграции для получения более подробной информации.

tf.compat.v1.tpu.experimental.initialize_tpu_system

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.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/tpu/experimental/initialize_tpu_system

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API