| /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 | 465 for (const Edge* out_edge : added_nodes.first->out_edges()) { in GetConstantGraph() local
466 if (node_map.count(out_edge->dst()) == 0) { in GetConstantGraph()
467 if (out_edge->IsControlEdge()) continue; in GetConstantGraph()
470 {{added_nodes.second[0], out_edge->src_output()}, in GetConstantGraph()
471 {added_nodes.first, out_edge->src_output()}}); in GetConstantGraph()
478 {{added_nodes.second[out_edge->src_output()], 0}, in GetConstantGraph()
479 {added_nodes.first, out_edge->src_output()}}); in GetConstantGraph()
543 for (const Edge* out_edge : n->out_edges()) { in ReplaceTensorWithConstant() local
544 if (out_edge->src_output() == tensor.second) { in ReplaceTensorWithConstant()
545 edges_to_remove.push_back(out_edge); in ReplaceTensorWithConstant()
|
| D | immutable_executor_state.cc | 325 for (const Edge* out_edge : curr_node->out_edges()) { in BuildControlFlowInfo() local
326 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/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 | 395 for (const Edge* out_edge : node->out_edges()) { in OptimizeControlFlowColocation() local
396 if (!out_edge->IsControlEdge()) { in OptimizeControlFlowColocation()
397 data_edge = out_edge; in OptimizeControlFlowColocation()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| D | segment.cc | 605 for (const SimpleEdge* out_edge : out_edges) { in ContractEdge() local
606 if (out_edge->IsControlEdge()) { in ContractEdge()
607 SimpleEdge* e = const_cast<SimpleEdge*>(out_edge); in ContractEdge()
610 SimpleEdge* e = const_cast<SimpleEdge*>(out_edge); in ContractEdge()
627 for (const auto& out_edge : dst->out_edges()) { in ContractEdge() local
628 remove_edges->push_back(out_edge); in ContractEdge()
834 for (const SimpleEdge* out_edge : node->out_edges()) { in SegmentGraph() local
835 VLOG(3) << "... out node " << out_edge->dst()->name() << " ( " in SegmentGraph()
836 << out_edge->dst()->id() << " <- " << node->id() << " )"; in SegmentGraph()
837 if (out_edge->IsControlEdge()) { in SegmentGraph()
[all …]
|
| D | segment_test.cc | 57 return [node_names](const Edge* out_edge) -> bool { in MakeOutputEdgeCandidateFn() argument
58 return node_names.find(out_edge->src()->name()) != node_names.end(); in MakeOutputEdgeCandidateFn()
|
| /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 | 1057 for (const Edge* out_edge : orig->out_edges()) { in MakePrunedGraphCopyAndInline() local
1058 auto iter = node_images->find(out_edge->dst()); in MakePrunedGraphCopyAndInline()
1062 ->AddEdge(image, out_edge->src_output(), iter->second, in MakePrunedGraphCopyAndInline()
1063 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 | 737 for (const OutEdgeInfo& out_edge : out_edge_info) { in ReplaceNode() local
738 g->AddEdge(new_node, out_edge.src_output, out_edge.dst, out_edge.dst_input); in ReplaceNode()
847 for (const Edge* out_edge : fwd_arg->out_edges()) { in RewriteTensorListWithConstElement() local
848 if (out_edge->dst()->type_string() == "TensorListPushBack") { in RewriteTensorListWithConstElement()
849 tl_push_nodes.push_back(out_edge->dst()); in RewriteTensorListWithConstElement()
884 for (const Edge* out_edge : bwd_arg->out_edges()) { in RewriteTensorListWithConstElement() local
885 if (out_edge->dst()->type_string() == "TensorListPopBack") { in RewriteTensorListWithConstElement()
886 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/tpu/graph_rewrite/ |
| D | host_training_loop_optimization_util.cc | 318 for (const auto out_edge : loop_switch_node->out_edges()) { in GetOrCreateBeforeEachIterationNode() local
319 if (out_edge->src_output() == 1) { in GetOrCreateBeforeEachIterationNode()
320 *node_out = out_edge->dst(); in GetOrCreateBeforeEachIterationNode()
|
| D | distributed_tpu_rewrite_pass.cc | 1866 for (const Edge* out_edge : body_node->out_edges()) { in InferXlaShardingFromNeighbors() local
1867 if (out_edge->IsControlEdge()) continue; in InferXlaShardingFromNeighbors()
1870 num_cores_per_replica, n.name(), *out_edge->dst(), &core, in InferXlaShardingFromNeighbors()
1873 TF_RETURN_IF_ERROR(parse_sharding_from_function(out_edge)); in InferXlaShardingFromNeighbors()
1941 for (const Edge* out_edge : edge->dst()->out_edges()) { in AssignArgsAndRetvalsToCores() local
1942 if (out_edge->dst()->type_string() == kTPUPartitionedOutput) { in AssignArgsAndRetvalsToCores()
1943 partitioned_output_nodes[edge->src_output()] = out_edge->dst(); in AssignArgsAndRetvalsToCores()
2956 for (const Edge* out_edge : edge->dst()->out_edges()) { in BuildExecuteNodes() local
2957 if (out_edge->dst()->type_string() == kTPUPartitionedOutput) { in BuildExecuteNodes()
2963 to_be_removed_nodes.push_back(out_edge->dst()); in BuildExecuteNodes()
[all …]
|
| D | encapsulate_tpu_computations_pass.cc | 2054 for (const Edge* out_edge : out_edges) { in RemoveOutputIdentityNodesForWhileV2() local
2055 if (out_edge->IsControlEdge()) { in RemoveOutputIdentityNodesForWhileV2()
2056 g->AddControlEdge(while_node, out_edge->dst()); in RemoveOutputIdentityNodesForWhileV2()
2058 Node* dst = out_edge->dst(); in RemoveOutputIdentityNodesForWhileV2()
2059 int dst_input = out_edge->dst_input(); in RemoveOutputIdentityNodesForWhileV2()
2060 g->RemoveEdge(out_edge); in RemoveOutputIdentityNodesForWhileV2()
|
| /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 | 163 bool operator()(const Edge* out_edge) const;
|
| D | convert_nodes.cc | 6599 bool OutputEdgeValidator::operator()(const Edge* out_edge) const { in operator ()()
6600 if (out_edge->IsControlEdge()) return true; in operator ()()
6601 if (out_edge->src()->type_string() == "Const") { in operator ()()
6602 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 | 797 for (const Edge* out_edge : node->out_edges()) { in ClusterizeNodes() local
798 const Node* dst_node = out_edge->dst(); in ClusterizeNodes()
802 strings::StrCat(node->name(), ":", out_edge->src_output()); in ClusterizeNodes()
|