Searched refs:name_to_node_map (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/cc/tools/
H A Dfreeze_saved_model.cc67 std::unordered_map<string, NodeDef*>* name_to_node_map) {
70 (*name_to_node_map)[node->name()] = node;
83 // name_to_node_map is needed to get the inputs from the NodeDef corresponding
86 std::unordered_map<string, NodeDef*> name_to_node_map; local
87 GetNodeNameToNodeDefMap(graph_def, &name_to_node_map);
102 NodeDef* node = name_to_node_map[node_name];
65 GetNodeNameToNodeDefMap( GraphDef* graph_def, std::unordered_map<string, NodeDef*>* name_to_node_map) argument

Completed in 205 milliseconds