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,
    tpu_compile_options_proto='',
    name=None
)

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

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