Home
last modified time | relevance | path

Searched refs:out_edges (Results 1 – 25 of 69) sorted by relevance

123

/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_view.cc68 DCHECK_LE(n->out_edges().size(), kint32max); in CountOutputEdges()
71 for (auto e : n->out_edges()) { in CountOutputEdges()
160 for (auto e : n->out_edges()) { in InitializeNode()
181 for (auto e : n->out_edges()) { in InitializeNode()
251 if (n->out_edges().size() > kint32max) { in Initialize()
254 " output edges. Node ", n->name(), " had ", n->out_edges().size(), in Initialize()
301 for (const auto& e : sa->out_edges()) { in SetScopedAllocatorAttrs()
318 for (const auto& e : use_node->out_edges()) { in SetScopedAllocatorAttrs()
418 for (const auto& e : n->out_edges()) { in SetAllocAttrs()
Dplacer.cc265 if (IsGeneratorNode(node) && !node->out_edges().empty()) { in Run()
266 const Node* output = (*node->out_edges().begin())->dst(); in Run()
270 node->out_edges().begin(), node->out_edges().end(), in Run()
Dfunction_utils.cc141 if (n->out_edges().empty()) continue; in RemoveIdentityNodes()
148 for (const Edge* out : n->out_edges()) { in RemoveIdentityNodes()
224 for (const Edge* e : n->out_edges()) { in RemoveListArrayConverter()
307 if (n->out_edges().empty()) { in ToGraphDef()
Dplacer_inspection_required_ops_utils.cc238 const EdgeSet& out_edges = node->out_edges(); in AddOutputIdentities() local
239 std::vector<const Edge*> edge_vector(out_edges.begin(), out_edges.end()); in AddOutputIdentities()
Dmkl_layout_pass.cc1141 for (const Edge* e : m->out_edges()) { in GetConv2DOrBiasAdd()
1182 for (const Edge* e : m->out_edges()) { in GetPadOrConv2D()
1231 for (const Edge* e : m->out_edges()) { in GetPadOrFusedConv2D()
1308 for (const Edge* e : g->out_edges()) { in GetConv2DBackpropFilterOrBiasAddGrad()
1325 for (const Edge* e : g->out_edges()) { in GetConv2DBackpropFilterOrBiasAddGrad()
1383 if (node->num_outputs() > 1 || node->out_edges().size() > 1) return false; in CheckForTranspose()
1390 for (const Edge* e : node->out_edges()) { in CheckForTranspose()
1459 if (node->out_edges().size() > 1) return false; in CheckForMklOp()
2242 for (const Edge* e : filter_node->out_edges()) { in SetUpContiguousInputs()
2502 for (const Edge* e : orig_node->out_edges()) { in AddWorkSpaceEdgeIfNeeded()
[all …]
Daccumulate_n_optimizer.cc215 for (const Edge* out_edge : n->out_edges()) { in RewriteIntoTempVariable()
271 for (const Edge* out_edge : n->out_edges()) { in RewriteIntoAddN()
Dreplicate_per_replica_nodes.cc165 if (replicated_node->IsArg() && replicated_node->out_edges().empty()) { in RemoveDeadReplicatedArgs()
205 for (const Edge* edge : node->out_edges()) { in ReplicateEdges()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util.cc139 for (const Edge* e : arg_node->out_edges()) { in ReplaceArgUsageWithConstNode()
500 Status SetNodeShardingFromNeighbors(Node* n, bool out_edges) { in SetNodeShardingFromNeighbors() argument
503 for (const Edge* edge : (out_edges ? n->out_edges() : n->in_edges())) { in SetNodeShardingFromNeighbors()
505 const Node* possible_match = out_edges ? edge->dst() : edge->src(); in SetNodeShardingFromNeighbors()
650 for (auto edge : node->out_edges()) { in RewriteAssociatedFunction()
722 std::vector<const Edge*> out_edges; in ReplaceNode() local
723 for (const Edge* edge : n->out_edges()) { in ReplaceNode()
724 out_edges.push_back(edge); in ReplaceNode()
728 for (const Edge* edge : out_edges) { in ReplaceNode()
803 for (const Edge* e : n->out_edges()) { in RewriteTensorListWithConstElement()
[all …]
Dfunctionalize_while.cc236 std::vector<const Edge*> edges(arg.enter->out_edges().begin(), in FunctionalizeLoop()
237 arg.enter->out_edges().end()); in FunctionalizeLoop()
277 for (const Edge* edge : frame->loop_cond->out_edges()) { in FunctionalizeLoop()
296 for (const Edge* e : arg.enter->out_edges()) { in FunctionalizeLoop()
345 for (const Edge* edge : arg.merge->out_edges()) { in FunctionalizeLoop()
366 for (const Edge* edge : arg.switch_node->out_edges()) { in FunctionalizeLoop()
393 for (const Edge* out : edge->dst()->out_edges()) { in FunctionalizeLoop()
479 std::vector<const Edge*> edges(arg.exit->out_edges().begin(), in FunctionalizeLoop()
480 arg.exit->out_edges().end()); in FunctionalizeLoop()
Drearrange_function_argument.cc136 for (const Edge* e : n->out_edges()) { in ReorderOutputEdges()
231 std::vector<const Edge*> out_edges; in RearrangeOutputEdges() local
232 for (const Edge* e : n->out_edges()) { in RearrangeOutputEdges()
234 out_edges.push_back(e); in RearrangeOutputEdges()
237 for (const Edge* e : out_edges) { in RearrangeOutputEdges()
474 std::vector<const Edge*> out_edges; in MaybeRewriteIfNode() local
475 for (const Edge* e : n->out_edges()) { in MaybeRewriteIfNode()
477 out_edges.push_back(e); in MaybeRewriteIfNode()
480 for (const Edge* e : out_edges) { in MaybeRewriteIfNode()
Dresource_util.cc68 for (const Edge* o : n.out_edges()) { in OutputEdgesByIndex()
90 for (const Edge* o : n.out_edges()) { in PropagateFromStackOrTensorArraySourceOp()
118 for (const Edge* o : n.out_edges()) { in PropagateFromArgOp()
233 for (const Edge* o : n.out_edges()) { in PropagateThroughIdentityOp()
241 for (const Edge* o : n.out_edges()) { in PropagateThroughIdentityOp()
Dfunctionalize_cond.cc454 for (const Edge* e : node->out_edges()) { in BuildArgumentNodes()
584 for (const Edge* e : n->out_edges()) { in ExtractBodies()
889 std::vector<const Edge*> edges(node->out_edges().begin(), in AddOutputEdges()
890 node->out_edges().end()); in AddOutputEdges()
1238 while (!node->out_edges().empty()) { in RemoveRedundantMerge()
1239 const Edge* oe = *node->out_edges().begin(); in RemoveRedundantMerge()
1291 while (!node->out_edges().empty()) { in RemoveRedundantSwitch()
1292 auto e = *node->out_edges().begin(); in RemoveRedundantSwitch()
1389 for (const Edge* e : s->out_edges()) { in DeleteReachableAndDeadNodes()
1408 for (const Edge* e : m->out_edges()) { in DeleteReachableAndDeadNodes()
[all …]
/external/tensorflow/tensorflow/compiler/jit/
Dbuild_xla_ops_pass.cc66 std::vector<const Edge*> out_edges(old_node->out_edges().begin(), in MoveOutgoingEdges() local
67 old_node->out_edges().end()); in MoveOutgoingEdges()
68 for (const Edge* edge : out_edges) { in MoveOutgoingEdges()
126 absl::c_copy_if(old_node->out_edges(), std::back_inserter(data_edges), in MergeOutgoingDataEdges()
182 absl::c_copy_if(old_node->out_edges(), std::back_inserter(ctrl_edges), in MergeOutgoingControlEdges()
328 for (const Edge* e : n->out_edges()) { in ReplaceFunctionCallWithPartitionedCall()
Dshape_inference.cc149 std::vector<const Edge*> out_edges(n->out_edges().begin(), in PropagateShapes() local
150 n->out_edges().end()); in PropagateShapes()
151 for (const Edge* e : out_edges) { in PropagateShapes()
Dpartially_decluster_pass.cc76 for (const Edge* e : n->out_edges()) { in FindNodesToDecluster()
122 for (const Edge* out_edge : n->out_edges()) { in PartiallyDeclusterNode()
157 if (n->out_edges().empty()) { in PartiallyDeclusterNode()
Dencapsulate_xla_computations_pass.cc94 for (const Edge* edge : node.out_edges()) { in AddControlOutputs()
295 for (const Edge* le : launch->out_edges()) { in BuildXlaLaunchOps()
306 for (const Edge* oe : output_node->out_edges()) { in BuildXlaLaunchOps()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc283 for (const Edge* edge : node.out_edges()) { in AddControlOutputs()
309 for (const Edge* e : n->out_edges()) { in RemoveIdentityNodesForArgRetval()
397 for (const Edge* e : n->out_edges()) { in MoveHeadOutsideCompilationToHost()
420 for (const Edge* e : n->out_edges()) { in MoveHeadOutsideCompilationToHost()
521 for (const Edge* e : n->out_edges()) { in MoveHeadOutsideCompilationToHost()
799 for (const Edge* e : n->out_edges()) { in RemoveUnusedXlaInput()
1010 for (const Edge* e : n->out_edges()) { in MoveTailOutsideCompilationToHost()
1086 for (const Edge* e : xla_node->out_edges()) { in MoveTailOutsideCompilationToHost()
1128 for (const Edge* e : n->out_edges()) { in MoveTailOutsideCompilationToHost()
1267 for (const Edge* e : n->out_edges()) { in ReplaceArgUsedByOutsideCompilationWithPlaceholder()
[all …]
Dhost_training_loop_optimization_util.cc41 for (const auto e : src->out_edges()) { in FindEdgeConnecting()
63 for (const Edge* e : node->out_edges()) { in IsExecuteNodeOrIdentityToExecuteNode()
95 for (const Edge* output_edge : enter_node->out_edges()) { in ResourceOnlyUsedForTPUExecuteInLoop()
283 for (const auto e_cond : loop_cond.out_edges()) { in FindLoopExitNodes()
287 for (const auto e_switch : switch_node->out_edges()) { in FindLoopExitNodes()
318 for (const auto out_edge : loop_switch_node->out_edges()) { in GetOrCreateBeforeEachIterationNode()
Dvariable_merger_pass.cc85 for (const Edge* e : nodes[i]->out_edges()) { in MergeVarHandleOps()
123 for (const Edge* e : nodes[i]->out_edges()) { in MergeReadVariableOps()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dscheduler.cc92 for (const Edge* out_edge : curr->out_edges()) { in ComputeAsap()
129 for (const Edge* edge : n->out_edges()) { in ComputeAlap()
136 pending_count[n->id()] = n->out_edges().size(); in ComputeAlap()
237 for (const Edge* out_edge : event.node->out_edges()) { in ComputeSchedule()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dsegment.cc88 const std::vector<SimpleEdge*>& out_edges() const { return out_edges_; } in out_edges() function in tensorflow::tensorrt::segment::__anon3618f86b0111::SimpleNode
151 out_edges_.reserve(node_->out_edges().size()); in SimpleNode()
603 std::vector<const SimpleEdge*> out_edges(dst->out_edges().begin(), in ContractEdge() local
604 dst->out_edges().end()); in ContractEdge()
605 for (const SimpleEdge* out_edge : out_edges) { in ContractEdge()
627 for (const auto& out_edge : dst->out_edges()) { in ContractEdge()
834 for (const SimpleEdge* out_edge : node->out_edges()) { in SegmentGraph()
957 for (const Edge* edge : node->out_edges()) { in SegmentGraph()
/external/tensorflow/tensorflow/core/graph/
Dalgorithm.cc69 for (const Edge* out_edge : n->out_edges()) { in DFSFromHelper()
79 for (const Edge* out_edge : n->out_edges()) { in DFSFromHelper()
264 if (!n->IsSink() && n->out_edges().empty()) { in FixupSourceAndSinkEdges()
Dcollective_order.cc52 node->out_edges().size()); in DiscoverDataDependencies()
54 for (const Edge* out_edge : node->out_edges()) { in DiscoverDataDependencies()
/external/tensorflow/tensorflow/core/kernels/data/
Dsingle_threaded_executor.cc156 arg_output_locations_[arg_index].reserve(arg_node->out_edges().size()); in Initialize()
157 for (const Edge* e : arg_node->out_edges()) { in Initialize()
176 for (const Edge* e : n->out_edges()) { in Initialize()
199 for (const Edge* e : n->out_edges()) { in Initialize()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc214 std::vector<const Edge*> out_edges(node->out_edges().begin(), in GetEngineInfo() local
215 node->out_edges().end()); in GetEngineInfo()
216 std::sort(out_edges.begin(), out_edges.end(), EdgePtrCompare()); in GetEngineInfo()
217 for (const auto edge : out_edges) { in GetEngineInfo()

123