Searched refs:nodes_to_keep (Results 1 – 3 of 3) sorted by relevance
169 nodes_to_keep = set()175 if node in nodes_to_keep:178 nodes_to_keep.add(node)181 return nodes_to_keep211 nodes_to_keep = _bfs_for_reachable_nodes(dest_nodes, name_to_input_name)214 list(nodes_to_keep), key=lambda n: name_to_seq_num[n])
235 std::vector<string> nodes_to_keep; in TEST_P() local236 nodes_to_keep.reserve(noop_nodes.size()); in TEST_P()241 nodes_to_keep.push_back(node->name()); in TEST_P()248 for (const auto &noop_node_name : nodes_to_keep) in TEST_P()
155 std::unordered_set<const Node*> nodes_to_keep; in RewriteAndPruneGraph() local157 &nodes_to_keep)); in RewriteAndPruneGraph()159 AddRetvalNodes(graph, node_map, config.fetch(), &nodes_to_keep)); in RewriteAndPruneGraph()161 PruneForReverseReachability(graph, std::move(nodes_to_keep)); in RewriteAndPruneGraph()