| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | debug_stripper.cc | 52 inp = AsControlDependency(NodeName(inp)); in Optimize()
68 *node.mutable_input(i) = AsControlDependency(NodeName(node.input(i))); in Optimize()
|
| D | shape_optimizer.cc | 118 size_node.set_input(1, AsControlDependency(node)); in Optimize()
186 node.set_input(0, AsControlDependency(node.input(0))); in Optimize()
187 node.set_input(1, AsControlDependency(node.input(1))); in Optimize()
|
| D | dependency_optimizer.cc | 85 if (consumer_input == AsControlDependency(node.name())) { in SafeToRemoveIdentity()
421 new_input = AsControlDependency(NodeName(input_to_forward)); in OptimizeNode()
433 consumer->add_input(AsControlDependency(input->name())); in OptimizeNode()
441 consumer, AsControlDependency(node_name), node_map_.get()); in OptimizeNode()
723 it->second->add_input(AsControlDependency(*input)); in GroupCrossDeviceControlEdges()
734 node->add_input(AsControlDependency(*entry.second)); in GroupCrossDeviceControlEdges()
|
| D | model_pruner.cc | 156 add_as_control ? AsControlDependency(NodeName(input)) : input; in ForwardInputsInternal()
165 add_as_control ? AsControlDependency(NodeName(input)) : input; in ForwardInputsInternal()
174 add_as_control ? AsControlDependency(NodeName(input)) : input; in ForwardInputsInternal()
|
| D | constant_folding.cc | 88 if (input == ctrl_input || AsControlDependency(input) == ctrl_input) { in MaybeAddControlInput()
107 const string old_input_ctrl_dep = AsControlDependency(NodeName(old_input)); in MaybeRemoveControlInput()
217 return AsControlDependency(node); in AddControlDependency()
228 return AsControlDependency(*output); in AddControlDependency()
252 return AsControlDependency(*added_node); in AddControlDependency()
306 consumer->add_input(AsControlDependency(node->name())); in ForwardInputs()
485 node->set_input(0, AsControlDependency(NodeName(node->input(0)))); in MaterializeShapes()
880 const string ctrl_dep = AsControlDependency(node->input(i)); in MaterializeConstantValuedNode()
1438 *const_out->add_input() = AsControlDependency(*node); in FoldMergeNode()
1451 *const_index->add_input() = AsControlDependency(*node); in FoldMergeNode()
[all …]
|
| D | constant_folding_test.cc | 260 {"in1", AsControlDependency("paddings1"), AsControlDependency("c1")}, in PaddingWithZeroSize()
2085 AddNode("s1", "Identity", {"in1", AsControlDependency("split_dim")}, {}, in TEST_F()
2134 {"in1", AsControlDependency("size_splits1"), in TEST_F()
2135 AsControlDependency("split_dim")}, in TEST_F()
2184 {"in2", AsControlDependency("in1"), AsControlDependency("p2")}, {}, in TEST_F()
2218 AddNode("s2", "Identity", {"in2", AsControlDependency("in1")}, {}, &want); in TEST_F()
2267 {"in2", AsControlDependency("in1"), AsControlDependency("a2")}, {}, in TEST_F()
2302 {"in1", AsControlDependency("begin"), AsControlDependency("size")}, in TEST_F()
2350 {"in1", AsControlDependency("begin1"), AsControlDependency("size")}, in TEST_F()
2399 {"in1", AsControlDependency("begin"), AsControlDependency("end"), in TEST_F()
[all …]
|
| D | arithmetic_optimizer.cc | 118 if (input == new_input || AsControlDependency(input) == new_input) { in MaybeAddControlInput()
2602 node->set_input(1, AsControlDependency(y->name())); in TrySimplify()
2623 node->add_input(AsControlDependency(y->name())); in TrySimplify()
2634 node->set_input(1, AsControlDependency(y->name())); in TrySimplify()
2639 node->set_input(1, AsControlDependency(y->name())); in TrySimplify()
2655 node->set_input(0, AsControlDependency(x->name())); in TrySimplify()
2656 node->set_input(1, AsControlDependency(y->name())); in TrySimplify()
2662 node->set_input(1, AsControlDependency(y->name())); in TrySimplify()
2667 node->set_input(1, AsControlDependency(y->name())); in TrySimplify()
2779 node->add_input(AsControlDependency(y->name())); in TrySimplifyInternal()
[all …]
|
| D | memory_optimizer.cc | 685 AsControlDependency(NodeName(node->input(min_input_id))); in SchedulingPass()
731 *node->add_input() = AsControlDependency(accum->name()); in SchedulingPass()
|
| D | layout_optimizer.cc | 909 *const_node->add_input() = AsControlDependency(depended_node); in AddNodePermNHWCToNCHW()
921 *const_node->add_input() = AsControlDependency(depended_node); in AddNodePermNCHWToNHWC()
1420 *node->add_input() = AsControlDependency(depended_node); in AddNodeShapeConst()
|
| D | loop_optimizer.cc | 148 *consumer->mutable_input(i) = AsControlDependency(*const_node); in HandleConst()
|
| D | generic_layout_optimizer_transposer.cc | 523 is_in_frame ? AsControlDependency(src_node_def->name()) : ""; in UpdateEdge()
1270 new_node.add_input(AsControlDependency(string(depended_node))); in AddNodeShapeConst()
|
| D | function_optimizer.cc | 518 existing_control_deps.insert(AsControlDependency(NodeName(input))); in RemovePushedDownConstInputs()
|
| D | arithmetic_optimizer_test.cc | 2767 AddNode("out0", "Const", {AsControlDependency("x")}, {}, &want); in TEST_F()
2810 AddNode("out1", "Log1p", {"x2", AsControlDependency("x3")}, {}, &want); in TEST_F()
2847 AddNode("exp1", "Exp", {"x1", AsControlDependency("x3")}, {}, &want); in TEST_F()
2848 AddNode("out1", "Expm1", {"x1", AsControlDependency("x3")}, {}, &want); in TEST_F()
|
| D | remapper.cc | 1468 *new_shape.add_input() = AsControlDependency(scale); in IsCpuCompatibleDataType()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils_test.cc | 216 TEST_F(UtilsTest, AsControlDependency) { in TEST_F() argument
219 EXPECT_EQ(AsControlDependency(node), "^foo"); in TEST_F()
220 EXPECT_EQ(AsControlDependency(node.name()), "^foo"); in TEST_F()
221 EXPECT_EQ(AsControlDependency("^foo"), "^foo"); in TEST_F()
|
| D | utils.h | 315 string AsControlDependency(const NodeDef& node);
319 string AsControlDependency(const string& node);
|
| D | utils.cc | 143 string AsControlDependency(const NodeDef& node) { in AsControlDependency() function
147 string AsControlDependency(const string& node_name) { in AsControlDependency() function
|
| D | mutable_graph_view.cc | 981 AsControlDependency(fanout.node->name())); in GetControllingFaninToAdd()
991 AsControlDependency(string(node_name))); in GetControllingFaninToAdd()
1467 node->set_input(pos, AsControlDependency(control->name())); in UpdateAllRegularFaninsToControlling()
|
| D | mutable_graph_view_test.cc | 1701 string fanout = IsControlInput(fanin) ? AsControlDependency("c") : "c"; in TestUpdateFaninFromFaninToNodeAsSwitchControl()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | graph_view.cc | 1178 node->add_input(AsControlDependency(string(fanin_node_name))); in AddControllingFaninInternal()
|