| /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 | 93 if (consumer_input == AsControlDependency(node.name())) { in SafeToRemoveIdentity()
429 new_input = AsControlDependency(NodeName(input_to_forward)); in OptimizeNode()
441 consumer->add_input(AsControlDependency(input->name())); in OptimizeNode()
449 consumer, AsControlDependency(node_name), node_map_.get()); in OptimizeNode()
731 it->second->add_input(AsControlDependency(*input)); in GroupCrossDeviceControlEdges()
742 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()
221 return AsControlDependency(*node); in AddControlDependency()
232 return AsControlDependency(*output); in AddControlDependency()
256 return AsControlDependency(*added_node); in AddControlDependency()
310 consumer->add_input(AsControlDependency(node->name())); in ForwardInputs()
489 node->set_input(0, AsControlDependency(NodeName(node->input(0)))); in MaterializeShapes()
884 const string ctrl_dep = AsControlDependency(node->input(i)); in MaterializeConstantValuedNode()
1464 *const_out->add_input() = AsControlDependency(*node); in FoldMergeNode()
1477 *const_index->add_input() = AsControlDependency(*node); in FoldMergeNode()
[all …]
|
| D | constant_folding_test.cc | 260 {"in1", AsControlDependency("paddings1"), AsControlDependency("c1")}, in PaddingWithZeroSize()
2090 AddNode("s1", "Identity", {"in1", AsControlDependency("split_dim")}, {}, in TEST_F()
2139 {"in1", AsControlDependency("size_splits1"), in TEST_F()
2140 AsControlDependency("split_dim")}, in TEST_F()
2189 {"in2", AsControlDependency("in1"), AsControlDependency("p2")}, {}, in TEST_F()
2223 AddNode("s2", "Identity", {"in2", AsControlDependency("in1")}, {}, &want); in TEST_F()
2272 {"in2", AsControlDependency("in1"), AsControlDependency("a2")}, {}, in TEST_F()
2307 {"in1", AsControlDependency("begin"), AsControlDependency("size")}, in TEST_F()
2355 {"in1", AsControlDependency("begin1"), AsControlDependency("size")}, in TEST_F()
2404 {"in1", AsControlDependency("begin"), AsControlDependency("end"), in TEST_F()
[all …]
|
| D | arithmetic_optimizer.cc | 121 if (input == new_input || AsControlDependency(input) == new_input) { in MaybeAddControlInput()
3199 node->set_input(1, AsControlDependency(y->name())); in TrySimplify()
3220 node->add_input(AsControlDependency(y->name())); in TrySimplify()
3231 node->set_input(1, AsControlDependency(y->name())); in TrySimplify()
3236 node->set_input(1, AsControlDependency(y->name())); in TrySimplify()
3252 node->set_input(0, AsControlDependency(x->name())); in TrySimplify()
3253 node->set_input(1, AsControlDependency(y->name())); in TrySimplify()
3259 node->set_input(1, AsControlDependency(y->name())); in TrySimplify()
3264 node->set_input(1, AsControlDependency(y->name())); in TrySimplify()
3376 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 | loop_optimizer.cc | 148 *consumer->mutable_input(i) = AsControlDependency(*const_node); in HandleConst()
|
| D | generic_layout_optimizer_transposer.cc | 519 is_in_frame ? AsControlDependency(src_node_def->name()) : ""; in UpdateEdge()
1265 new_node.add_input(AsControlDependency(string(depended_node))); in AddNodeShapeConst()
|
| D | function_optimizer.cc | 521 existing_control_deps.insert(AsControlDependency(NodeName(input))); in RemovePushedDownConstInputs()
|
| D | arithmetic_optimizer_test.cc | 3226 AddNode("out0", "Const", {AsControlDependency("x")}, {}, &want); in TEST_F()
3269 AddNode("out1", "Log1p", {"x2", AsControlDependency("x3")}, {}, &want); in TEST_F()
3306 AddNode("exp1", "Exp", {"x1", AsControlDependency("x3")}, {}, &want); in TEST_F()
3307 AddNode("out1", "Expm1", {"x1", AsControlDependency("x3")}, {}, &want); in TEST_F()
|
| D | remapper.cc | 2930 *new_shape.add_input() = AsControlDependency(scale); in AddBatchNormNodes()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils_test.cc | 219 TEST_F(UtilsTest, AsControlDependency) { in TEST_F() argument
222 EXPECT_EQ(AsControlDependency(node), "^foo"); in TEST_F()
223 EXPECT_EQ(AsControlDependency(node.name()), "^foo"); in TEST_F()
224 EXPECT_EQ(AsControlDependency("^foo"), "^foo"); in TEST_F()
|
| D | utils.h | 329 string AsControlDependency(const NodeDef& node);
333 string AsControlDependency(const string& node);
|
| D | utils.cc | 128 string AsControlDependency(const NodeDef& node) { in AsControlDependency() function
132 string AsControlDependency(const string& node_name) { in AsControlDependency() function
|
| D | mutable_graph_view.cc | 1003 AsControlDependency(fanout.node->name())); in GetControllingFaninToAdd()
1013 AsControlDependency(string(node_name))); in GetControllingFaninToAdd()
1483 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()
|