Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/common_runtime/
Dplacer.cc203 if (IsGeneratorNode(node) && !node->out_edges().empty()) { in Run()
204 const Node* output = (*node->out_edges().begin())->dst(); in Run()
208 node->out_edges().begin(), node->out_edges().end(), in Run()
Dparallel_concat_optimizer.cc111 for (auto* e : n->out_edges()) { in Run()
Daccumulate_n_optimizer.cc183 for (const Edge* out_edge : n->out_edges()) { in rewriteNode()
Dexecutor.cc412 const size_t num_output_edges = n->out_edges().size(); in NodeItemBytes()
463 const size_t num_output_edges = n->out_edges().size(); in InitializeNode()
479 for (auto e : n->out_edges()) { in InitializeNode()
707 for (const auto& e : sa->out_edges()) { in SetScopedAllocatorAttrs()
724 for (const auto& e : use_node->out_edges()) { in SetScopedAllocatorAttrs()
756 for (const auto& e : n->out_edges()) { in SetAllocAttrs()
1467 for (const Edge* out_edge : curr_node->out_edges()) { in BuildControlFlowInfo()
2591 for (const Edge* e : node->out_edges()) { in DeleteFrame()
/external/tensorflow/tensorflow/compiler/jit/
Dbuild_xla_ops_pass.cc49 std::vector<const Edge*> out_edges(old_node->out_edges().begin(), in MoveOutgoingEdges() local
50 old_node->out_edges().end()); in MoveOutgoingEdges()
51 for (const Edge* edge : out_edges) { in MoveOutgoingEdges()
85 absl::c_copy_if(old_node->out_edges(), std::back_inserter(data_edges), in MergeOutgoingDataEdges()
114 absl::c_copy_if(old_node->out_edges(), std::back_inserter(ctrl_edges), in MergeOutgoingControlEdges()
260 for (const Edge* e : n->out_edges()) { in ReplaceFunctionCallWithPartionedCall()
Dpartially_decluster_pass.cc64 for (const Edge* e : n->out_edges()) { in FindNodesToDecluster()
110 for (const Edge* out_edge : n->out_edges()) { in PartiallyDeclusterNode()
145 if (n->out_edges().empty()) { in PartiallyDeclusterNode()
Dencapsulate_xla_computations_pass.cc98 for (const Edge* edge : node.out_edges()) { in AddControlOutputs()
300 for (const Edge* le : launch->out_edges()) { in BuildXlaLaunchOps()
311 for (const Edge* oe : output_node->out_edges()) { in BuildXlaLaunchOps()
Dencapsulate_util.cc224 for (auto e : n->out_edges()) { in PostprocessDataEdgesBetweenOutsideCompilations()
341 for (const Edge* e : g->source_node()->out_edges()) { in PreprocessEdgesBetweenOutsideCompilations()
Dclone_constants_for_better_clustering.cc103 if (is_small_host_constant && input->out_edges().size() != 1) { in CloneSmallHostConstantInputs()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dsegment.cc81 const std::vector<SimpleEdge*>& out_edges() const { return out_edges_; } in out_edges() function in tensorflow::tensorrt::segment::SimpleNode
144 out_edges_.reserve(node_->out_edges().size()); in SimpleNode()
375 std::vector<const SimpleEdge*> out_edges(dst->out_edges().begin(), in ContractEdge() local
376 dst->out_edges().end()); in ContractEdge()
377 for (const SimpleEdge* out_edge : out_edges) { in ContractEdge()
399 for (const auto& out_edge : dst->out_edges()) { in ContractEdge()
493 for (const SimpleEdge* out_edge : node->out_edges()) { in SegmentGraph()
602 for (const Edge* edge : node->out_edges()) { in SegmentGraph()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_while.cc308 std::vector<const Edge*> edges(arg.enter->out_edges().begin(), in FunctionalizeLoop()
309 arg.enter->out_edges().end()); in FunctionalizeLoop()
349 for (const Edge* edge : frame->loop_cond->out_edges()) { in FunctionalizeLoop()
368 for (const Edge* e : arg.enter->out_edges()) { in FunctionalizeLoop()
417 for (const Edge* edge : arg.merge->out_edges()) { in FunctionalizeLoop()
442 for (const Edge* edge : arg.switch_node->out_edges()) { in FunctionalizeLoop()
469 for (const Edge* out : edge->dst()->out_edges()) { in FunctionalizeLoop()
561 std::vector<const Edge*> edges(arg.exit->out_edges().begin(), in FunctionalizeLoop()
562 arg.exit->out_edges().end()); in FunctionalizeLoop()
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()
651 for (auto edge : node->out_edges()) { in RewriteAssociatedFunction()
723 std::vector<const Edge*> out_edges; in ReplaceNode() local
724 for (const Edge* edge : n->out_edges()) { in ReplaceNode()
725 out_edges.push_back(edge); in ReplaceNode()
729 for (const Edge* edge : out_edges) { in ReplaceNode()
Dfunctionalize_cond.cc438 for (const Edge* e : node->out_edges()) { in BuildArgumentNodes()
568 for (const Edge* e : n->out_edges()) { in ExtractBodies()
848 std::vector<const Edge*> edges(node->out_edges().begin(), in AddOutputEdges()
849 node->out_edges().end()); in AddOutputEdges()
1191 while (!node->out_edges().empty()) { in RemoveRedundantMerge()
1192 const Edge* oe = *node->out_edges().begin(); in RemoveRedundantMerge()
1244 while (!node->out_edges().empty()) { in RemoveRedundantSwitch()
1245 auto e = *node->out_edges().begin(); in RemoveRedundantSwitch()
1341 for (const Edge* e : s->out_edges()) { in DeleteReachableAndDeadNodes()
1354 for (const Edge* e : m->out_edges()) { in DeleteReachableAndDeadNodes()
[all …]
Dfunctionalize_control_flow_util.cc60 for (const Edge* out : current_node->out_edges()) { in CheckNodeNotInCycle()
Dtf2xla_util.h55 Status SetNodeShardingFromNeighbors(Node* n, bool out_edges);
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_stream_util.cc60 << " fanout " << e->src()->out_edges().size(); in AssignStreams()
85 const size_t fanout = e->src()->out_edges().size(); in AssignStreams()
/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/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()
258 if (!n->IsSink() && n->out_edges().empty()) { in FixupSourceAndSinkEdges()
Dmkl_layout_pass.cc877 for (const Edge* e : m->out_edges()) { in GetConv2DOrBiasAdd()
913 for (const Edge* e : m->out_edges()) { in GetPadOrConv2D()
962 for (const Edge* e : m->out_edges()) { in GetPadOrFusedConv2D()
1033 for (const Edge* e : g->out_edges()) { in GetConv2DBackpropFilterOrBiasAddGrad()
1048 for (const Edge* e : g->out_edges()) { in GetConv2DBackpropFilterOrBiasAddGrad()
1090 if (node->num_outputs() > 1 || node->out_edges().size() > 1) return false; in CheckForTranspose()
1097 for (const Edge* e : node->out_edges()) { in CheckForTranspose()
1171 if (node->out_edges().size() > 1) return false; in CheckForMklOp()
1732 for (const Edge* e : filter_node->out_edges()) { in SetUpContiguousInputs()
1946 for (const Edge* e : orig_node->out_edges()) { in AddWorkSpaceEdgeIfNeeded()
[all …]
Dcollective_order.cc52 node->out_edges().size()); in DiscoverDataDependencies()
54 for (const Edge* out_edge : node->out_edges()) { in DiscoverDataDependencies()
Dvalidate.cc93 for (const Edge* out : node->out_edges()) { in ValidateGraphHasNoCycle()
Dcontrol_flow.cc118 for (const Edge* out_edge : curr_node->out_edges()) { in BuildControlFlowInfo()
Doptimizer_cse.cc213 for (const Edge* e : n->out_edges()) { in Optimize()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc299 std::vector<const Edge*> out_edges(node->out_edges().begin(), in GetEngineInfo() local
300 node->out_edges().end()); in GetEngineInfo()
301 std::sort(out_edges.begin(), out_edges.end(), EdgePtrCompare()); in GetEngineInfo()
302 for (const auto edge : out_edges) { in GetEngineInfo()
663 for (auto edge : node->out_edges()) { in RegisterSegmentFunctionToFunctionLibrary()
/external/tensorflow/tensorflow/core/nccl/
Dnccl_rewrite.cc52 for (const auto& edge : node->out_edges()) { in ReplaceReduce()
126 for (const auto& edge : node->out_edges()) { in ReplaceBroadcast()

12