| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_while.cc | 452 const Edge* in_edge; in FunctionalizeLoop() local
453 TF_RETURN_IF_ERROR(arg.enter->input_edge(0, &in_edge)); in FunctionalizeLoop()
454 if (in_edge->IsControlEdge()) { in FunctionalizeLoop()
455 builder.ControlInput(in_edge->src()->name()); in FunctionalizeLoop()
458 in_edge->src()->name(), in_edge->src_output(), arg_types[i])); in FunctionalizeLoop()
468 const Edge* in_edge; in FunctionalizeLoop() local
469 TF_RETURN_IF_ERROR(arg.enter->input_edge(0, &in_edge)); in FunctionalizeLoop()
470 if (in_edge->IsControlEdge()) { in FunctionalizeLoop()
471 graph->AddControlEdge(in_edge->src(), while_node); in FunctionalizeLoop()
473 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()
733 for (const Edge* in_edge : n->in_edges()) { in ReplaceNode() local
734 g->AddEdge(in_edge->src(), in_edge->src_output(), new_node, in ReplaceNode()
735 in_edge->dst_input()); in ReplaceNode()
|
| D | functionalize_cond.cc | 781 const Edge* in_edge; in BuildIfNode() local
782 TF_RETURN_IF_ERROR(arg->input_edge(0, &in_edge)); in BuildIfNode()
783 if (in_edge->IsControlEdge()) { in BuildIfNode()
784 builder.ControlInput(in_edge->src()->name()); in BuildIfNode()
789 in_edge->src()->name(), in_edge->src_output(), dtype)); in BuildIfNode()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | algorithm.cc | 166 for (const Edge* in_edge : n->in_edges()) { in ReverseDFSFromHelper() local
167 if (!edge_filter || edge_filter(*in_edge)) { in ReverseDFSFromHelper()
168 nodes_sorted.emplace_back(in_edge->src()); in ReverseDFSFromHelper()
176 for (const Edge* in_edge : n->in_edges()) { in ReverseDFSFromHelper() local
177 if (!edge_filter || edge_filter(*in_edge)) { in ReverseDFSFromHelper()
178 add_work(in_edge->src()); in ReverseDFSFromHelper()
|
| D | graph_partition.cc | 374 for (const Edge* in_edge : node->in_edges()) { in OptimizeControlFlowColocation() local
375 if (in_edge->dst_input() == 0) { in OptimizeControlFlowColocation()
378 in_edge->src()->assigned_device_name()); in OptimizeControlFlowColocation()
383 for (const Edge* in_edge : node->in_edges()) { in OptimizeControlFlowColocation() local
384 if (!in_edge->IsControlEdge()) { in OptimizeControlFlowColocation()
387 in_edge->src()->assigned_device_name()); in OptimizeControlFlowColocation()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | accumulate_n_optimizer.cc | 247 for (const Edge* in_edge : n->in_edges()) { in RewriteIntoAddN() local
248 if (in_edge->IsControlEdge()) { in RewriteIntoAddN()
249 control_inputs.push_back(in_edge->src()); in RewriteIntoAddN()
251 data_inputs.emplace_back(in_edge->src(), in_edge->src_output()); in RewriteIntoAddN()
|
| D | constant_folding.cc | 390 for (const Edge* in_edge : n->in_edges()) { in AddNodeToConstantGraph() local
392 if (!in_edge->IsControlEdge()) { in AddNodeToConstantGraph()
393 Node* in = in_edge->src(); in AddNodeToConstantGraph()
398 constant_graph->AddEdge(it->second[0], in_edge->src_output(), added[0], in AddNodeToConstantGraph()
399 in_edge->dst_input()); in AddNodeToConstantGraph()
405 constant_graph->AddEdge(it->second[in_edge->src_output()], 0, added[0], in AddNodeToConstantGraph()
406 in_edge->dst_input()); in AddNodeToConstantGraph()
|
| D | direct_session.cc | 1274 for (const Edge* in_edge : n->in_edges()) { in CheckFetch() local
1275 const Node* in_node = in_edge->src(); in CheckFetch()
1276 if (pending_feeds.count({in_node->name(), in_edge->src_output()}) > 0) { in CheckFetch()
1278 in_edge->src_output(), in CheckFetch()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | partially_decluster_pass.cc | 146 for (const Edge* in_edge : n->in_edges()) { in PartiallyDeclusterNode() local
147 graph->AddEdge(in_edge->src(), in_edge->src_output(), cloned_node, in PartiallyDeclusterNode()
148 in_edge->dst_input()); in PartiallyDeclusterNode()
|
| D | deadness_analysis.cc | 905 for (const Edge* in_edge : n->in_edges()) { in GetInputPreds() local
908 (in_edge->IsControlEdge() && edge_kind == EdgeKind::kControlOnly) || in GetInputPreds()
909 (!in_edge->IsControlEdge() && edge_kind == EdgeKind::kDataOnly); in GetInputPreds()
912 auto it = predicate_map_.find(InputEdgeToTensorId(in_edge)); in GetInputPreds()
920 return errors::Internal("Could not find input ", in_edge->DebugString(), in GetInputPreds()
|
| /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/compiler/tf2tensorrt/segment/ |
| D | segment.cc | 585 for (const SimpleEdge* in_edge : in_edges) { in ContractEdge() local
586 if (in_edge->IsControlEdge()) { in ContractEdge()
587 if (in_edge->src() != src) { in ContractEdge()
588 SimpleEdge* e = const_cast<SimpleEdge*>(in_edge); in ContractEdge()
592 if (in_edge->src() != src) { in ContractEdge()
593 SimpleEdge* e = const_cast<SimpleEdge*>(in_edge); in ContractEdge()
624 for (const auto& in_edge : dst->in_edges()) { in ContractEdge() local
625 remove_edges->push_back(in_edge); in ContractEdge()
|
| D | segment_test.cc | 50 return [node_names](const Edge* in_edge) -> bool { in MakeInputEdgeCandidateFn() argument
51 return node_names.find(in_edge->dst()->name()) != node_names.end(); in MakeInputEdgeCandidateFn()
|
| /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 | 1076 for (const Edge* in_edge : n->in_edges()) { in CheckFetches() local
1077 const Node* in_node = in_edge->src(); in CheckFetches()
1078 if (pending_feeds.count({in_node->name(), in_edge->src_output()}) > 0) { in CheckFetches()
1080 in_edge->src_output(), in CheckFetches()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | vectorization_utils.cc | 601 const Edge* in_edge = nullptr; in AddUnstackedTensorMappings() local
602 TF_RETURN_IF_ERROR(ret_node->input_edge(0, &in_edge)); in AddUnstackedTensorMappings()
603 AddUnstackedTensorMappingsHelper({in_edge->src(), in_edge->src_output()}, in AddUnstackedTensorMappings()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_execute_utils.cc | 774 for (const Edge* in_edge : node->in_edges()) { in ClusterizeNodes() local
775 const Node* src_node = in_edge->src(); in ClusterizeNodes()
780 strings::StrCat(src_node->name(), ":", in_edge->src_output()); in ClusterizeNodes()
|