| 175 | } |
| 176 | |
| 177 | bool AddNodeToTargets(const string& node_or_tensor_name, |
| 178 | const NameIndex& name_index, |
| 179 | std::unordered_set<const Node*>* targets) { |
| 180 | TensorId id = ParseTensorName(node_or_tensor_name); |
| 181 | auto iter = name_index.find(id.first); |
| 182 | if (iter == name_index.end()) { |
| 183 | return false; |
| 184 | } |
| 185 | const Node* n = iter->second; |
| 186 | CHECK_EQ(n->name(), id.first); |
| 187 | targets->insert(n); |
| 188 | return true; |
| 189 | } |
| 190 | |
| 191 | Status PruneForTargets(Graph* g, const NameIndex& name_index, |
| 192 | const std::vector<Node*>& fetch_nodes, |
no test coverage detected