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