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 есть только одно задание; если это предположение неверно, будет возвращена ошибка.
`compilation_failure_closes_chips`	Задаёт, следует ли отключать чипы TPU при сбое компиляции.
`tpu_cancellation_closes_chips`	Задаёт, следует ли отключать чипы TPU при отмене выполнения TPU. Если значение равно None, поведение определяется флагом командной строки `tpu_cancellation_closes_chips` для рабочего процесса TPU. ПРЕДУПРЕЖДЕНИЕ: этот аргумент применяется только к среде выполнения TPU TFRT.

Возвращает
Сериализованный объект `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