Searched refs:in_edge (Results 1 - 11 of 11) sorted by relevance
/external/tensorflow/tensorflow/contrib/tensorrt/segment/ |
H A D | segment.cc | 82 for (const tensorflow::Edge* in_edge : in_edges) { 83 if (in_edge->src() != src) { 84 tensorflow::Edge* e = const_cast<tensorflow::Edge*>(in_edge); 109 for (const auto& in_edge : dst->in_edges()) { 110 remove_edges->push_back(in_edge);
|
/external/tensorflow/tensorflow/compiler/tf2xla/ |
H A D | functionalize_control_flow.cc | 502 const Edge* in_edge; local 503 TF_RETURN_IF_ERROR(arg.enter->input_edge(0, &in_edge)); 504 if (in_edge->IsControlEdge()) { 505 builder.ControlInput(in_edge->src()->name()); 508 in_edge->src()->name(), in_edge->src_output(), arg_types[i])); 518 const Edge* in_edge; local 519 TF_RETURN_IF_ERROR(arg.enter->input_edge(0, &in_edge)); 520 if (in_edge->IsControlEdge()) { 521 graph->AddControlEdge(in_edge 1177 const Edge* in_edge; local 1244 const Edge* in_edge; local 1274 const Edge* in_edge; local [all...] |
H A D | tf2xla_util.cc | 226 for (const string& in_edge : map_entry.second->input()) { 227 auto id = ParseTensorName(in_edge);
|
/external/tensorflow/tensorflow/contrib/nccl/kernels/ |
H A D | nccl_rewrite.cc | 142 const Edge* in_edge = nullptr; local 143 TF_RETURN_IF_ERROR(node->input_edge(0, &in_edge)); 144 Node* in_node = in_edge->src(); 145 int in_index = in_edge->src_output();
|
/external/tensorflow/tensorflow/core/common_runtime/ |
H A D | constant_folding.cc | 357 for (const Edge* in_edge : n->in_edges()) { 359 if (!in_edge->IsControlEdge()) { 360 Node* in = in_edge->src(); 365 constant_graph->AddEdge(it->second[0], in_edge->src_output(), added[0], 366 in_edge->dst_input()); 372 constant_graph->AddEdge(it->second[in_edge->src_output()], 0, added[0], 373 in_edge->dst_input());
|
H A D | direct_session.cc | 1047 for (const Edge* in_edge : n->in_edges()) { 1048 const Node* in_node = in_edge->src(); 1049 if (pending_feeds.count({in_node->name(), in_edge->src_output()}) > 0) { 1051 in_edge->src_output(),
|
/external/tensorflow/tensorflow/core/distributed_runtime/ |
H A D | scheduler.cc | 148 for (const Edge* in_edge : curr->in_edges()) { 151 Node* src = in_edge->src(); 152 if (!in_edge->IsControlEdge() &&
|
H A D | master_session.cc | 874 for (const Edge* in_edge : n->in_edges()) { 875 const Node* in_node = in_edge->src(); 876 if (pending_feeds.count({in_node->name(), in_edge->src_output()}) > 0) { 878 in_edge->src_output(),
|
/external/tensorflow/tensorflow/compiler/jit/ |
H A D | encapsulate_subgraphs_pass.cc | 1668 for (const Edge* in_edge : node_in->in_edges()) { 1669 if (!in_edge->IsControlEdge()) { 1670 Node* src = in_edge->src(); 1674 graph_out->AddEdge((*copied_node_images)[in_edge->src()], 1675 in_edge->src_output(), node_out, 1676 in_edge->dst_input()); 1679 graph_out->AddEdge(iter->second, 0, node_out, in_edge->dst_input()); 1740 for (const Edge* in_edge : n->in_edges()) { 1741 if (!in_edge->IsControlEdge()) { 1742 Node* src_node = in_edge [all...] |
/external/tensorflow/tensorflow/core/graph/ |
H A D | graph_partition.cc | 331 for (const Edge* in_edge : node->in_edges()) { 332 if (in_edge->dst_input() == 0) { 335 in_edge->src()->assigned_device_name()); 340 for (const Edge* in_edge : node->in_edges()) { 341 if (!in_edge->IsControlEdge()) { 344 in_edge->src()->assigned_device_name());
|
/external/tensorflow/tensorflow/core/kernels/ |
H A D | remote_fused_graph_execute_utils.cc | 772 for (const Edge* in_edge : node->in_edges()) { 773 const Node* src_node = in_edge->src(); 778 strings::StrCat(src_node->name(), ":", in_edge->src_output());
|
Completed in 655 milliseconds