| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils.cc | 92 bool IsControlInput(absl::string_view name) { in IsControlInput() function
96 bool IsControlInput(const TensorId& tensor_id) { return tensor_id.index() < 0; } in IsControlInput() function
178 if (num_inputs > 0 && IsControlInput(node.input(num_inputs - 1))) { in HasControlInputs()
186 if (num_inputs > 0 && !IsControlInput(node.input(0))) { in HasRegularInputs()
196 if (IsControlInput(input)) { in NumNonControlInputs()
207 if (!IsControlInput(input)) { in NumControlInputs()
217 if (IsControlInput(node_as_input)) break; in HasRegularOutputs()
232 if (!IsControlInput(node_as_input)) break; in HasControlOutputs()
248 if (!IsControlInput(node_as_input)) break; in NumControlOutputs()
263 if (IsControlInput(node_as_input)) { in NumNonControlOutputs()
[all …]
|
| D | utils.h | 302 bool IsControlInput(absl::string_view name);
305 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 | 170 if (IsControlInput(input_node_name)) { in NumEdgesIfBypassed()
181 if (IsControlInput(consumer_input)) { in NumEdgesIfBypassed()
307 if (IsControlInput(old_input)) { in OptimizeNode()
413 (is_multi_input_identity && !IsControlInput(node->input(i)))) { in OptimizeNode()
417 CHECK(!IsControlInput(input_to_forward)); in OptimizeNode()
565 if (IsControlInput(input)) { in TransitiveReduction()
669 if (IsControlInput(node->input(j))) { in GroupCrossDeviceControlEdges()
714 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()
275 if (IsControlInput(input) && consumers.size() > 1) { in ForwardInputs()
289 if (IsControlInput(consumer_input)) { in ForwardInputs()
976 if (IsControlInput(input)) { in IsFoldableUncached()
1430 if (IsControlInput(input)) { in FoldMergeNode()
1440 if (IsControlInput(fanin_of_input)) { in FoldMergeNode()
1539 if (IsControlInput(input)) { in FoldNode()
1546 if (!IsControlInput(input)) { in FoldNode()
1910 if (IsControlInput(node->input(i))) { in ReplaceOperationWithBroadcastTo()
[all …]
|
| D | function_optimizer.cc | 474 if (IsControlInput(input)) in PushDownConstInputs()
483 if (IsControlInput(input)) continue; in PushDownConstInputs()
551 if (IsControlInput(input)) break; in RemovePushedDownConstInputTypes()
661 if (IsControlInput(input)) break; in InitializeFunctionSpecializationSignature()
|
| D | arithmetic_optimizer.cc | 209 if (IsControlInput(node.input(i))) { in AllRegularInputsEqual()
295 if (IsControlInput(src->input(i))) { in ForwardControlDependencies()
324 [](const string& input) { return IsControlInput(input); }); in IsDrivenByControlDependency()
430 if (IsControlInput(absorbed_node_input)) continue; in AbsorbInputByOptimizedNodesGroup()
457 if (IsControlInput(input_i)) continue; in CreateOptimizedNodesGroup()
866 if (IsControlInput(node->input(i))) { in GetCommonFactors()
936 if (IsControlInput(input)) { in GetUniqueFactors()
1642 if (!IsControlInput(input)) break; in GatherControlInputs()
1652 if (!IsControlInput(existing_input)) break; in AddControlInputs()
1674 if (IsControlInput(node.input(input_port))) { in InitializeChains()
[all …]
|
| 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 | auto_mixed_precision.cc | 273 if (!IsControlInput(node.input(i))) { in NonControlInputs()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | make_deterministic.cc | 196 if (IsControlInput(input)) { in ConvertMapOrInterleave()
309 DCHECK(!IsControlInput(map_node->input(i))); in SplitMap()
314 DCHECK(!IsControlInput(map_node->input(i))); in SplitMap()
319 DCHECK(IsControlInput(map_node->input(i))); in SplitMap()
422 if (IsControlInput(orig_node.input(i))) { in ConvertMapAndBatch()
|
| D | split_utils.cc | 184 if (IsControlInput(input_str)) { in RewriteInput()
479 if (!IsControlInput(input_str) && !IsFunctionArgument(input_str) && in SplitFunction()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | canonicalizer.cc | 32 if (IsControlInput(node->input(index))) { in CanonicalizeNode()
|
| D | functions.cc | 472 if (IsControlInput(graph_def_input)) { in AsFunctionDefInput()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | analytical_cost_estimator.cc | 86 if (IsControlInput(input)) { in AddCostNode()
|
| D | utils.cc | 95 if (IsControlInput(input_name)) { in ExtractExtraProperties()
|
| D | virtual_scheduler.cc | 511 if (curr_node_device == in_device || IsControlInput(input_node_name)) { in Init()
615 if (IsControlInput(input_source_name)) { in MaybeUpdateInputOutput()
|
| D | graph_properties.cc | 823 if (IsControlInput(input_tensor)) { in UpdateFunction()
1339 if (!IsControlInput(input)) { in AddFunction()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_constructor.cc | 366 static bool IsControlInput(const InputInfo& input) { in IsControlInput() function
1256 auto first_control = absl::c_find_if(inputs, &InputInfo::IsControlInput); in Convert()
|