| /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 | 462 for (const Edge* out_edge : added_nodes.first->out_edges()) { in GetConstantGraph() local
463 if (node_map.count(out_edge->dst()) == 0) { in GetConstantGraph()
464 if (out_edge->IsControlEdge()) continue; in GetConstantGraph()
467 {{added_nodes.second[0], out_edge->src_output()}, in GetConstantGraph()
468 {added_nodes.first, out_edge->src_output()}}); in GetConstantGraph()
475 {{added_nodes.second[out_edge->src_output()], 0}, in GetConstantGraph()
476 {added_nodes.first, out_edge->src_output()}}); in GetConstantGraph()
540 for (const Edge* out_edge : n->out_edges()) { in ReplaceTensorWithConstant() local
541 if (out_edge->src_output() == tensor.second) { in ReplaceTensorWithConstant()
542 edges_to_remove.push_back(out_edge); in ReplaceTensorWithConstant()
|
| D | forward_type_inference.cc | 281 for (const auto& out_edge : n->out_edges()) { in Run() local
282 if (out_edge->IsControlEdge()) { in Run()
285 Node* c = out_edge->dst(); in Run()
|
| D | immutable_executor_state.cc | 331 for (const Edge* out_edge : curr_node->out_edges()) { in BuildControlFlowInfo() local
332 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 | 394 for (const Edge* out_edge : node->out_edges()) { in OptimizeControlFlowColocation() local
395 if (!out_edge->IsControlEdge()) { in OptimizeControlFlowColocation()
396 data_edge = out_edge; in OptimizeControlFlowColocation()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| D | segment.cc | 610 for (const SimpleEdge* out_edge : out_edges) { in ContractEdge() local
611 if (out_edge->IsControlEdge()) { in ContractEdge()
612 SimpleEdge* e = const_cast<SimpleEdge*>(out_edge); in ContractEdge()
615 SimpleEdge* e = const_cast<SimpleEdge*>(out_edge); in ContractEdge()
632 for (const auto& out_edge : dst->out_edges()) { in ContractEdge() local
633 remove_edges->push_back(out_edge); in ContractEdge()
994 for (const SimpleEdge* out_edge : node->out_edges()) { in SegmentGraph() local
995 VLOG(3) << "... out node " << out_edge->dst()->name() << " ( " in SegmentGraph()
996 << out_edge->dst()->id() << " <- " << node->id() << " )"; in SegmentGraph()
997 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 | 1031 for (const Edge* out_edge : orig->out_edges()) { in MakePrunedGraphCopyAndInline() local
1032 auto iter = node_images->find(out_edge->dst()); in MakePrunedGraphCopyAndInline()
1036 ->AddEdge(image, out_edge->src_output(), iter->second, in MakePrunedGraphCopyAndInline()
1037 out_edge->dst_input()); in MakePrunedGraphCopyAndInline()
|
| D | deadness_analysis.cc | 1288 for (const Edge* out_edge : curr_node->out_edges()) { in GetFrameBasedTopologicalOrder() local
1289 Node* out = out_edge->dst(); in GetFrameBasedTopologicalOrder()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_util.cc | 907 for (const OutEdgeInfo& out_edge : out_edge_info) { in ReplaceNode() local
908 g->AddEdge(new_node, out_edge.src_output, out_edge.dst, out_edge.dst_input); in ReplaceNode()
1035 for (const Edge* out_edge : fwd_arg->out_edges()) { in RewriteTensorListWithConstElement() local
1036 if (out_edge->dst()->type_string() == "TensorListPushBack") { in RewriteTensorListWithConstElement()
1037 tl_push_nodes.push_back(out_edge->dst()); in RewriteTensorListWithConstElement()
1072 for (const Edge* out_edge : bwd_arg->out_edges()) { in RewriteTensorListWithConstElement() local
1073 if (out_edge->dst()->type_string() == "TensorListPopBack") { in RewriteTensorListWithConstElement()
1074 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 | 315 for (const auto out_edge : loop_switch_node->out_edges()) { in GetOrCreateBeforeEachIterationNode() local
316 if (out_edge->src_output() == 1) { in GetOrCreateBeforeEachIterationNode()
317 *node_out = out_edge->dst(); in GetOrCreateBeforeEachIterationNode()
|
| D | distributed_tpu_rewrite_pass.cc | 2079 for (const Edge* out_edge : body_node->out_edges()) { in InferXlaShardingFromNeighbors() local
2080 if (out_edge->IsControlEdge()) continue; in InferXlaShardingFromNeighbors()
2083 num_cores_per_replica, n.name(), *out_edge->dst(), &core, in InferXlaShardingFromNeighbors()
2086 TF_RETURN_IF_ERROR(parse_sharding_from_function(out_edge)); in InferXlaShardingFromNeighbors()
2154 for (const Edge* out_edge : edge->dst()->out_edges()) { in AssignArgsAndRetvalsToCores() local
2155 if (out_edge->dst()->type_string() == kTPUPartitionedOutput) { in AssignArgsAndRetvalsToCores()
2156 partitioned_output_nodes[edge->src_output()] = out_edge->dst(); in AssignArgsAndRetvalsToCores()
3316 for (const Edge* out_edge : edge->dst()->out_edges()) { in BuildExecuteNodes() local
3317 if (out_edge->dst()->type_string() == kTPUPartitionedOutput) { in BuildExecuteNodes()
3323 to_be_removed_nodes.push_back(out_edge->dst()); in BuildExecuteNodes()
[all …]
|
| D | encapsulate_tpu_computations_pass.cc | 2008 for (const Edge* out_edge : out_edges) { in RemoveOutputIdentityNodesForWhileV2() local
2009 if (out_edge->IsControlEdge()) { in RemoveOutputIdentityNodesForWhileV2()
2010 g->AddControlEdge(while_node, out_edge->dst()); in RemoveOutputIdentityNodesForWhileV2()
2012 Node* dst = out_edge->dst(); in RemoveOutputIdentityNodesForWhileV2()
2013 int dst_input = out_edge->dst_input(); in RemoveOutputIdentityNodesForWhileV2()
2014 g->RemoveEdge(out_edge); in RemoveOutputIdentityNodesForWhileV2()
|
| /external/tensorflow/tensorflow/dtensor/cc/ |
| D | dtensor_device_util.cc | 889 for (const Edge* out_edge : node->out_edges()) { in IdentifyAllFunctionsToExecute() local
890 if (out_edge->IsControlEdge()) continue; in IdentifyAllFunctionsToExecute()
892 const Node* retval_or_identity_node = out_edge->dst(); in IdentifyAllFunctionsToExecute()
902 output_edges[global_index] = out_edge; in IdentifyAllFunctionsToExecute()
|
| D | dtensor_device.cc | 1190 for (const Edge* out_edge : n->out_edges()) { in SelectGraphToExecute() local
1191 if (out_edge->IsControlEdge()) continue; in SelectGraphToExecute()
1192 Node* out_node = out_edge->dst(); in SelectGraphToExecute()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_functional_ops.cc | 1187 for (const Edge* out_edge : flatten_reshape_node->out_edges()) { in InsertReshapeNodePairs() local
1188 if (out_edge->dst() == recover_reshape_node) { in InsertReshapeNodePairs()
1189 (*tpu_input_shapes)[out_edge].push_back(dimension); in InsertReshapeNodePairs()
2010 for (const Edge* out_edge : node->out_edges()) { in InferShapesWithResourceVar() local
2011 Node* read_node = out_edge->dst(); in InferShapesWithResourceVar()
2026 [graph = shape_inference_graph_interim.get(), out_edge, read_node] { in InferShapesWithResourceVar()
2027 graph->RemoveEdge(out_edge); in InferShapesWithResourceVar()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.h | 186 bool operator()(const Edge* out_edge) const;
|
| D | convert_nodes.cc | 6160 bool OutputEdgeValidator::operator()(const Edge* out_edge) const { in operator ()()
6161 if (out_edge->IsControlEdge()) return true; in operator ()()
6162 if (out_edge->src()->type_string() == "Const") { in operator ()()
6163 VLOG(1) << "--> Need to remove output node " << out_edge->src()->name() in operator ()()
|