Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmemory_optimizer.cc163 RecomputedSubGraph current_recomputation; in GetOpGroupsToRecompute() local
179 current_recomputation.target_nodes.insert(output); in GetOpGroupsToRecompute()
184 current_recomputation.recomputed_source_nodes.insert( in GetOpGroupsToRecompute()
199 &current_recomputation.recomputed_source_nodes); in GetOpGroupsToRecompute()
200 if (current_recomputation.target_nodes.empty()) { in GetOpGroupsToRecompute()
203 subgraphs_to_recompute.push_back(current_recomputation); in GetOpGroupsToRecompute()