Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/contrib/tensorrt/segment/
Dsegment.cc94 std::vector<const tensorflow::Edge*> out_edges(dst->out_edges().begin(), in ContractEdge() local
95 dst->out_edges().end()); in ContractEdge()
96 for (const tensorflow::Edge* out_edge : out_edges) { in ContractEdge()
112 for (const auto& out_edge : dst->out_edges()) { in ContractEdge()
171 for (const tensorflow::Edge* out_edge : node->out_edges()) { in SegmentGraph()
/external/tensorflow/tensorflow/compiler/jit/
Dbuild_xla_launch_ops_pass.cc101 std::vector<const Edge*> out_edges(node->out_edges().begin(), in ReplaceNodeWithXlaLaunch() local
102 node->out_edges().end()); in ReplaceNodeWithXlaLaunch()
103 for (const Edge* edge : out_edges) { in ReplaceNodeWithXlaLaunch()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util.cc256 Status SetNodeShardingFromNeighbors(Node* n, bool out_edges) { in SetNodeShardingFromNeighbors() argument
259 for (const Edge* edge : (out_edges ? n->out_edges() : n->in_edges())) { in SetNodeShardingFromNeighbors()
261 const Node* possible_match = out_edges ? edge->dst() : edge->src(); in SetNodeShardingFromNeighbors()
Dfunctionalize_control_flow.cc301 std::vector<const Edge*> edges(arg.enter->out_edges().begin(), in FunctionalizeLoop()
302 arg.enter->out_edges().end()); in FunctionalizeLoop()
341 for (const Edge* edge : frame->loop_cond->out_edges()) { in FunctionalizeLoop()
360 for (const Edge* e : arg.enter->out_edges()) { in FunctionalizeLoop()
407 for (const Edge* edge : arg.merge->out_edges()) { in FunctionalizeLoop()
432 for (const Edge* edge : arg.switch_node->out_edges()) { in FunctionalizeLoop()
458 for (const Edge* out : edge->dst()->out_edges()) { in FunctionalizeLoop()
529 std::vector<const Edge*> edges(arg.exit->out_edges().begin(), in FunctionalizeLoop()
530 arg.exit->out_edges().end()); in FunctionalizeLoop()
691 for (const Edge* e : node->out_edges()) { in IsDeadSwitch()
[all …]
Dtf2xla_util.h52 Status SetNodeShardingFromNeighbors(Node* n, bool out_edges);
Dtf2xla.cc105 for (const Edge* edge : feed_node->out_edges()) { in AddArgNodes()
/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/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/graph/
Dmkl_layout_pass.cc692 for (const Edge* e : bias_add_grad_inp->out_edges()) { in IsBiasAddGradInConv2DWithBiasContext()
754 for (const Edge* fe : first_inp_of_filter->out_edges()) { in IsBiasAddGradInConv2DWithBiasContext()
757 for (const Edge* ie : second_inp_of_input->out_edges()) { in IsBiasAddGradInConv2DWithBiasContext()
780 for (const Edge* fe : first_inp_of_filter->out_edges()) { in IsBiasAddGradInConv2DWithBiasContext()
1152 for (const Edge* e : filter_node->out_edges()) { in SetUpContiguousInputs()
1377 for (const Edge* e : orig_node->out_edges()) { in AddWorkSpaceEdgeIfNeeded()
1770 if (pred->out_edges().size() != 1) { in MergeNode()
1776 for (const Edge* e : pred->out_edges()) { in MergeNode()
1856 for (const Edge* e : pred->out_edges()) { in MergeNode()
1863 for (const Edge* e : succ->out_edges()) { in MergeNode()
[all …]
Dcontrol_flow.cc58 for (const Edge* out_edge : curr_node->out_edges()) { in BuildControlFlowInfo()
Doptimizer_cse.cc212 for (const Edge* e : n->out_edges()) { in Optimize()
Dalgorithm.cc223 if (!n->IsSink() && n->out_edges().empty()) { in FixupSourceAndSinkEdges()
Dsubgraph.cc114 for (const Edge* e : n->out_edges()) { in FeedInputs()
Dgradients.cc241 for (const Edge* e : n->out_edges()) { in InitBackprop()
/external/tensorflow/tensorflow/contrib/nccl/kernels/
Dnccl_rewrite.cc50 for (const auto& edge : node->out_edges()) { in ReplaceReduce()
124 for (const auto& edge : node->out_edges()) { in ReplaceBroadcast()
/external/tensorflow/tensorflow/core/common_runtime/
Dplacer.cc822 const Node* output = (*node->out_edges().begin())->dst(); in Run()
826 node->out_edges().begin(), node->out_edges().end(), in Run()
Dparallel_concat_optimizer.cc110 for (auto* e : n->out_edges()) { in Run()
Dfunction.cc1019 if (n->out_edges().empty()) continue; in RemoveIdentityNodes()
1026 for (const Edge* out : n->out_edges()) { in RemoveIdentityNodes()
1092 for (const Edge* e : n->out_edges()) { in RemoveListArrayConverter()
1248 for (const Edge* e : arg->out_edges()) { in InlineFunctionBody()
1293 for (const Edge* e : caller->out_edges()) { in InlineFunctionBody()
1360 if (n->out_edges().empty()) { in ToGraphDef()
Daccumulate_n_optimizer.cc167 for (const Edge* out_edge : n->out_edges()) { in rewriteNode()
Dconstant_folding.cc432 for (const Edge* out_edge : added_nodes.first->out_edges()) { in GetConstantGraph()
506 for (const Edge* out_edge : n->out_edges()) { in ReplaceTensorWithConstant()
Dexecutor.cc457 const size_t num_output_edges = n->out_edges().size(); in NodeItemBytes()
507 const size_t num_output_edges = n->out_edges().size(); in InitializeNode()
523 for (auto e : n->out_edges()) { in InitializeNode()
693 for (auto e : n->out_edges()) { in SetAllocAttrs()
1376 for (const Edge* out_edge : curr_node->out_edges()) { in BuildControlFlowInfo()
2340 for (const Edge* e : node->out_edges()) { in DeleteFrame()
/external/valgrind/coregrind/
Dm_transtab.c305 OutEdgeArr out_edges; member
948 OutEdgeArr__add(&from_tteC->out_edges, &oe); in VG_()
1019 m = OutEdgeArr__size(&from_tteC->out_edges); in unchain_in_preparation_for_deletion()
1022 OutEdge* oe = OutEdgeArr__index(&from_tteC->out_edges, j); in unchain_in_preparation_for_deletion()
1028 OutEdgeArr__deleteIndex(&from_tteC->out_edges, j); in unchain_in_preparation_for_deletion()
1032 n = OutEdgeArr__size(&here_tteC->out_edges); in unchain_in_preparation_for_deletion()
1034 OutEdge* oe = OutEdgeArr__index(&here_tteC->out_edges, i); in unchain_in_preparation_for_deletion()
1051 OutEdgeArr__makeEmpty(&here_tteC->out_edges); in unchain_in_preparation_for_deletion()
/external/tensorflow/tensorflow/cc/framework/
Dgradients.cc217 for (const Edge* e : n->out_edges()) { in GetStopBackpropNodes()
307 for (const Edge* e : n->out_edges()) { in Initialize()
/external/tensorflow/tensorflow/contrib/tensorrt/convert/
Dconvert_graph.cc82 for (const tensorflow::Edge* edge : node->out_edges()) { in GetSubGraphOutgoingEdges()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_constants_lib.cc242 for (const Edge* out_edge : node->out_edges()) { in FoldConstants()

12