Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dshape_optimizer.cc79 fanout.node->set_input(1, AsControlDependency(node)); in Optimize()
127 node.set_input(0, AsControlDependency(node.input(0))); in Optimize()
128 node.set_input(1, AsControlDependency(node.input(1))); in Optimize()
Ddebug_stripper.cc41 inp = AsControlDependency(NodeName(inp)); in Optimize()
57 *node.mutable_input(i) = AsControlDependency(NodeName(node.input(i))); in Optimize()
Ddependency_optimizer.cc85 if (consumer_input == AsControlDependency(node.name())) { in SafeToRemoveIdentity()
397 new_input = AsControlDependency(NodeName(input_to_forward)); in OptimizeNode()
409 consumer->add_input(AsControlDependency(input->name())); in OptimizeNode()
417 updated_consumer |= RemoveInput(consumer, AsControlDependency(node_name), in OptimizeNode()
657 it->second->add_input(AsControlDependency(*input)); in GroupCrossDeviceControlEdges()
668 node->add_input(AsControlDependency(*entry.second)); in GroupCrossDeviceControlEdges()
Dmodel_pruner.cc138 add_as_control ? AsControlDependency(NodeName(input)) : input; in ForwardInputsInternal()
147 add_as_control ? AsControlDependency(NodeName(input)) : input; in ForwardInputsInternal()
156 add_as_control ? AsControlDependency(NodeName(input)) : input; in ForwardInputsInternal()
Dfunction_optimizer.cc607 existing_control_deps.insert(AsControlDependency(NodeName(input))); in RemovePushedDownConstInputs()
1086 *func_body_node.add_input() = AsControlDependency(func_inputs->name()); in InlineDirectFunctionCall()
1224 *inlined_node.add_input() = AsControlDependency( in InlineSymbolicGradient()
1776 AsControlDependency(inlined_node_name(inputs_ready_node->name()))); in InlineIndirectFunctionCall()
1801 AsControlDependency(inlined_node_name(empty_inputs_hook[0])); in InlineIndirectFunctionCall()
1868 AsControlDependency(tensor.node())); in InlineIndirectFunctionCall()
1880 AsControlDependency(node_name)); in InlineIndirectFunctionCall()
1926 AsControlDependency(inlined_node_name(kSideEffectsExecutedNodeName))); in InlineIndirectFunctionCall()
1989 add_ctrl_inputs.insert(AsControlDependency(override)); in RestoreGraphInvariants()
Dconstant_folding.cc105 if (input == ctrl_input || AsControlDependency(input) == ctrl_input) { in MaybeAddControlInput()
124 const string old_input_ctrl_dep = AsControlDependency(NodeName(old_input)); in MaybeRemoveControlInput()
245 return AsControlDependency(*node); in AddControlDependency()
257 return AsControlDependency(*output); in AddControlDependency()
281 return AsControlDependency(*added_node); in AddControlDependency()
434 node->set_input(0, AsControlDependency(NodeName(node->input(0)))); in MaterializeShapes()
822 const string ctrl_dep = AsControlDependency(node->input(i)); in MaterializeConstantValuedNode()
1276 *const_out->add_input() = AsControlDependency(*node); in FoldMergeNode()
1289 *const_index->add_input() = AsControlDependency(*node); in FoldMergeNode()
1407 *output->mutable_input(i) = AsControlDependency(*constant_output); in FoldNode()
[all …]
Dconstant_folding_test.cc1771 AddNode("s1", "Identity", {"in1", AsControlDependency("split_dim")}, {}, in TEST_F()
1820 {"in1", AsControlDependency("size_splits1"), in TEST_F()
1821 AsControlDependency("split_dim")}, in TEST_F()
1870 {"in2", AsControlDependency("in1"), AsControlDependency("p2")}, {}, in TEST_F()
1904 AddNode("s2", "Identity", {"in2", AsControlDependency("in1")}, {}, &want); in TEST_F()
1953 {"in2", AsControlDependency("in1"), AsControlDependency("a2")}, {}, in TEST_F()
1988 {"in1", AsControlDependency("begin"), AsControlDependency("size")}, in TEST_F()
2036 {"in1", AsControlDependency("begin1"), AsControlDependency("size")}, in TEST_F()
2085 {"in1", AsControlDependency("begin"), AsControlDependency("end"), in TEST_F()
2086 AsControlDependency("strides")}, in TEST_F()
[all …]
Darithmetic_optimizer.cc116 if (input == new_input || AsControlDependency(input) == new_input) { in MaybeAddControlInput()
2506 node->set_input(1, AsControlDependency(y->name())); in TrySimplify()
2514 node->set_input(1, AsControlDependency(y->name())); in TrySimplify()
2519 node->set_input(1, AsControlDependency(y->name())); in TrySimplify()
2545 node->set_input(0, AsControlDependency(x->name())); in TrySimplify()
2546 node->set_input(1, AsControlDependency(y->name())); in TrySimplify()
2553 node->set_input(1, AsControlDependency(y->name())); in TrySimplify()
2558 node->set_input(1, AsControlDependency(y->name())); in TrySimplify()
2671 node->add_input(AsControlDependency(y->name())); in TrySimplifyInternal()
2757 node->set_input(1, AsControlDependency(ones->name())); in TrySimplify()
[all …]
Darithmetic_optimizer_test.cc2674 AddNode("out2", "Square", {"x", AsControlDependency("y2")}, {}, &want); in TEST_F()
2675 AddNode("out1", "Identity", {"x", AsControlDependency("y1")}, {}, &want); in TEST_F()
2676 AddNode("out.5", "Sqrt", {"x", AsControlDependency("y.5")}, {}, &want); in TEST_F()
2678 {AsControlDependency("x"), AsControlDependency("y0")}, {}, &want); in TEST_F()
2679 AddNode("out_.5", "Rsqrt", {"x", AsControlDependency("y_.5")}, {}, &want); in TEST_F()
2680 AddNode("out_1", "Reciprocal", {"x", AsControlDependency("y_1")}, {}, &want); in TEST_F()
2723 {"x2", AsControlDependency("x1"), AsControlDependency("x3")}, {}, in TEST_F()
2762 AddNode("exp1", "Exp", {"x1", AsControlDependency("x3")}, {}, &want); in TEST_F()
2764 {"x1", AsControlDependency("x2"), AsControlDependency("x3")}, {}, in TEST_F()
Dlayout_optimizer.cc888 *const_node->add_input() = AsControlDependency(depended_node); in AddNodePermNHWCToNCHW()
900 *const_node->add_input() = AsControlDependency(depended_node); in AddNodePermNCHWToNHWC()
1361 *node->add_input() = AsControlDependency(depended_node); in AddNodeShapeConst()
Dmemory_optimizer.cc654 AsControlDependency(NodeName(node->input(min_input_id))); in SchedulingPass()
700 *node->add_input() = AsControlDependency(accum->name()); in SchedulingPass()
Dloop_optimizer.cc149 *consumer->mutable_input(i) = AsControlDependency(*const_node); in HandleConst()
Dremapper.cc649 *new_shape->add_input() = AsControlDependency(scale); in AddBatchNormNodes()
/external/tensorflow/tensorflow/core/grappler/
Dutils_test.cc214 TEST_F(UtilsTest, AsControlDependency) { in TEST_F() argument
217 EXPECT_EQ(AsControlDependency(node), "^foo"); in TEST_F()
218 EXPECT_EQ(AsControlDependency(node.name()), "^foo"); in TEST_F()
219 EXPECT_EQ(AsControlDependency("^foo"), "^foo"); in TEST_F()
Dutils.h236 string AsControlDependency(const NodeDef& node);
240 string AsControlDependency(const string& node);
Dutils.cc208 string AsControlDependency(const NodeDef& node) { in AsControlDependency() function
212 string AsControlDependency(const string& node_name) { in AsControlDependency() function
Dmutable_graph_view.cc978 AsControlDependency(fanout.node->name())); in GetControllingFaninToAdd()
988 AsControlDependency(string(node_name))); in GetControllingFaninToAdd()
1464 node->set_input(pos, AsControlDependency(control->name())); in UpdateAllRegularFaninsToControlling()
Dmutable_graph_view_test.cc1701 string fanout = IsControlInput(fanin) ? AsControlDependency("c") : "c"; in TestUpdateFaninFromFaninToNodeAsSwitchControl()