tf.compat.v1.tpu.initialize_system

Инициализирует распределённую систему TPU для использования с TensorFlow.

tf.compat.v1.tpu.initialize_system(
    embedding_config: Optional[embedding_pb2.TPUEmbeddingConfiguration] = None,
    job: Optional[Text] = None,
    compilation_failure_closes_chips: bool = True,
    tpu_cancellation_closes_chips: Optional[bool] = None
) -> core_types.Tensor

Аргументы
`embedding_config`	Если не равно None, то `TPUEmbeddingConfiguration` протокол, описывающий желаемую конфигурацию таблиц поиска аппаратных вхождений. Если embedding_config равно None, то аппаратные вхождения использовать нельзя.
`job`	Задача (XXX в спецификации устройства TensorFlow /job:XXX), содержащая устройства TPU, которые будут инициализированы. Если job=None, предполагается, что в TensorFlow flock существует только одна задача, и будет возвращена ошибка, если это предположение не верно.
`compilation_failure_closes_chips`	Устанавливает конфигурацию, определяющую, хотим ли мы закрывать чипы TPU при возникновении ошибки компиляции.
`tpu_cancellation_closes_chips`	Устанавливает конфигурацию, определяющую, хотим ли мы закрывать чипы TPU при отмене выполнения TPU. Если значение равно None, поведение будет определяться командной строкой `tpu_cancellation_closes_chips` для TPU работника. ПРЕДУПРЕЖДЕНИЕ: этот аргумент применим только к TFRT TPU runtime.

Возвращаемое значение
Сериализованный `TopologyProto`, описывающий систему TPU. Примечание: топология должна быть оценена с помощью `Session.run` перед её использованием.

© 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/compat/v1/tpu/initialize_system