| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | accumulate_n_optimizer.cc | 215 for (const Edge* out_edge : n->out_edges()) { in RewriteIntoTempVariable() local
216 if (out_edge->IsControlEdge()) { in RewriteIntoTempVariable()
217 g->AddControlEdge(clean_up_accumulator, out_edge->dst()); in RewriteIntoTempVariable()
219 g->AddEdge(clean_up_accumulator, 0, out_edge->dst(), in RewriteIntoTempVariable()
220 out_edge->dst_input()); in RewriteIntoTempVariable()
271 for (const Edge* out_edge : n->out_edges()) { in RewriteIntoAddN() local
272 if (out_edge->IsControlEdge()) { in RewriteIntoAddN()
273 g->AddControlEdge(add_n_node, out_edge->dst()); in RewriteIntoAddN()
275 g->AddEdge(add_n_node, 0, out_edge->dst(), out_edge->dst_input()); in RewriteIntoAddN()
|
| D | constant_folding.cc | 438 for (const Edge* out_edge : added_nodes.first->out_edges()) { in GetConstantGraph() local
439 if (node_map.count(out_edge->dst()) == 0) { in GetConstantGraph()
440 if (out_edge->IsControlEdge()) continue; in GetConstantGraph()
443 {{added_nodes.second[0], out_edge->src_output()}, in GetConstantGraph()
444 {added_nodes.first, out_edge->src_output()}}); in GetConstantGraph()
451 {{added_nodes.second[out_edge->src_output()], 0}, in GetConstantGraph()
452 {added_nodes.first, out_edge->src_output()}}); in GetConstantGraph()
516 for (const Edge* out_edge : n->out_edges()) { in ReplaceTensorWithConstant() local
517 if (out_edge->src_output() == tensor.second) { in ReplaceTensorWithConstant()
518 edges_to_remove.push_back(out_edge); in ReplaceTensorWithConstant()
|
| D | executor.cc | 1546 for (const Edge* out_edge : curr_node->out_edges()) { in BuildControlFlowInfo() local
1547 Node* out = out_edge->dst(); in BuildControlFlowInfo()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | scheduler.cc | 92 for (const Edge* out_edge : curr->out_edges()) { in ComputeAsap() local
95 const Node* out = out_edge->dst(); in ComputeAsap()
96 if (!out_edge->IsControlEdge() && in ComputeAsap()
110 bool is_ready = UpdatePending(out_edge, &pending_count); in ComputeAsap()
237 for (const Edge* out_edge : event.node->out_edges()) { in ComputeSchedule() local
239 const Node* out = out_edge->dst(); in ComputeSchedule()
240 if (!out_edge->IsControlEdge() && in ComputeSchedule()
252 bool is_ready = UpdatePending(out_edge, &pending_count); in ComputeSchedule()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| D | segment.cc | 377 for (const SimpleEdge* out_edge : out_edges) { in ContractEdge() local
378 if (out_edge->IsControlEdge()) { in ContractEdge()
379 SimpleEdge* e = const_cast<SimpleEdge*>(out_edge); in ContractEdge()
382 SimpleEdge* e = const_cast<SimpleEdge*>(out_edge); in ContractEdge()
399 for (const auto& out_edge : dst->out_edges()) { in ContractEdge() local
400 remove_edges->push_back(out_edge); in ContractEdge()
497 for (const SimpleEdge* out_edge : node->out_edges()) { in SegmentGraph() local
498 VLOG(3) << "... out node " << out_edge->dst()->name() << " ( " in SegmentGraph()
499 << out_edge->dst()->id() << " <- " << node->id() << " )"; in SegmentGraph()
500 if (out_edge->IsControlEdge()) { in SegmentGraph()
[all …]
|
| D | segment_test.cc | 58 return [node_names](const Edge* out_edge) -> bool { in MakeOutputEdgeCandidateFn() argument
59 return node_names.find(out_edge->src()->name()) != node_names.end(); in MakeOutputEdgeCandidateFn()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | algorithm.cc | 69 for (const Edge* out_edge : n->out_edges()) { in DFSFromHelper() local
70 if (!edge_filter || edge_filter(*out_edge)) { in DFSFromHelper()
71 nodes_sorted.emplace_back(out_edge->dst()); in DFSFromHelper()
79 for (const Edge* out_edge : n->out_edges()) { in DFSFromHelper() local
80 if (!edge_filter || edge_filter(*out_edge)) { in DFSFromHelper()
81 add_work(out_edge->dst()); in DFSFromHelper()
|
| D | control_flow.cc | 118 for (const Edge* out_edge : curr_node->out_edges()) { in BuildControlFlowInfo() local
119 const Node* out = out_edge->dst(); in BuildControlFlowInfo()
|
| D | collective_order.cc | 54 for (const Edge* out_edge : node->out_edges()) { in DiscoverDataDependencies() local
55 auto& child_deps = (*data_dependencies)[out_edge->dst()]; in DiscoverDataDependencies()
|
| D | graph_partition.cc | 393 for (const Edge* out_edge : node->out_edges()) { in OptimizeControlFlowColocation() local
394 if (!out_edge->IsControlEdge()) { in OptimizeControlFlowColocation()
395 data_edge = out_edge; in OptimizeControlFlowColocation()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | partially_decluster_pass.cc | 122 for (const Edge* out_edge : n->out_edges()) { in PartiallyDeclusterNode() local
123 if (out_edge->IsControlEdge()) { in PartiallyDeclusterNode()
127 Node* dst = out_edge->dst(); in PartiallyDeclusterNode()
131 out_edges_to_clone.push_back(out_edge); in PartiallyDeclusterNode()
|
| D | node_matchers.cc | 348 OutEdge out_edge, in MatchAndExplain() argument
351 if (!src_matcher_.MatchAndExplain(out_edge.first, &inner_listener)) { in MatchAndExplain()
362 if (out_edge.second != src_oidx_) { in MatchAndExplain()
365 << " but found " << out_edge.second; in MatchAndExplain()
|
| D | encapsulate_subgraphs_pass.cc | 1060 for (const Edge* out_edge : orig->out_edges()) { in MakePrunedGraphCopyAndInline() local
1061 auto iter = node_images->find(out_edge->dst()); in MakePrunedGraphCopyAndInline()
1065 ->AddEdge(image, out_edge->src_output(), iter->second, in MakePrunedGraphCopyAndInline()
1066 out_edge->dst_input()); in MakePrunedGraphCopyAndInline()
|
| D | deadness_analysis.cc | 1290 for (const Edge* out_edge : curr_node->out_edges()) { in GetFrameBasedTopologicalOrder() local
1291 Node* out = out_edge->dst(); in GetFrameBasedTopologicalOrder()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_util.cc | 731 for (const OutEdgeInfo& out_edge : out_edge_info) { in ReplaceNode() local
732 g->AddEdge(new_node, out_edge.src_output, out_edge.dst, out_edge.dst_input); in ReplaceNode()
841 for (const Edge* out_edge : fwd_arg->out_edges()) { in RewriteTensorListWithConstElement() local
842 if (out_edge->dst()->type_string() == "TensorListPushBack") { in RewriteTensorListWithConstElement()
843 tl_push_nodes.push_back(out_edge->dst()); in RewriteTensorListWithConstElement()
878 for (const Edge* out_edge : bwd_arg->out_edges()) { in RewriteTensorListWithConstElement() local
879 if (out_edge->dst()->type_string() == "TensorListPopBack") { in RewriteTensorListWithConstElement()
880 tl_pop_nodes.push_back(out_edge->dst()); in RewriteTensorListWithConstElement()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_constants_lib.cc | 267 for (const Edge* out_edge : node->out_edges()) { in FoldConstants() local
268 if (out_edge->IsControlEdge()) continue; in FoldConstants()
270 const int output_idx = out_edge->src_output(); in FoldConstants()
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | graph_transferer.cc | 393 for (const Edge* out_edge : original_input_node->out_edges()) { in TransformGraphToAddAggregatedInputNode() local
394 Node* dst = out_edge->dst(); in TransformGraphToAddAggregatedInputNode()
395 int dst_port = out_edge->dst_input(); in TransformGraphToAddAggregatedInputNode()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.h | 156 bool operator()(const Edge* out_edge) const;
|
| D | convert_nodes.cc | 5937 bool OutputEdgeValidator::operator()(const Edge* out_edge) const { in operator ()()
5938 if (out_edge->IsControlEdge()) return true; in operator ()()
5939 if (out_edge->src()->type_string() == "Const") { in operator ()()
5940 VLOG(1) << "--> Need to remove output node " << out_edge->src()->name() in operator ()()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_execute_utils.cc | 795 for (const Edge* out_edge : node->out_edges()) { in ClusterizeNodes() local
796 const Node* dst_node = out_edge->dst(); in ClusterizeNodes()
800 strings::StrCat(node->name(), ":", out_edge->src_output()); in ClusterizeNodes()
|