tf.raw_ops.TPUReplicateMetadata
Метаданные, указывающие, как следует дублировать вычисление TPU.
tf.raw_ops.TPUReplicateMetadata(
num_replicas, num_cores_per_replica=1, topology='', use_tpu=True,
device_assignment=[], computation_shape=[], host_compute_core=[],
padding_map=[], step_marker_location='STEP_MARK_AT_ENTRY',
allow_soft_placement=False, name=None
)
Данная операция содержит метаданные, общие для операций подграфа вычислений tpu.replicate().
| Аргументы | |
|---|---|
num_replicas |
Целое число, которое представляет собой int. Количество реплик вычисления |
num_cores_per_replica |
Необязательное целое число. По умолчанию 1. Количество ядер на реплику. Используется для параллелизма моделей. |
topology |
Необязательный протокол. По умолчанию "". TopologyProto, указывающий топологию среза подсети TPU. |
use_tpu |
Необязательное логическое значение. По умолчанию True. Необходимо ли размещать вычисление на TPU. |
device_assignment |
Необязательный список строк. По умолчанию []. Назначение устройств для вычисления. |
computation_shape |
Необязательный список целых чисел. По умолчанию []. УСТАРЕЛО. Используйте num_cores_per_replica вместо этого. |
host_compute_core |
Необязательный список целых чисел. По умолчанию []. |
padding_map |
Необязательный список целых чисел. По умолчанию []. |
step_marker_location |
Необязательный строковый тип данных. По умолчанию "STEP_MARK_AT_ENTRY". |
allow_soft_placement |
Необязательный строковый тип данных. По умолчанию False. |
name |
Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
| Созданная операция. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/raw_ops/TPUReplicateMetadata
