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.cc95 ConstantFolding::AddControlDependency(ctrl_input, graph, node_map); in MaybeAddControlInput()
209 string ConstantFolding::AddControlDependency(const string& input_name, in AddControlDependency() function in tensorflow::grappler::ConstantFolding
448 AddControlDependency(node->input(0), graph_, node_map_.get()); in MaterializeShapes()
486 node->set_input(1, AddControlDependency(NodeName(node->input(1)), in MaterializeShapes()
538 string ctrl_dep = AddControlDependency(shape_n_node->name(), in MaterializeShapes()
703 AddControlDependency(node.name(), graph_, node_map_.get()); in MaterializeBroadcastGradientArgs()
817 AddControlDependency(node->input(1), graph_, node_map_.get()); in MaterializeReductionIndices()
1845 string ctrl_dep = AddControlDependency(NodeName(node->input(i)), graph, in ReplaceOperationWithBroadcastTo()
1865 AddControlDependency(node->input(i), graph, node_map_.get()); in ReplaceOperationWithBroadcastTo()
1895 AddControlDependency(node->input(i), graph, node_map_.get()); in ReplaceOperationWithIdentity()
[all …]
Dconstant_folding.h45 static string AddControlDependency(const string& input_name, GraphDef* graph,
Dloop_optimizer.cc180 const string ctrl_dep = ConstantFolding::AddControlDependency( in HandleConst()
539 const string ctrl_dep = ConstantFolding::AddControlDependency( in RemoveStackOps()
Ddependency_optimizer.cc310 const string ctrl_input = ConstantFolding::AddControlDependency( in OptimizeNode()
Darithmetic_optimizer.cc125 ConstantFolding::AddControlDependency(new_input, graph, node_map); in MaybeAddControlInput()