Home
last modified time | relevance | path

Searched refs:current_recomputation (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dmemory_optimizer.cc182 RecomputedSubGraph current_recomputation; in GetOpGroupsToRecompute() local
199 current_recomputation.target_nodes.insert(output); in GetOpGroupsToRecompute()
204 current_recomputation.recomputed_source_nodes.insert( in GetOpGroupsToRecompute()
219 &current_recomputation.recomputed_source_nodes); in GetOpGroupsToRecompute()
220 if (current_recomputation.target_nodes.empty()) { in GetOpGroupsToRecompute()
223 subgraphs_to_recompute.push_back(current_recomputation); in GetOpGroupsToRecompute()