Gets a map from string node name to NodeDef.
| 68 | |
| 69 | // Gets a map from string node name to NodeDef. |
| 70 | void GetNodeNameToNodeDefMap( |
| 71 | GraphDef* graph_def, |
| 72 | std::unordered_map<string, NodeDef*>* name_to_node_map) { |
| 73 | for (size_t i = 0; i < graph_def->node_size(); i++) { |
| 74 | NodeDef* node = graph_def->mutable_node(i); |
| 75 | (*name_to_node_map)[node->name()] = node; |
| 76 | } |
| 77 | } |
| 78 | |
| 79 | // Strips off the tensor part of the tensor_name to get the node_name. |
| 80 | const string GetNodeNameFromTensorName(string tensor_name) { |