Searched refs:replica_index (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_feed.py | 861 for replica_index in range(number_of_replicas):
862 flattened_inputs = sharded_inputs[replica_index]
867 for x, dims in zip(sharded_inputs[replica_index],
877 task_id=self._host_id, logical_core=0)[replica_index]
900 replica_index, logical_core),
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.h | 470 int replica_index, const std::vector<Node*>& outside_compilation_nodes,
|
| D | distributed_tpu_rewrite_pass.cc | 3463 int replica_index, const std::vector<Node*>& outside_compilation_nodes, in CopyOutsideCompilationNodes() argument
3470 const string suffix = strings::StrCat("/R", replica_index); in CopyOutsideCompilationNodes()
3477 image->AddAttr(kXlaReplicaIdAttrName, replica_index); in CopyOutsideCompilationNodes()
3501 node_image_vector.resize(replica_index + 1); in CopyOutsideCompilationNodes()
3502 node_image_vector[replica_index] = image; in CopyOutsideCompilationNodes()
|