Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A 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
452 AddControlDependency(node->input(0), graph_, node_map_.get()); in MaterializeShapes()
490 node->set_input(1, AddControlDependency(NodeName(node->input(1)), in MaterializeShapes()
542 string ctrl_dep = AddControlDependency(shape_n_node->name(), in MaterializeShapes()
707 AddControlDependency(node.name(), graph_, node_map_.get()); in MaterializeBroadcastGradientArgs()
821 AddControlDependency(node->input(1), graph_, node_map_.get()); in MaterializeReductionIndices()
1894 string ctrl_dep = AddControlDependency(NodeName(node->input(i)), graph, in ReplaceOperationWithBroadcastTo()
1914 AddControlDependency(node->input(i), graph, node_map_.get()); in ReplaceOperationWithBroadcastTo()
1944 AddControlDependency(node->input(i), graph, node_map_.get()); in ReplaceOperationWithIdentity()
[all …]
H A Dloop_optimizer.cc180 const string ctrl_dep = ConstantFolding::AddControlDependency( in HandleConst()
539 const string ctrl_dep = ConstantFolding::AddControlDependency( in RemoveStackOps()
978 const string ctrl_dep = ConstantFolding::AddControlDependency( in RemoveDeadBranches()
H A Dconstant_folding.h45 static string AddControlDependency(const string& input_name, GraphDef* graph,
H A Ddependency_optimizer.cc318 const string ctrl_input = ConstantFolding::AddControlDependency( in OptimizeNode()
H A Darithmetic_optimizer.cc128 ConstantFolding::AddControlDependency(new_input, graph, node_map); in MaybeAddControlInput()