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.

Возвращаемое значение
Сериализованный `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/api_docs/python/tf/compat/v1/tpu/initialize_system