Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmemory_optimizer.cc331 NodeDef* current_trigger_node = nullptr; in AddRecomputeControlDependencyNodes() local
340 if (current_trigger_node != nullptr) { in AddRecomputeControlDependencyNodes()
342 strings::StrCat("^", current_trigger_node->name()); in AddRecomputeControlDependencyNodes()
344 current_trigger_node = new_trigger_node; in AddRecomputeControlDependencyNodes()
345 triggers[original_recomputed_node] = current_trigger_node; in AddRecomputeControlDependencyNodes()
352 *current_trigger_node->add_input() = in AddRecomputeControlDependencyNodes()
354 VLOG(2) << " Recomputation trigger " << current_trigger_node->name() in AddRecomputeControlDependencyNodes()