Searched refs:node_name_map (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | host_training_loop_optimization_util.cc | 455 const auto node_name_map = graph->BuildNodeNameIndex(); in AddReshardOp() local
456 const auto node_it = node_name_map.find(compile_node_name); in AddReshardOp()
457 TF_RET_CHECK(node_it != node_name_map.end()) in AddReshardOp()
492 auto loop_cond_node_it = node_name_map.find(cond_node_name); in AddReshardOp()
493 TF_RET_CHECK(loop_cond_node_it != node_name_map.end()) in AddReshardOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 314 std::unordered_map<string, Node*>* node_name_map);
318 std::unordered_map<string, Node*>* node_name_map);
322 std::unordered_map<string, Node*>* node_name_map);
427 const std::unordered_map<string, Node*>& node_name_map);
433 const std::unordered_map<string, Node*>& node_name_map,
587 const std::unordered_map<string, Node*>& node_name_map) { in GetUniqueNodeName() argument
591 while (node_name_map.find(new_node_name) != node_name_map.end()) { in GetUniqueNodeName()
725 const std::unordered_map<string, Node*>& node_name_map) { in CreatePlaceholderNodeForFeed() argument
730 : GetUniqueNodeName(node->name(), index, node_name_map); in CreatePlaceholderNodeForFeed()
774 const std::unordered_map<string, Node*>& node_name_map, in GetInputOutputNodes() argument
[all …]
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | node_def_util.cc | 1017 const std::map<absl::string_view, absl::string_view>& node_name_map, in MaybeUpdateColocationConstraintsWithMap() argument
1028 if (node_name_map.find(original) != node_name_map.end()) { in MaybeUpdateColocationConstraintsWithMap()
1030 strings::StrCat(kColocationGroupPrefix, node_name_map.at(original)); in MaybeUpdateColocationConstraintsWithMap()
|
| D | node_def_util.h | 440 const std::map<absl::string_view, absl::string_view>& node_name_map,
|