tf.raw_ops.TPUReplicateMetadata

Метаданные, указывающие, как следует реплицировать вычисление TPU.

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

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

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

tf.compat.v1.raw_ops.TPUReplicateMetadata

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, use_spmd_for_xla_partitioning=False, name=None
)

Данная операция содержит общие метаданные для операций подграфа вычислений tpu.replicate().

Аргументы
`num_replicas`	Число реплик вычисления
`num_cores_per_replica`	Необязательное число ядер на реплику. По умолчанию `1`. Используется для параллелизма модели.
`topology`	Необязательный TopologyProto, указывающий топологию среза узла TPU. По умолчанию `""`.
`use_tpu`	Необязательный флаг размещения вычислений на TPU. По умолчанию `True`.
`device_assignment`	Необязательный список назначений устройств для вычисления. По умолчанию `[]`.
`computation_shape`	Необязательный список (УСТЕРЕЖДЕНО). Используйте вместо этого num_cores_per_replica. По умолчанию `[]`.
`host_compute_core`	Необязательный список. По умолчанию `[]`.
`padding_map`	Необязательный список. По умолчанию `[]`.
`step_marker_location`	Необязательный элемент. По умолчанию `"STEP_MARK_AT_ENTRY"`.
`allow_soft_placement`	Необязательный элемент. По умолчанию `False`.
`use_spmd_for_xla_partitioning`	Необязательный элемент. По умолчанию `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.4/api_docs/python/tf/raw_ops/TPUReplicateMetadata