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.
Args |
num_replicas
|
An int that is >= 0 .
Number of replicas of the computation
|
num_cores_per_replica
|
An optional int . Defaults to 1 .
Number of cores per replica. Used for model parallelism.
|
topology
|
An optional string . Defaults to "" .
TopologyProto indicating the topology of the TPU pod slice.
|
use_tpu
|
An optional bool . Defaults to True .
Whether to place the computation on the TPU.
|
device_assignment
|
An optional list of ints . Defaults to [] .
The assignment of devices for the computation.
|
computation_shape
|
An optional list of ints . Defaults to [] .
DEPRECATED. Use num_cores_per_replica instead.
|
host_compute_core
|
An optional list of strings . Defaults to [] .
|
padding_map
|
An optional list of strings . Defaults to [] .
|
step_marker_location
|
An optional string . Defaults to "STEP_MARK_AT_ENTRY" .
|
allow_soft_placement
|
An optional bool . Defaults to False .
|
use_spmd_for_xla_partitioning
|
An optional bool . Defaults to False .
|
tpu_compile_options_proto
|
An optional string . Defaults to "" .
|
name
|
A name for the operation (optional).
|
Returns |
The created Operation.
|