Home
last modified time | relevance | path

Searched refs:TPUREPLICATE_MIRRORED_VAR_INDICES_ATTR (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/
Dtpu_defs.cc24 const char* const TPUREPLICATE_MIRRORED_VAR_INDICES_ATTR = variable
Dtpu_defs.h44 extern const char* const TPUREPLICATE_MIRRORED_VAR_INDICES_ATTR;
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc335 if (xla_node->attrs().Find(TPUREPLICATE_MIRRORED_VAR_INDICES_ATTR) != in UpdateMirroredVariableIndices()
338 TPUREPLICATE_MIRRORED_VAR_INDICES_ATTR, in UpdateMirroredVariableIndices()
345 xla_node->ClearAttr(TPUREPLICATE_MIRRORED_VAR_INDICES_ATTR); in UpdateMirroredVariableIndices()
346 xla_node->AddAttr(TPUREPLICATE_MIRRORED_VAR_INDICES_ATTR, in UpdateMirroredVariableIndices()
766 if (xla_node->attrs().Find(TPUREPLICATE_MIRRORED_VAR_INDICES_ATTR) != in RemoveUnusedXlaInput()
769 TPUREPLICATE_MIRRORED_VAR_INDICES_ATTR, in RemoveUnusedXlaInput()
932 xla_node->ClearAttr(TPUREPLICATE_MIRRORED_VAR_INDICES_ATTR); in RemoveUnusedXlaInput()
933 xla_node->AddAttr(TPUREPLICATE_MIRRORED_VAR_INDICES_ATTR, in RemoveUnusedXlaInput()
2827 AddNodeAttr(TPUREPLICATE_MIRRORED_VAR_INDICES_ATTR, in BuildTPUReplicateOps()
Ddistributed_tpu_rewrite_pass.cc4032 TPUREPLICATE_MIRRORED_VAR_INDICES_ATTR, in GetIOTypes()