Home
last modified time | relevance | path

Searched refs:IsControlEdge (Results 1 – 25 of 72) sorted by relevance

123

/external/tensorflow/tensorflow/core/distributed_runtime/
Dscheduler.cc39 if (edge->IsControlEdge()) { in InitializePending()
55 if (edge->IsControlEdge()) { in UpdatePending()
96 if (!out_edge->IsControlEdge() && in ComputeAsap()
130 if (edge->IsControlEdge()) { in ComputeAlap()
152 if (!in_edge->IsControlEdge() && in ComputeAlap()
240 if (!out_edge->IsControlEdge() && in ComputeSchedule()
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction_utils.cc105 if (e->IsControlEdge() || ret) { in GetTheOnlyDataEdge()
149 if (out->IsControlEdge()) { in RemoveIdentityNodes()
193 if (e->IsControlEdge()) { in RemoveListArrayConverter()
225 if (e->IsControlEdge()) { in RemoveListArrayConverter()
330 if (e->IsControlEdge()) { in ToGraphDef()
351 } else if (e->IsControlEdge()) { in ToGraphDef()
Dgradients.cc114 if (e->IsControlEdge()) continue; in AddSymGrad()
273 if (e->IsControlEdge()) continue; in InitBackprop()
295 if (e->IsControlEdge()) continue; in InitBackprop()
375 if (e->IsControlEdge()) continue; in Compute()
386 if (e->IsControlEdge()) continue; in Compute()
395 if (e->IsControlEdge()) continue; in Compute()
Dgraph_view.cc73 if (e->IsControlEdge()) { in CountOutputEdges()
161 if (e->IsControlEdge()) continue; in InitializeNode()
182 if (!e->IsControlEdge() || IsSink(e->dst())) continue; in InitializeNode()
302 if (IsSink(e->dst()) || !e->IsControlEdge()) { in SetScopedAllocatorAttrs()
319 if (IsSink(e->dst()) || !e->IsControlEdge()) { in SetScopedAllocatorAttrs()
419 if (!e->IsControlEdge()) { in SetAllocAttrs()
Daccumulate_n_optimizer.cc133 if (input_edge->IsControlEdge()) { in RewriteIntoTempVariable()
216 if (out_edge->IsControlEdge()) { in RewriteIntoTempVariable()
248 if (in_edge->IsControlEdge()) { in RewriteIntoAddN()
272 if (out_edge->IsControlEdge()) { in RewriteIntoAddN()
Dmkl_layout_pass.cc1142 if (!e->IsControlEdge() && in GetConv2DOrBiasAdd()
1183 if (!e->IsControlEdge() && e->dst()->type_string() == csinfo_.conv2d) { in GetPadOrConv2D()
1194 if (!e->IsControlEdge() && e->src()->type_string() == csinfo_.pad) { in GetPadOrConv2D()
1232 if (!e->IsControlEdge() && in GetPadOrFusedConv2D()
1244 if (!e->IsControlEdge() && e->src()->type_string() == csinfo_.pad) { in GetPadOrFusedConv2D()
1309 if (!e->IsControlEdge() && in GetConv2DBackpropFilterOrBiasAddGrad()
1326 if (!e->IsControlEdge() && in GetConv2DBackpropFilterOrBiasAddGrad()
1391 if (e->IsControlEdge()) { in CheckForTranspose()
1398 if (e->IsControlEdge()) { in CheckForTranspose()
1407 if (!e->IsControlEdge()) { in CheckForTranspose()
[all …]
Dinline_function_utils.cc127 if (edge->IsControlEdge()) continue; in InputDevices()
572 if (e->IsControlEdge()) { in InlineFunctionBody()
644 return !e->src()->IsSource() && e->IsControlEdge(); in InlineFunctionBody()
702 if (e->IsControlEdge()) { in InlineFunctionBody()
737 if (!e->IsControlEdge()) { in InlineFunctionBody()
750 if (e->IsControlEdge()) { in InlineFunctionBody()
759 caller->out_edges(), [](const Edge* e) { return e->IsControlEdge(); }); in InlineFunctionBody()
803 if (e->IsControlEdge()) { in InlineFunctionBody()
Dparallel_concat_optimizer.cc79 if (input_edge->IsControlEdge()) { in Run()
105 if (e->IsControlEdge()) { in Run()
Dconstant_folding.cc69 if (in->IsControlEdge()) continue; in ReadPartialShapesFromShapeMap()
310 if (!in->IsControlEdge() && in ConsiderConstantFoldableNode()
392 if (!in_edge->IsControlEdge()) { in AddNodeToConstantGraph()
467 if (out_edge->IsControlEdge()) continue; in GetConstantGraph()
/external/tensorflow/tensorflow/compiler/jit/
Dbuild_xla_ops_pass.cc127 [](const Edge* e) { return !e->IsControlEdge(); }); in MergeOutgoingDataEdges()
183 [](const Edge* e) { return e->IsControlEdge(); }); in MergeOutgoingControlEdges()
268 if (e->IsControlEdge()) { in CopyIncomingControlEdges()
279 [](const Edge* e) { return e->IsControlEdge(); }); in RemoveAllIncomingControlEdges()
303 n->in_edges(), [](const Edge* e) { return !e->IsControlEdge(); }); in ReplaceFunctionCallWithPartitionedCall()
307 if (!e->IsControlEdge()) { in ReplaceFunctionCallWithPartitionedCall()
321 if (e->IsControlEdge()) { in ReplaceFunctionCallWithPartitionedCall()
330 if (e->IsControlEdge()) { in ReplaceFunctionCallWithPartitionedCall()
431 if (e->IsControlEdge()) { in PredicateInt32Inputs()
Dencapsulate_xla_computations_pass.cc86 if (edge->IsControlEdge()) { in AddControlInputs()
95 if (edge->IsControlEdge()) { in AddControlOutputs()
212 if (!e->IsControlEdge() && in Encapsulate()
296 if (le->IsControlEdge()) { in BuildXlaLaunchOps()
307 TF_RET_CHECK(!oe->IsControlEdge()); in BuildXlaLaunchOps()
Dclone_constants_for_better_clustering.cc46 if (e->IsControlEdge()) { in CloneNode()
121 if (e->IsControlEdge()) { in CloneSmallHostConstantInputs()
Dencapsulate_subgraphs_pass.cc98 atleast_one_non_control_edge || !in->IsControlEdge(); in AreAllParentsGuaranteedConst()
99 if (!in->IsControlEdge() && runtime_const_nodes.count(in->src()) == 0) { in AreAllParentsGuaranteedConst()
721 if (edge->IsControlEdge()) { in CopySubgraphEdges()
732 if (!edge->IsControlEdge()) { in CopySubgraphEdges()
743 if (edge->IsControlEdge()) { in CopySubgraphEdges()
755 if (!edge->IsControlEdge()) { in CopySubgraphEdges()
768 if (!edge->IsControlEdge()) { in CopySubgraphEdges()
925 if (edge->IsControlEdge()) { in CopyEdgeToOutputGraph()
Dencapsulate_util.cc78 if (!e->IsControlEdge()) { in PreprocessControlEdgesBetweenOutsideCompilations()
123 if (e->IsControlEdge()) { in PreprocessDataEdgesBetweenOutsideCompilations()
231 if (e->IsControlEdge()) { in PostprocessDataEdgesBetweenOutsideCompilations()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_while.cc101 int src_output = squash_src_outputs[e->src()->id()] && !e->IsControlEdge() in CopySubgraph()
239 if (edges[i]->IsControlEdge() && edges[i]->dst()->IsSink()) { in FunctionalizeLoop()
242 TF_RET_CHECK(!edges[i]->IsControlEdge()) << edges[i]->src()->name(); in FunctionalizeLoop()
252 e->IsControlEdge() ? Graph::kControlSlot : 0); in FunctionalizeLoop()
278 if (!edge->IsControlEdge() && IsSwitch(edge->dst()) && in FunctionalizeLoop()
300 if (e->IsControlEdge() && e->dst()->IsSink()) { in FunctionalizeLoop()
454 if (in_edge->IsControlEdge()) { in FunctionalizeLoop()
470 if (in_edge->IsControlEdge()) { in FunctionalizeLoop()
Dresource_util.cc91 if (o->IsControlEdge()) continue; in PropagateFromStackOrTensorArraySourceOp()
119 if (o->IsControlEdge()) continue; in PropagateFromArgOp()
234 if (o->IsControlEdge()) continue; in PropagateThroughIdentityOp()
242 if (o->IsControlEdge()) continue; in PropagateThroughIdentityOp()
Dfunctionalize_cond.cc455 if (e->IsControlEdge()) continue; in BuildArgumentNodes()
483 if (!e->IsControlEdge()) { in BuildArgumentNodes()
548 if (e->IsControlEdge()) continue; in ExtractBodies()
592 if (e->IsControlEdge()) { in ExtractBodies()
646 } else if (e->IsControlEdge()) { in ExtractBodies()
707 if (e->IsControlEdge()) { in ExtractBodies()
732 if (e->IsControlEdge()) continue; in ExtractBodies()
783 if (in_edge->IsControlEdge()) { in BuildIfNode()
900 bool control_edge = edge->IsControlEdge(); in AddOutputEdges()
1148 if (e->IsControlEdge()) { in StateAlongEdge()
[all …]
Drearrange_function_argument.cc112 if (e->IsControlEdge()) { in ReorderInputEdges()
137 if (e->IsControlEdge()) { in ReorderOutputEdges()
233 if (!e->IsControlEdge()) { in RearrangeOutputEdges()
410 if (e->IsControlEdge() || e->dst_input() == 0) { in MaybeRewriteIfNode()
476 if (!e->IsControlEdge()) { in MaybeRewriteIfNode()
/external/tensorflow/tensorflow/core/graph/
Dgraph_partition.cc118 if (e->IsControlEdge()) { in EdgeType()
127 if (edge->IsControlEdge()) { in NeedSameDeviceSendRecv()
152 if (edge->IsControlEdge()) return false; in IsDstInputOnHost()
207 if (!edge->IsControlEdge()) { in AddSend()
271 if (!edge->IsControlEdge()) { in AddRecv()
323 } else if (edge->IsControlEdge()) { in AddRecv()
384 if (!in_edge->IsControlEdge()) { in OptimizeControlFlowColocation()
396 if (!out_edge->IsControlEdge()) { in OptimizeControlFlowColocation()
1043 if (edge->IsControlEdge()) { in Partition()
1076 if (edge->IsControlEdge() || in Partition()
[all …]
/external/tensorflow/tensorflow/cc/framework/
Dgradients.cc182 if (e->IsControlEdge()) continue; in GetReachableNodes()
219 if (e->IsControlEdge() || !reachable_nodes[e->dst()->id()]) continue; in GetStopBackpropNodes()
312 if (e->IsControlEdge() || !reachable_nodes[e->dst()->id()]) continue; in Initialize()
431 if (e->IsControlEdge()) continue; in ProcessWhileLoop()
469 if (e->IsControlEdge()) continue; in AddGradients()
497 if (e->IsControlEdge()) continue; in AddGradients()
526 if (e->IsControlEdge()) continue; in AddGradients()
Dops.cc31 if (e->IsControlEdge()) continue; in input()
57 if (e->IsControlEdge()) continue; in GetInputs()
/external/tensorflow/tensorflow/core/nccl/
Dnccl_rewrite.cc47 if (edge->IsControlEdge()) { in ReplaceReduce()
62 if (edge->IsControlEdge()) { in ReplaceReduce()
127 int dst_dev = edge->IsControlEdge() in ReplaceBroadcast()
173 if (edge->IsControlEdge()) { in ReplaceBroadcast()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dsegment.cc72 bool IsControlEdge() const { return control_; } in IsControlEdge() function in tensorflow::tensorrt::segment::__anon8f933a440111::SimpleEdge
175 bool is_control = e->IsControlEdge(); in SimpleGraph()
586 if (in_edge->IsControlEdge()) { in ContractEdge()
606 if (out_edge->IsControlEdge()) { in ContractEdge()
837 if (out_edge->IsControlEdge()) { in SegmentGraph()
947 if (!edge->IsControlEdge() && !edge->src()->IsSource() && in SegmentGraph()
958 if (!edge->dst()->IsSink() && !edge->IsControlEdge() && in SegmentGraph()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc275 if (edge->IsControlEdge()) { in AddControlInputs()
284 if (edge->IsControlEdge()) { in AddControlOutputs()
313 if (e->IsControlEdge()) { in RemoveIdentityNodesForArgRetval()
398 if (!e->IsControlEdge() && in MoveHeadOutsideCompilationToHost()
421 if (!e->IsControlEdge()) { in MoveHeadOutsideCompilationToHost()
522 if (!e->IsControlEdge() && in MoveHeadOutsideCompilationToHost()
1043 if (!e->IsControlEdge() && in MoveTailOutsideCompilationToHost()
1268 if (e->IsControlEdge() || in ReplaceArgUsedByOutsideCompilationWithPlaceholder()
1498 if (!e->IsControlEdge() && e->src()->IsArg() && e->dst()->IsRetval()) { in RemoveEdgesBetweenArgAndRetval()
1508 if (!e->IsControlEdge()) { in RemoveEdgesBetweenArgAndRetval()
[all …]
Dhost_training_loop_optimization_util.cc64 if (e->IsControlEdge()) continue; in IsExecuteNodeOrIdentityToExecuteNode()
97 if (output_edge->IsControlEdge() || output_node->IsExit()) continue; in ResourceOnlyUsedForTPUExecuteInLoop()
284 if (e_cond->IsControlEdge() || !e_cond->dst()->IsSwitch()) continue; in FindLoopExitNodes()
288 if (e_switch->IsControlEdge() || !e_switch->dst()->IsExit()) continue; in FindLoopExitNodes()

123