Searched refs:node_name_map (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | host_training_loop_optimization_util.cc | 442 const auto node_name_map = graph->BuildNodeNameIndex(); in AddReshardOp() local
443 const auto node_it = node_name_map.find(compile_node_name); in AddReshardOp()
444 TF_RET_CHECK(node_it != node_name_map.end()) in AddReshardOp()
479 auto loop_cond_node_it = node_name_map.find(cond_node_name); in AddReshardOp()
480 TF_RET_CHECK(loop_cond_node_it != node_name_map.end()) in AddReshardOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 264 std::unordered_map<string, Node*>* node_name_map);
268 std::unordered_map<string, Node*>* node_name_map);
272 std::unordered_map<string, Node*>* node_name_map);
377 const std::unordered_map<string, Node*>& node_name_map);
383 const std::unordered_map<string, Node*>& node_name_map,
539 const std::unordered_map<string, Node*>& node_name_map) { in GetUniqueNodeName() argument
543 while (node_name_map.find(new_node_name) != node_name_map.end()) { in GetUniqueNodeName()
616 const std::unordered_map<string, Node*>& node_name_map) { in CreatePlaceholderNodeForFeed() argument
621 : GetUniqueNodeName(node->name(), index, node_name_map); in CreatePlaceholderNodeForFeed()
665 const std::unordered_map<string, Node*>& node_name_map, in GetInputOutputNodes() argument
[all …]
|