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.cc584 const std::unordered_map<string, Node*>& node_name_map) { in GetUniqueNodeName()
720 const std::unordered_map<string, Node*>& node_name_map) { in CreatePlaceholderNodeForFeed()
769 const std::unordered_map<string, Node*>& node_name_map, in GetInputOutputNodes()
810 std::unordered_map<string, Node*>* node_name_map) { in AddNodesToShapeRefiner()
1385 std::unordered_map<string, Node*>* node_name_map) { in PruneUnreachableNodes()
1402 std::unordered_map<string, Node*>* node_name_map) { in ConvertFeedsToPlaceholders()
1470 auto node_name_map = graph_->BuildNodeNameIndex(); in PrepareConvert() local
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dhost_training_loop_optimization_util.cc457 const auto node_name_map = graph->BuildNodeNameIndex(); in AddReshardOp() local
/external/tensorflow/tensorflow/core/framework/
Dnode_def_util.cc960 const std::map<absl::string_view, absl::string_view>& node_name_map, in MaybeUpdateColocationConstraintsWithMap()