Home
last modified time | relevance | path

Searched defs:node_name_map (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc587 const std::unordered_map<string, Node*>& node_name_map) { in GetUniqueNodeName()
725 const std::unordered_map<string, Node*>& node_name_map) { in CreatePlaceholderNodeForFeed()
774 const std::unordered_map<string, Node*>& node_name_map, in GetInputOutputNodes()
815 std::unordered_map<string, Node*>* node_name_map) { in AddNodesToShapeRefiner()
1438 std::unordered_map<string, Node*>* node_name_map) { in PruneUnreachableNodes()
1455 std::unordered_map<string, Node*>* node_name_map) { in ConvertFeedsToPlaceholders()
1523 auto node_name_map = graph_->BuildNodeNameIndex(); in PrepareConvert() local
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dhost_training_loop_optimization_util.cc455 const auto node_name_map = graph->BuildNodeNameIndex(); in AddReshardOp() local
/external/tensorflow/tensorflow/core/framework/
Dnode_def_util.cc1017 const std::map<absl::string_view, absl::string_view>& node_name_map, in MaybeUpdateColocationConstraintsWithMap()