tf.raw_ops.TPUReplicateMetadata
bookmark_border bookmark
Stay organized with collections
Save and categorize content based on your preferences.
Metadata indicating how the TPU computation should be replicated.
View aliases
Compat aliases for migration
See
Migration guide for
more details.
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
)
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.
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License , and code samples are licensed under the Apache 2.0 License . For details, see the Google Developers Site Policies . Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license .
Last updated 2024-01-23 UTC.
[[["Easy to understand","easyToUnderstand","thumb-up"],["Solved my problem","solvedMyProblem","thumb-up"],["Other","otherUp","thumb-up"]],[["Missing the information I need","missingTheInformationINeed","thumb-down"],["Too complicated / too many steps","tooComplicatedTooManySteps","thumb-down"],["Out of date","outOfDate","thumb-down"],["Samples / code issue","samplesCodeIssue","thumb-down"],["Other","otherDown","thumb-down"]],["Last updated 2024-01-23 UTC."],[],[]]