Searched refs:recomputed_source_nodes (Results 1 – 1 of 1) sorted by relevance
165 std::unordered_set<const NodeDef*> recomputed_source_nodes; member204 current_recomputation.recomputed_source_nodes.insert( in GetOpGroupsToRecompute()219 ¤t_recomputation.recomputed_source_nodes); in GetOpGroupsToRecompute()234 const std::unordered_set<const NodeDef*>& recomputed_source_nodes, in GetMaxDownstreamComponents() argument239 for (const NodeDef* original_recompute_node : recomputed_source_nodes) { in GetMaxDownstreamComponents()259 recomputed_source_nodes.begin(), recomputed_source_nodes.end()); in GetMaxDownstreamComponents()278 if (recomputed_source_nodes.count(output) == 0) { in GetMaxDownstreamComponents()302 const std::unordered_set<const NodeDef*>& recomputed_source_nodes, in AddRecomputeControlDependencyNodes() argument310 recomputed_source_nodes.begin(), recomputed_source_nodes.end()); in AddRecomputeControlDependencyNodes()337 recomputed_source_nodes.count(target_input) != 0 || in AddRecomputeControlDependencyNodes()[all …]