Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc112 ConstantFolding::AddControlDependency(ctrl_input, graph, node_map); in MaybeAddControlInput()
237 string ConstantFolding::AddControlDependency(const string& input_name, in AddControlDependency() function in tensorflow::grappler::ConstantFolding
397 AddControlDependency(node->input(0), graph_, node_map_.get()); in MaterializeShapes()
435 node->set_input(1, AddControlDependency(NodeName(node->input(1)), in MaterializeShapes()
485 string ctrl_dep = AddControlDependency(shape_n_node->name(), in MaterializeShapes()
649 AddControlDependency(node.name(), graph_, node_map_.get()); in MaterializeBroadcastGradientArgs()
762 AddControlDependency(node->input(1), graph_, node_map_.get()); in MaterializeReductionIndices()
1658 AddControlDependency(node->input(i), graph, node_map_.get()); in ReplaceOperationWithIdentity()
1689 AddControlDependency(node->input(i), graph, node_map_.get()); in ReplaceOperationWithSnapshot()
1701 AddControlDependency(node->input(1), graph, node_map_.get()); in ReplaceDivisionOfOnesByReciprocal()
[all …]
Dconstant_folding.h42 static string AddControlDependency(const string& input_name, GraphDef* graph,
Dloop_optimizer.cc182 const string ctrl_dep = ConstantFolding::AddControlDependency( in HandleConst()
540 const string ctrl_dep = ConstantFolding::AddControlDependency( in RemoveStackOps()
Ddependency_optimizer.cc288 const string ctrl_input = ConstantFolding::AddControlDependency( in OptimizeNode()
Darithmetic_optimizer.cc123 ConstantFolding::AddControlDependency(new_input, graph, node_map); in MaybeAddControlInput()