Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Ddependency_optimizer.cc288 const string ctrl_input = ConstantFolding::AddControlDependency( in OptimizeNode() local
290 ctrl_inputs.insert(ctrl_input); in OptimizeNode()
291 node->set_input(pos, ctrl_input); in OptimizeNode()
292 node_map_->UpdateInput(node_name, old_input, ctrl_input); in OptimizeNode()
Dconstant_folding.cc101 bool MaybeAddControlInput(const string& ctrl_input, NodeDef* node, in MaybeAddControlInput() argument
105 if (input == ctrl_input || AsControlDependency(input) == ctrl_input) { in MaybeAddControlInput()
112 ConstantFolding::AddControlDependency(ctrl_input, graph, node_map); in MaybeAddControlInput()
114 node_map->AddOutput(NodeName(ctrl_input), node->name()); in MaybeAddControlInput()
Dfunction_optimizer.cc1999 for (const string& ctrl_input : add_ctrl_inputs) { in RestoreGraphInvariants() local
2000 node.add_input(ctrl_input); in RestoreGraphInvariants()