| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_while.cc | 451 const Edge* in_edge; in FunctionalizeLoop() local
452 TF_RETURN_IF_ERROR(arg.enter->input_edge(0, &in_edge)); in FunctionalizeLoop()
453 if (in_edge->IsControlEdge()) { in FunctionalizeLoop()
454 builder.ControlInput(in_edge->src()->name()); in FunctionalizeLoop()
457 in_edge->src()->name(), in_edge->src_output(), arg_types[i])); in FunctionalizeLoop()
467 const Edge* in_edge; in FunctionalizeLoop() local
468 TF_RETURN_IF_ERROR(arg.enter->input_edge(0, &in_edge)); in FunctionalizeLoop()
469 if (in_edge->IsControlEdge()) { in FunctionalizeLoop()
470 graph->AddControlEdge(in_edge->src(), while_node); in FunctionalizeLoop()
472 graph->AddEdge(in_edge->src(), in_edge->src_output(), while_node, i); in FunctionalizeLoop()
|
| D | tf2xla_util.cc | 465 for (const string& in_edge : map_entry.second->input()) { in PruneGraphDefInto() local
466 auto id = ParseTensorName(in_edge); in PruneGraphDefInto()
727 for (const Edge* in_edge : n->in_edges()) { in ReplaceNode() local
728 g->AddEdge(in_edge->src(), in_edge->src_output(), new_node, in ReplaceNode()
729 in_edge->dst_input()); in ReplaceNode()
|
| D | functionalize_cond.cc | 773 const Edge* in_edge; in BuildIfNode() local
774 TF_RETURN_IF_ERROR(arg->input_edge(0, &in_edge)); in BuildIfNode()
775 if (in_edge->IsControlEdge()) { in BuildIfNode()
776 builder.ControlInput(in_edge->src()->name()); in BuildIfNode()
781 in_edge->src()->name(), in_edge->src_output(), dtype)); in BuildIfNode()
|
| /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 | 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 | 1260 for (const Edge* in_edge : n->in_edges()) { in CheckFetch() local
1261 const Node* in_node = in_edge->src(); in CheckFetch()
1262 if (pending_feeds.count({in_node->name(), in_edge->src_output()}) > 0) { in CheckFetch()
1264 in_edge->src_output(), in CheckFetch()
|
| /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 | 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/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 | 372 for (const Edge* in_edge : node->in_edges()) { in OptimizeControlFlowColocation() local
373 if (in_edge->dst_input() == 0) { in OptimizeControlFlowColocation()
376 in_edge->src()->assigned_device_name()); in OptimizeControlFlowColocation()
381 for (const Edge* in_edge : node->in_edges()) { in OptimizeControlFlowColocation() local
382 if (!in_edge->IsControlEdge()) { in OptimizeControlFlowColocation()
385 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 | 1074 for (const Edge* in_edge : n->in_edges()) { in CheckFetches() local
1075 const Node* in_node = in_edge->src(); in CheckFetches()
1076 if (pending_feeds.count({in_node->name(), in_edge->src_output()}) > 0) { in CheckFetches()
1078 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 | 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()
|