Home
last modified time | relevance | path

Searched refs:IsControlInput (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/
Dutils.cc172 bool IsControlInput(const string& name) { in IsControlInput() function
176 bool IsControlInput(const TensorId& tensor_id) { return tensor_id.index() < 0; } in IsControlInput() function
258 if (num_inputs > 0 && IsControlInput(node.input(num_inputs - 1))) { in HasControlInputs()
267 if (IsControlInput(input)) { in NumNonControlInputs()
278 if (IsControlInput(node_as_input)) { in NumNonControlOutputs()
301 if (!IsControlInput(input) && NodeName(input) == node.name()) { in NumNonControlDataOutputs()
331 (!follow_control_input && IsControlInput(current->input(0)))) { in GetTailOfChain()
369 if (!inputs.insert(NodeName(input)).second && IsControlInput(input)) { in DedupControlInputs()
Dutils.h109 bool IsControlInput(const string& name);
112 bool IsControlInput(const TensorId& tensor_id);
Dmutable_graph_view_test.cc1701 string fanout = IsControlInput(fanin) ? AsControlDependency("c") : "c"; in TestUpdateFaninFromFaninToNodeAsSwitchControl()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Ddependency_optimizer.cc140 if (IsControlInput(input_node_name)) { in NumEdgesIfBypassed()
151 if (IsControlInput(consumer_input)) { in NumEdgesIfBypassed()
277 if (IsControlInput(old_input)) { in OptimizeNode()
381 (is_multi_input_identity && !IsControlInput(node->input(i)))) { in OptimizeNode()
385 CHECK(!IsControlInput(input_to_forward)); in OptimizeNode()
498 if (IsControlInput(input)) { in TransitiveReduction()
611 if (IsControlInput(node->input(j))) { in GroupCrossDeviceControlEdges()
646 if (IsControlInput(input_name)) { in GroupCrossDeviceControlEdges()
Ddebug_stripper.cc40 if (!IsControlInput(inp)) { in Optimize()
56 if (!IsControlInput(node.input(i))) { in Optimize()
Dmodel_pruner.cc152 add_as_control || IsControlInput(input), new_node, in ForwardInputsInternal()
176 if (!IsControlInput(new_node->input(i))) { in ForwardInputs()
226 if (IsControlInput(input) && in IdentityNTerminalPorts()
317 if (IsControlInput(input)) { in RewriteIdentityNAndInputsOutputs()
Dloop_optimizer.cc93 if (IsControlInput(input)) { in HandleInvariantEnter()
148 if (IsControlInput(consumer->input(i))) { in HandleConst()
195 if (IsControlInput(node->input(i))) { in HandleInvariantNode()
287 if (IsControlInput(input) && NodeName(input) == node->name()) { in RevertInvariantNodes()
308 if (IsControlInput(input) && !IsConstant(*producer) && in RevertInvariantNodes()
350 if (!IsControlInput(input)) { in FindInvariantNodes()
Darithmetic_optimizer.cc224 if (IsControlInput(src->input(i))) { in ForwardControlDependencies()
245 [](const string& input) { return IsControlInput(input); }); in IsDrivenByControlDependency()
343 if (IsControlInput(absorbed_node_input)) continue; in AbsorbInputByOptimizedNodesGroup()
370 if (IsControlInput(input_i)) continue; in CreateOptimizedNodesGroup()
778 if (IsControlInput(node->input(i))) { in GetCommonFactors()
847 if (IsControlInput(input)) { in GetUniqueFactors()
1516 if (!IsControlInput(input)) break; in GatherControlInputs()
1526 if (!IsControlInput(existing_input)) break; in AddControlInputs()
1544 if (IsControlInput(node.input(input_port))) { in InitializeChains()
1558 if (IsControlInput(output->input(0))) continue; in InitializeChains()
[all …]
Dconstant_folding.cc128 if (IsControlInput(input)) { in MaybeRemoveControlInput()
240 if (IsControlInput(input_name)) { in AddControlDependency()
940 if (IsControlInput(input)) { in IsFoldable()
1242 if (IsControlInput(input)) { in FoldMergeNode()
1252 if (IsControlInput(fanin_of_input)) { in FoldMergeNode()
1343 if (IsControlInput(input) && in FoldNode()
1350 if (IsControlInput(fanin_of_input) && in FoldNode()
1654 if (IsControlInput(node->input(i))) { in ReplaceOperationWithIdentity()
1685 if (IsControlInput(node->input(i))) { in ReplaceOperationWithSnapshot()
1737 if (IsControlInput(node->input(i))) { in ReplaceOperationWithConstant()
[all …]
Ddebug_stripper_test.cc68 if (IsControlInput(input)) { in TEST_F()
Dfunction_optimizer.cc559 if (IsControlInput(input)) in PushDownConstInputs()
568 if (IsControlInput(input)) continue; in PushDownConstInputs()
637 if (IsControlInput(input)) break; in RemovePushedDownConstInputTypes()
748 if (IsControlInput(input)) break; in InitializeFunctionSpecializationSignature()
Dmemory_optimizer.cc611 if (IsControlInput(node->input(i))) { in SchedulingPass()
625 if (!IsControlInput(node->input(i))) { in SchedulingPass()
678 if (!IsControlInput(input)) { in SchedulingPass()
Dscoped_allocator_optimizer.cc134 if (!IsControlInput(input_name)) { in GetInputs()
261 if (IsControlInput(input_name)) { in ClearInternalControlInputs()
Dlayout_optimizer.cc330 if (!IsControlInput(node.input(i))) { in NonControlInputs()
367 if (node.input_size() > 0 && !IsControlInput(node.input(0))) { in DataInputPos()
Dconstant_folding_test.cc867 EXPECT_TRUE(IsControlInput(node.input(0))); in TEST_F()
868 EXPECT_TRUE(IsControlInput(node.input(1))); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/costs/
Danalytical_cost_estimator.cc83 if (IsControlInput(input)) { in AddCostNode()
Dutils.cc93 if (IsControlInput(input_name)) { in ExtractExtraProperties()
Dgraph_properties.cc624 if (IsControlInput(input_tensor)) { in UpdateFunction()
1057 if (!IsControlInput(input)) { in AddFunction()
Dvirtual_scheduler.cc488 if (IsControlInput(input_source_name)) { in MaybeUpdateInputOutput()
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.cc102 if (IsControlInput(func_def_input)) { in ExpandFunctionDefInput()
225 if (IsControlInput(graph_def_input)) { in AsFunctionDefInput()