Searched refs:unpruned_recompute_nodes (Results 1 – 1 of 1) sorted by relevance
167 std::unordered_set<const NodeDef*> unpruned_recompute_nodes; in GetOpGroupsToRecompute() local168 unpruned_recompute_nodes.insert(recompute_node); in GetOpGroupsToRecompute()172 should_recompute, &unpruned_recompute_nodes); in GetOpGroupsToRecompute()173 visited_nodes.insert(unpruned_recompute_nodes.begin(), in GetOpGroupsToRecompute()174 unpruned_recompute_nodes.end()); in GetOpGroupsToRecompute()175 for (const NodeDef* recompute_node : unpruned_recompute_nodes) { in GetOpGroupsToRecompute()196 [&unpruned_recompute_nodes](const NodeDef& node) { in GetOpGroupsToRecompute()197 return unpruned_recompute_nodes.count(&node) != 0; in GetOpGroupsToRecompute()