Searched refs:recomputed_source_nodes (Results 1 – 1 of 1) sorted by relevance
146 std::unordered_set<const NodeDef*> recomputed_source_nodes; member184 current_recomputation.recomputed_source_nodes.insert( in GetOpGroupsToRecompute()199 ¤t_recomputation.recomputed_source_nodes); in GetOpGroupsToRecompute()214 const std::unordered_set<const NodeDef*>& recomputed_source_nodes, in GetMaxDownstreamComponents() argument219 for (const NodeDef* original_recompute_node : recomputed_source_nodes) { in GetMaxDownstreamComponents()239 recomputed_source_nodes.begin(), recomputed_source_nodes.end()); in GetMaxDownstreamComponents()258 if (recomputed_source_nodes.count(output) == 0) { in GetMaxDownstreamComponents()282 const std::unordered_set<const NodeDef*>& recomputed_source_nodes, in AddRecomputeControlDependencyNodes() argument290 recomputed_source_nodes.begin(), recomputed_source_nodes.end()); in AddRecomputeControlDependencyNodes()317 recomputed_source_nodes.count(target_input) != 0 || in AddRecomputeControlDependencyNodes()[all …]