{ }
Metadata indicating how the TPU computation should be replicated.
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
)
This operation holds the metadata common to operations of a tpu.replicate()
computation subgraph.
Returns | |
---|---|
The created Operation. |