Searched refs:GetNodeId (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | host_training_loop_optimization_util.cc | 333 "TPUVariableReshard/before_iteration", "/_", internal::GetNodeId()))); in GetOrCreateBeforeEachIterationNode()
371 "TPUVariableReshard/last_iteration", "/_", internal::GetNodeId()))); in AddNoOpAfterLastIteration()
504 "TPUVariableReshard/default_shard_state", "/_", internal::GetNodeId()))); in AddReshardOp()
524 "TPUVariableReshard/last_iteration", "/_", internal::GetNodeId()))); in AddReshardOp()
534 "TPUVariableReshard/reshard", "/_", internal::GetNodeId()))); in AddReshardOp()
568 "TPUVariableReshard/reshard_state", "/_", internal::GetNodeId()))); in AddReshardOp()
588 "TPUVariableReshard/unshard", "/_", internal::GetNodeId()))); in AddReshardOp()
|
| D | distributed_tpu_rewrite_pass_internal.h | 33 uint64 GetNodeId();
|
| D | distributed_tpu_rewrite_pass_internal.cc | 35 uint64 GetNodeId() { in GetNodeId() function
|
| D | distributed_tpu_rewrite_pass.cc | 292 return graph->NewName(strings::StrCat(prefix, "/_", internal::GetNodeId())); in UniqueNodeName()
|