| 96 | } |
| 97 | |
| 98 | NodeDef* NodeMap::GetNode(const string& name) const { |
| 99 | const string node_name = NodeName(name); |
| 100 | auto it = nodes_.find(node_name); |
| 101 | if (it == nodes_.end()) { |
| 102 | return nullptr; |
| 103 | } |
| 104 | return it->second; |
| 105 | } |
| 106 | |
| 107 | bool NodeMap::NodeExists(const string& name) const { |
| 108 | const string node_name = NodeName(name); |