Searched refs:unpruned_recompute_nodes (Results 1 – 1 of 1) sorted by relevance
186 std::unordered_set<const NodeDef*> unpruned_recompute_nodes; in GetOpGroupsToRecompute() local187 unpruned_recompute_nodes.insert(recompute_node); in GetOpGroupsToRecompute()191 should_recompute, &unpruned_recompute_nodes); in GetOpGroupsToRecompute()192 visited_nodes.insert(unpruned_recompute_nodes.begin(), in GetOpGroupsToRecompute()193 unpruned_recompute_nodes.end()); in GetOpGroupsToRecompute()194 for (const NodeDef* unpruned_recompute_node : unpruned_recompute_nodes) { in GetOpGroupsToRecompute()216 [&unpruned_recompute_nodes](const NodeDef& node) { in GetOpGroupsToRecompute()217 return unpruned_recompute_nodes.count(&node) != 0; in GetOpGroupsToRecompute()