Searched refs:recompute_node (Results 1 – 1 of 1) sorted by relevance
159 for (const NodeDef* recompute_node : candidate_recompute_nodes) { in GetOpGroupsToRecompute() local160 if (visited_nodes.count(recompute_node) > 0) { in GetOpGroupsToRecompute()168 unpruned_recompute_nodes.insert(recompute_node); in GetOpGroupsToRecompute()175 for (const NodeDef* recompute_node : unpruned_recompute_nodes) { in GetOpGroupsToRecompute() local177 for (NodeDef* output : node_map.GetOutputs(recompute_node->name())) { in GetOpGroupsToRecompute()185 recompute_node); in GetOpGroupsToRecompute()