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`	Необязательный булево. По умолчанию `True`. Нужно ли размещать вычисления на TPU.
`device_assignment`	Необязательный список строк. По умолчанию `[]`. Распределение устройств для вычислений.
`computation_shape`	Необязательный список целых чисел. Устарело. Используйте num_cores_per_replica вместо него. По умолчанию `[]`.
`host_compute_core`	Необязательный список строк. По умолчанию `[]`.
`padding_map`	Необязательный список строк. По умолчанию `[]`.
`step_marker_location`	Необязательный TopologyProto. По умолчанию `"STEP_MARK_AT_ENTRY"`.
`allow_soft_placement`	Необязательный TopologyProto. По умолчанию `False`.
`use_spmd_for_xla_partitioning`	Необязательный TopologyProto. По умолчанию `False`.
`name`	Название операции (необязательно).

Возвращаемое значение
Созданная операция.

© 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/raw_ops/TPUReplicateMetadata