| 63 | } |
| 64 | |
| 65 | void MapNamesToNodes(const GraphDef& graph_def, |
| 66 | std::map<string, const NodeDef*>* result) { |
| 67 | for (const NodeDef& node : graph_def.node()) { |
| 68 | (*result)[node.name()] = &node; |
| 69 | } |
| 70 | } |
| 71 | |
| 72 | void MapNodesToOutputs(const GraphDef& graph_def, |
| 73 | std::map<string, std::vector<const NodeDef*>>* result) { |