Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmemory_optimizer.cc362 const std::unordered_set<string>& recomputed_node_names, in RecomputedOrOriginalNodeName() argument
364 if (recomputed_node_names.find(original_node_name) == in RecomputedOrOriginalNodeName()
365 recomputed_node_names.end()) { in RecomputedOrOriginalNodeName()
380 std::unordered_set<string> recomputed_node_names; in RecomputeSubgraph() local
388 recomputed_node_names.insert(original_node->name()); in RecomputeSubgraph()
406 recomputed_node_names, original_input_name); in RecomputeSubgraph()
417 target_input_name = RecomputedOrOriginalNodeName(recomputed_node_names, in RecomputeSubgraph()