| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_while.cc | 534 const Edge* in_edge; in FunctionalizeLoop() local
535 TF_RETURN_IF_ERROR(arg.enter->input_edge(0, &in_edge)); in FunctionalizeLoop()
536 if (in_edge->IsControlEdge()) { in FunctionalizeLoop()
537 builder.ControlInput(in_edge->src()->name()); in FunctionalizeLoop()
540 in_edge->src()->name(), in_edge->src_output(), arg_types[i])); in FunctionalizeLoop()
550 const Edge* in_edge; in FunctionalizeLoop() local
551 TF_RETURN_IF_ERROR(arg.enter->input_edge(0, &in_edge)); in FunctionalizeLoop()
552 if (in_edge->IsControlEdge()) { in FunctionalizeLoop()
553 graph->AddControlEdge(in_edge->src(), while_node); in FunctionalizeLoop()
555 graph->AddEdge(in_edge->src(), in_edge->src_output(), while_node, i); in FunctionalizeLoop()
|
| D | tf2xla_util.cc | 470 for (const string& in_edge : map_entry.second->input()) { in PruneGraphDefInto() local
471 auto id = ParseTensorName(in_edge); in PruneGraphDefInto()
734 for (const Edge* in_edge : n->in_edges()) { in ReplaceNode() local
735 g->AddEdge(in_edge->src(), in_edge->src_output(), new_node, in ReplaceNode()
736 in_edge->dst_input()); in ReplaceNode()
|
| D | functionalize_cond.cc | 755 const Edge* in_edge; in BuildIfNode() local
756 TF_RETURN_IF_ERROR(arg->input_edge(0, &in_edge)); in BuildIfNode()
757 if (in_edge->IsControlEdge()) { in BuildIfNode()
758 builder.ControlInput(in_edge->src()->name()); in BuildIfNode()
763 in_edge->src()->name(), in_edge->src_output(), dtype)); in BuildIfNode()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| D | segment.cc | 357 for (const SimpleEdge* in_edge : in_edges) { in ContractEdge() local
358 if (in_edge->IsControlEdge()) { in ContractEdge()
359 if (in_edge->src() != src) { in ContractEdge()
360 SimpleEdge* e = const_cast<SimpleEdge*>(in_edge); in ContractEdge()
364 if (in_edge->src() != src) { in ContractEdge()
365 SimpleEdge* e = const_cast<SimpleEdge*>(in_edge); in ContractEdge()
396 for (const auto& in_edge : dst->in_edges()) { in ContractEdge() local
397 remove_edges->push_back(in_edge); in ContractEdge()
|
| D | segment_test.cc | 51 return [node_names](const Edge* in_edge) -> bool { in MakeInputEdgeCandidateFn() argument
52 return node_names.find(in_edge->dst()->name()) != node_names.end(); in MakeInputEdgeCandidateFn()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | partially_decluster_pass.cc | 134 for (const Edge* in_edge : n->in_edges()) { in PartiallyDeclusterNode() local
135 graph->AddEdge(in_edge->src(), in_edge->src_output(), cloned_node, in PartiallyDeclusterNode()
136 in_edge->dst_input()); in PartiallyDeclusterNode()
|
| D | encapsulate_subgraphs_pass.cc | 1891 for (const Edge* in_edge : node_in->in_edges()) { in CopyShapeInferenceNodeToGraph() local
1892 if (!in_edge->IsControlEdge()) { in CopyShapeInferenceNodeToGraph()
1893 Node* src = in_edge->src(); in CopyShapeInferenceNodeToGraph()
1897 graph_out->AddEdge((*copied_node_images)[in_edge->src()], in CopyShapeInferenceNodeToGraph()
1898 in_edge->src_output(), node_out, in CopyShapeInferenceNodeToGraph()
1899 in_edge->dst_input()); in CopyShapeInferenceNodeToGraph()
1902 graph_out->AddEdge(iter->second, 0, node_out, in_edge->dst_input()); in CopyShapeInferenceNodeToGraph()
1977 for (const Edge* in_edge : n->in_edges()) { in DoStaticShapeInferenceForOutsideCompilationSend() local
1978 if (!in_edge->IsControlEdge()) { in DoStaticShapeInferenceForOutsideCompilationSend()
1979 Node* src_node = in_edge->src(); in DoStaticShapeInferenceForOutsideCompilationSend()
[all …]
|
| D | deadness_analysis.cc | 761 for (const Edge* in_edge : n->in_edges()) { in GetInputPreds() local
764 (in_edge->IsControlEdge() && edge_kind == EdgeKind::kControlOnly) || in GetInputPreds()
765 (!in_edge->IsControlEdge() && edge_kind == EdgeKind::kDataOnly); in GetInputPreds()
768 auto it = predicate_map_.find(InputEdgeToTensorId(in_edge)); in GetInputPreds()
776 return errors::Internal("Could not find input ", in_edge->DebugString(), in GetInputPreds()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | constant_folding.cc | 363 for (const Edge* in_edge : n->in_edges()) { in AddNodeToConstantGraph() local
365 if (!in_edge->IsControlEdge()) { in AddNodeToConstantGraph()
366 Node* in = in_edge->src(); in AddNodeToConstantGraph()
371 constant_graph->AddEdge(it->second[0], in_edge->src_output(), added[0], in AddNodeToConstantGraph()
372 in_edge->dst_input()); in AddNodeToConstantGraph()
378 constant_graph->AddEdge(it->second[in_edge->src_output()], 0, added[0], in AddNodeToConstantGraph()
379 in_edge->dst_input()); in AddNodeToConstantGraph()
|
| D | direct_session.cc | 1168 for (const Edge* in_edge : n->in_edges()) { in CheckFetch() local
1169 const Node* in_node = in_edge->src(); in CheckFetch()
1170 if (pending_feeds.count({in_node->name(), in_edge->src_output()}) > 0) { in CheckFetch()
1172 in_edge->src_output(), in CheckFetch()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | algorithm.cc | 163 for (const Edge* in_edge : n->in_edges()) { in ReverseDFSFromHelper() local
164 nodes_sorted.emplace_back(in_edge->src()); in ReverseDFSFromHelper()
171 for (const Edge* in_edge : n->in_edges()) { in ReverseDFSFromHelper() local
172 add_work(in_edge->src()); in ReverseDFSFromHelper()
|
| D | graph_partition.cc | 347 for (const Edge* in_edge : node->in_edges()) { in OptimizeControlFlowColocation() local
348 if (in_edge->dst_input() == 0) { in OptimizeControlFlowColocation()
351 in_edge->src()->assigned_device_name()); in OptimizeControlFlowColocation()
356 for (const Edge* in_edge : node->in_edges()) { in OptimizeControlFlowColocation() local
357 if (!in_edge->IsControlEdge()) { in OptimizeControlFlowColocation()
360 in_edge->src()->assigned_device_name()); in OptimizeControlFlowColocation()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_rewrite.cc | 144 const Edge* in_edge = nullptr; in ReplaceBroadcast() local
145 TF_RETURN_IF_ERROR(node->input_edge(0, &in_edge)); in ReplaceBroadcast()
146 Node* in_node = in_edge->src(); in ReplaceBroadcast()
147 int in_index = in_edge->src_output(); in ReplaceBroadcast()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | scheduler.cc | 148 for (const Edge* in_edge : curr->in_edges()) { in ComputeAlap() local
151 const Node* src = in_edge->src(); in ComputeAlap()
152 if (!in_edge->IsControlEdge() && in ComputeAlap()
|
| D | master_session.cc | 1019 for (const Edge* in_edge : n->in_edges()) { in CheckFetches() local
1020 const Node* in_node = in_edge->src(); in CheckFetches()
1021 if (pending_feeds.count({in_node->name(), in_edge->src_output()}) > 0) { in CheckFetches()
1023 in_edge->src_output(), in CheckFetches()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | vectorization_utils.cc | 608 const Edge* in_edge = nullptr; in AddUnstackedTensorMappings() local
609 TF_RETURN_IF_ERROR(ret_node->input_edge(0, &in_edge)); in AddUnstackedTensorMappings()
610 AddUnstackedTensorMappingsHelper({in_edge->src(), in_edge->src_output()}, in AddUnstackedTensorMappings()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_execute_utils.cc | 773 for (const Edge* in_edge : node->in_edges()) { in ClusterizeNodes() local
774 const Node* src_node = in_edge->src(); in ClusterizeNodes()
779 strings::StrCat(src_node->name(), ":", in_edge->src_output()); in ClusterizeNodes()
|