| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils.cc | 107 bool IsControlInput(const string& name) { in IsControlInput() function
111 bool IsControlInput(const TensorId& tensor_id) { return tensor_id.index() < 0; } in IsControlInput() function
193 if (num_inputs > 0 && IsControlInput(node.input(num_inputs - 1))) { in HasControlInputs()
201 if (num_inputs > 0 && !IsControlInput(node.input(0))) { in HasRegularInputs()
211 if (IsControlInput(input)) { in NumNonControlInputs()
222 if (!IsControlInput(input)) { in NumControlInputs()
232 if (IsControlInput(node_as_input)) break; in HasRegularOutputs()
247 if (!IsControlInput(node_as_input)) break; in HasControlOutputs()
263 if (!IsControlInput(node_as_input)) break; in NumControlOutputs()
278 if (IsControlInput(node_as_input)) { in NumNonControlOutputs()
[all …]
|
| D | utils.h | 288 bool IsControlInput(const string& name);
291 bool IsControlInput(const TensorId& tensor_id);
|
| D | mutable_graph_view_test.cc | 1701 string fanout = IsControlInput(fanin) ? AsControlDependency("c") : "c"; in TestUpdateFaninFromFaninToNodeAsSwitchControl()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | debug_stripper.cc | 51 if (!IsControlInput(inp)) { in Optimize()
67 if (!IsControlInput(node.input(i))) { in Optimize()
|
| D | dependency_optimizer.cc | 162 if (IsControlInput(input_node_name)) { in NumEdgesIfBypassed()
173 if (IsControlInput(consumer_input)) { in NumEdgesIfBypassed()
299 if (IsControlInput(old_input)) { in OptimizeNode()
405 (is_multi_input_identity && !IsControlInput(node->input(i)))) { in OptimizeNode()
409 CHECK(!IsControlInput(input_to_forward)); in OptimizeNode()
557 if (IsControlInput(input)) { in TransitiveReduction()
661 if (IsControlInput(node->input(j))) { in GroupCrossDeviceControlEdges()
706 if (IsControlInput(input_name)) { in GroupCrossDeviceControlEdges()
|
| D | model_pruner.cc | 170 add_as_control || IsControlInput(input), new_node, in ForwardInputsInternal()
194 if (!IsControlInput(new_node->input(i))) { in ForwardInputs()
244 if (IsControlInput(input) && in IdentityNTerminalPorts()
335 if (IsControlInput(input)) { in RewriteIdentityNAndInputsOutputs()
|
| D | loop_optimizer.cc | 92 if (IsControlInput(input)) { in HandleInvariantEnter()
147 if (IsControlInput(consumer->input(i))) { in HandleConst()
193 if (IsControlInput(node->input(i))) { in HandleInvariantNode()
285 if (IsControlInput(input) && NodeName(input) == node->name()) { in RevertInvariantNodes()
306 if (IsControlInput(input) && !IsConstant(*producer) && in RevertInvariantNodes()
348 if (!IsControlInput(input)) { in FindInvariantNodes()
646 if (IsControlInput(input)) continue; in CheckForDeadFanout()
|
| D | constant_folding.cc | 111 if (IsControlInput(input)) { in MaybeRemoveControlInput()
212 if (IsControlInput(input_name)) { in AddControlDependency()
271 if (IsControlInput(input) && consumers.size() > 1) { in ForwardInputs()
285 if (IsControlInput(consumer_input)) { in ForwardInputs()
972 if (IsControlInput(input)) { in IsFoldableUncached()
1404 if (IsControlInput(input)) { in FoldMergeNode()
1414 if (IsControlInput(fanin_of_input)) { in FoldMergeNode()
1513 if (IsControlInput(input)) { in FoldNode()
1520 if (!IsControlInput(input)) { in FoldNode()
1861 if (IsControlInput(node->input(i))) { in ReplaceOperationWithBroadcastTo()
[all …]
|
| D | arithmetic_optimizer.cc | 233 if (IsControlInput(src->input(i))) { in ForwardControlDependencies()
262 [](const string& input) { return IsControlInput(input); }); in IsDrivenByControlDependency()
368 if (IsControlInput(absorbed_node_input)) continue; in AbsorbInputByOptimizedNodesGroup()
395 if (IsControlInput(input_i)) continue; in CreateOptimizedNodesGroup()
804 if (IsControlInput(node->input(i))) { in GetCommonFactors()
874 if (IsControlInput(input)) { in GetUniqueFactors()
1580 if (!IsControlInput(input)) break; in GatherControlInputs()
1590 if (!IsControlInput(existing_input)) break; in AddControlInputs()
1612 if (IsControlInput(node.input(input_port))) { in InitializeChains()
1626 if (IsControlInput(output->input(0))) continue; in InitializeChains()
[all …]
|
| D | function_optimizer.cc | 471 if (IsControlInput(input)) in PushDownConstInputs()
480 if (IsControlInput(input)) continue; in PushDownConstInputs()
548 if (IsControlInput(input)) break; in RemovePushedDownConstInputTypes()
658 if (IsControlInput(input)) break; in InitializeFunctionSpecializationSignature()
|
| D | scoped_allocator_optimizer_test.cc | 335 if (IsControlInput(input)) { in ValidateSAControlInput()
352 if (IsControlInput(input)) { in NumControlInputs()
|
| D | implementation_selector.cc | 176 if (!IsControlInput(node_def->input(i))) break; in UpdateNodeDef()
|
| D | scoped_allocator_optimizer.cc | 269 if (!IsControlInput(input_name)) { in GetInputs()
320 if (IsControlInput(input_name)) { in GetDataInputs()
428 if (IsControlInput(input_name)) { in ClearInternalControlInputs()
|
| D | debug_stripper_test.cc | 68 if (IsControlInput(input)) { in TEST_F()
|
| D | memory_optimizer.cc | 638 if (IsControlInput(node->input(i))) { in SchedulingPass()
652 if (!IsControlInput(node->input(i))) { in SchedulingPass()
709 if (!IsControlInput(input)) { in SchedulingPass()
|
| D | layout_optimizer.cc | 338 if (!IsControlInput(node.input(i))) { in NonControlInputs()
375 if (node.input_size() > 0 && !IsControlInput(node.input(0))) { in DataInputPos()
|
| D | auto_mixed_precision.cc | 210 if (!IsControlInput(node.input(i))) { in NonControlInputs()
|
| D | constant_folding_test.cc | 1181 EXPECT_TRUE(IsControlInput(node.input(0))); in TEST_F()
1182 EXPECT_TRUE(IsControlInput(node.input(1))); in TEST_F()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | canonicalizer.cc | 32 if (IsControlInput(node->input(index))) { in CanonicalizeNode()
|
| D | functions.cc | 468 if (IsControlInput(graph_def_input)) { in AsFunctionDefInput()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | analytical_cost_estimator.cc | 85 if (IsControlInput(input)) { in AddCostNode()
|
| D | utils.cc | 94 if (IsControlInput(input_name)) { in ExtractExtraProperties()
|
| D | virtual_scheduler.cc | 506 if (curr_node_device == in_device || IsControlInput(input_node_name)) { in Init()
610 if (IsControlInput(input_source_name)) { in MaybeUpdateInputOutput()
|
| D | graph_properties.cc | 794 if (IsControlInput(input_tensor)) { in UpdateFunction()
1301 if (!IsControlInput(input)) { in AddFunction()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_constructor.cc | 369 static bool IsControlInput(const InputInfo& input) { in IsControlInput() function
1259 auto first_control = absl::c_find_if(inputs, &InputInfo::IsControlInput); in Convert()
|