| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | placer.cc | 203 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()
|
| D | parallel_concat_optimizer.cc | 111 for (auto* e : n->out_edges()) { in Run()
|
| D | accumulate_n_optimizer.cc | 183 for (const Edge* out_edge : n->out_edges()) { in rewriteNode()
|
| D | executor.cc | 412 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/ |
| D | build_xla_ops_pass.cc | 49 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()
|
| D | partially_decluster_pass.cc | 64 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()
|
| D | encapsulate_xla_computations_pass.cc | 98 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()
|
| D | encapsulate_util.cc | 224 for (auto e : n->out_edges()) { in PostprocessDataEdgesBetweenOutsideCompilations()
341 for (const Edge* e : g->source_node()->out_edges()) { in PreprocessEdgesBetweenOutsideCompilations()
|
| D | clone_constants_for_better_clustering.cc | 103 if (is_small_host_constant && input->out_edges().size() != 1) { in CloneSmallHostConstantInputs()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| D | segment.cc | 81 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/ |
| D | functionalize_while.cc | 308 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()
|
| D | tf2xla_util.cc | 139 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()
|
| D | functionalize_cond.cc | 438 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 …]
|
| D | functionalize_control_flow_util.cc | 60 for (const Edge* out : current_node->out_edges()) { in CheckNodeNotInCycle()
|
| D | tf2xla_util.h | 55 Status SetNodeShardingFromNeighbors(Node* n, bool out_edges);
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_stream_util.cc | 60 << " 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/ |
| D | scheduler.cc | 92 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/ |
| D | algorithm.cc | 69 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()
|
| D | mkl_layout_pass.cc | 877 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 …]
|
| D | collective_order.cc | 52 node->out_edges().size()); in DiscoverDataDependencies()
54 for (const Edge* out_edge : node->out_edges()) { in DiscoverDataDependencies()
|
| D | validate.cc | 93 for (const Edge* out : node->out_edges()) { in ValidateGraphHasNoCycle()
|
| D | control_flow.cc | 118 for (const Edge* out_edge : curr_node->out_edges()) { in BuildControlFlowInfo()
|
| D | optimizer_cse.cc | 213 for (const Edge* e : n->out_edges()) { in Optimize()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 299 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/ |
| D | nccl_rewrite.cc | 52 for (const auto& edge : node->out_edges()) { in ReplaceReduce()
126 for (const auto& edge : node->out_edges()) { in ReplaceBroadcast()
|