Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc539 const std::unordered_map<string, Node*>& node_name_map) { in GetUniqueNodeName()
616 const std::unordered_map<string, Node*>& node_name_map) { in CreatePlaceholderNodeForFeed()
665 const std::unordered_map<string, Node*>& node_name_map, in GetInputOutputNodes()
706 std::unordered_map<string, Node*>* node_name_map) { in AddNodesToShapeRefiner()
1330 std::unordered_map<string, Node*>* node_name_map) { in PruneUnreachableNodes()
1347 std::unordered_map<string, Node*>* node_name_map) { in ConvertFeedsToPlaceholders()
1399 auto node_name_map = graph_->BuildNodeNameIndex(); in PrepareConvert() local
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dhost_training_loop_optimization_util.cc442 const auto node_name_map = graph->BuildNodeNameIndex(); in AddReshardOp() local