/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()
|