Home
last modified time | relevance | path

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

1234

/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/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()
317 if (e->IsControlEdge()) { in ReplaceFunctionCallWithPartitionedCall()
326 if (e->IsControlEdge()) { in ReplaceFunctionCallWithPartitionedCall()
427 if (e->IsControlEdge()) { in PredicateInt32Inputs()
Dencapsulate_xla_computations_pass.cc91 if (edge->IsControlEdge()) { in AddControlInputs()
100 if (edge->IsControlEdge()) { in AddControlOutputs()
218 if (!e->IsControlEdge() && in Encapsulate()
302 if (le->IsControlEdge()) { in BuildXlaLaunchOps()
313 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()
723 if (edge->IsControlEdge()) { in CopySubgraphEdges()
734 if (!edge->IsControlEdge()) { in CopySubgraphEdges()
745 if (edge->IsControlEdge()) { in CopySubgraphEdges()
757 if (!edge->IsControlEdge()) { in CopySubgraphEdges()
770 if (!edge->IsControlEdge()) { in CopySubgraphEdges()
928 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()
232 if (edges[i]->IsControlEdge() && edges[i]->dst()->IsSink()) { in FunctionalizeLoop()
235 TF_RET_CHECK(!edges[i]->IsControlEdge()) << edges[i]->src()->name(); in FunctionalizeLoop()
245 e->IsControlEdge() ? Graph::kControlSlot : 0); in FunctionalizeLoop()
271 if (!edge->IsControlEdge() && IsSwitch(edge->dst()) && in FunctionalizeLoop()
293 if (e->IsControlEdge() && e->dst()->IsSink()) { in FunctionalizeLoop()
453 if (in_edge->IsControlEdge()) { in FunctionalizeLoop()
469 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()
Drearrange_function_argument.cc112 if (e->IsControlEdge()) { in ReorderInputEdges()
137 if (e->IsControlEdge()) { in ReorderOutputEdges()
233 if (!e->IsControlEdge()) { in RearrangeOutputEdges()
408 if (e->IsControlEdge() || e->dst_input() == 0) { in MaybeRewriteIfNode()
470 if (!e->IsControlEdge()) { in MaybeRewriteIfNode()
Dfunctionalize_cond.cc447 if (e->IsControlEdge()) continue; in BuildArgumentNodes()
475 if (!e->IsControlEdge()) { in BuildArgumentNodes()
540 if (e->IsControlEdge()) continue; in ExtractBodies()
584 if (e->IsControlEdge()) { in ExtractBodies()
638 } else if (e->IsControlEdge()) { in ExtractBodies()
699 if (e->IsControlEdge()) { in ExtractBodies()
724 if (e->IsControlEdge()) continue; in ExtractBodies()
775 if (in_edge->IsControlEdge()) { in BuildIfNode()
888 bool control_edge = edge->IsControlEdge(); in AddOutputEdges()
1136 if (e->IsControlEdge()) { in StateAlongEdge()
[all …]
Dconst_analysis.cc204 if (!pred->IsControlEdge() && edge_filter(*pred)) { in BackwardsConstAnalysis()
231 if (!edge->IsControlEdge() && in BackwardsConstAnalysis()
/external/tensorflow/tensorflow/core/graph/
Dgradients.cc113 if (e->IsControlEdge()) continue; in AddSymGrad()
272 if (e->IsControlEdge()) continue; in InitBackprop()
294 if (e->IsControlEdge()) continue; in InitBackprop()
374 if (e->IsControlEdge()) continue; in Compute()
385 if (e->IsControlEdge()) continue; in Compute()
394 if (e->IsControlEdge()) continue; in Compute()
Dgraph_partition.cc118 if (e->IsControlEdge()) { in EdgeType()
127 if (edge->IsControlEdge()) { in NeedSameDeviceSendRecv()
152 if (edge->IsControlEdge()) return false; in IsDstInputOnHost()
205 if (!edge->IsControlEdge()) { in AddSend()
269 if (!edge->IsControlEdge()) { in AddRecv()
321 } else if (edge->IsControlEdge()) { in AddRecv()
382 if (!in_edge->IsControlEdge()) { in OptimizeControlFlowColocation()
394 if (!out_edge->IsControlEdge()) { in OptimizeControlFlowColocation()
1041 if (edge->IsControlEdge()) { in Partition()
1074 if (edge->IsControlEdge() || in Partition()
[all …]
Dmkl_layout_pass.cc1121 if (!e->IsControlEdge() && in GetConv2DOrBiasAdd()
1157 if (!e->IsControlEdge() && e->dst()->type_string() == csinfo_.conv2d) { in GetPadOrConv2D()
1168 if (!e->IsControlEdge() && e->src()->type_string() == csinfo_.pad) { in GetPadOrConv2D()
1206 if (!e->IsControlEdge() && in GetPadOrFusedConv2D()
1218 if (!e->IsControlEdge() && e->src()->type_string() == csinfo_.pad) { in GetPadOrFusedConv2D()
1277 if (!e->IsControlEdge() && in GetConv2DBackpropFilterOrBiasAddGrad()
1292 if (!e->IsControlEdge() && in GetConv2DBackpropFilterOrBiasAddGrad()
1341 if (e->IsControlEdge()) { in CheckForTranspose()
1348 if (e->IsControlEdge()) { in CheckForTranspose()
1357 if (!e->IsControlEdge()) { in CheckForTranspose()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
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()
Dparallel_concat_optimizer.cc83 if (input_edge->IsControlEdge()) { in Run()
109 if (e->IsControlEdge()) { in Run()
Dconstant_folding.cc67 if (in->IsControlEdge()) continue; in ReadPartialShapesFromShapeMap()
283 if (!in->IsControlEdge() && in ConsiderConstantFoldableNode()
365 if (!in_edge->IsControlEdge()) { in AddNodeToConstantGraph()
440 if (out_edge->IsControlEdge()) continue; in GetConstantGraph()
Dfunction.cc1330 if (e->IsControlEdge() || ret) { in GetTheOnlyDataEdge()
1374 if (out->IsControlEdge()) { in RemoveIdentityNodes()
1418 if (e->IsControlEdge()) { in RemoveListArrayConverter()
1450 if (e->IsControlEdge()) { in RemoveListArrayConverter()
1514 if (edge->IsControlEdge()) continue; in InputDevices()
1944 if (e->IsControlEdge()) { in InlineFunctionBody()
2009 return !e->src()->IsSource() && e->IsControlEdge(); in InlineFunctionBody()
2066 if (e->IsControlEdge()) { in InlineFunctionBody()
2100 if (!e->IsControlEdge()) { in InlineFunctionBody()
2113 if (e->IsControlEdge()) { in InlineFunctionBody()
[all …]
Dlower_if_op.cc193 if (e->IsControlEdge()) { in AddInputs()
241 if (e->IsControlEdge()) { in AddOutputs()
/external/tensorflow/tensorflow/cc/framework/
Dgradients.cc181 if (e->IsControlEdge()) continue; in GetReachableNodes()
218 if (e->IsControlEdge() || !reachable_nodes[e->dst()->id()]) continue; in GetStopBackpropNodes()
311 if (e->IsControlEdge() || !reachable_nodes[e->dst()->id()]) continue; in Initialize()
430 if (e->IsControlEdge()) continue; in ProcessWhileLoop()
468 if (e->IsControlEdge()) continue; in AddGradients()
496 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.cc65 bool IsControlEdge() const { return control_; } in IsControlEdge() function in tensorflow::tensorrt::segment::SimpleEdge
168 bool is_control = e->IsControlEdge(); in SimpleGraph()
358 if (in_edge->IsControlEdge()) { in ContractEdge()
378 if (out_edge->IsControlEdge()) { in ContractEdge()
500 if (out_edge->IsControlEdge()) { in SegmentGraph()
596 if (!edge->IsControlEdge() && !edge->src()->IsSource() && in SegmentGraph()
607 if (!edge->dst()->IsSink() && !edge->IsControlEdge() && in SegmentGraph()
/external/v8/src/compiler/
Dnode-properties.cc58 bool NodeProperties::IsControlEdge(Edge edge) { in IsControlEdge() function in v8::internal::compiler::NodeProperties
69 if (!NodeProperties::IsControlEdge(edge)) continue; in IsExceptionalCall()
83 if (!NodeProperties::IsControlEdge(edge)) continue; in FindSuccessfulControlProjection()
182 if (IsControlEdge(edge)) { in ReplaceUses()
266 if (!IsControlEdge(edge)) continue; in CollectControlProjections()
Dcontrol-equivalence.cc102 if (NodeProperties::IsControlEdge(edge)) { in RunUndirectedDFS()
132 if (NodeProperties::IsControlEdge(edge)) { in RunUndirectedDFS()

1234