Searched refs:in_edge (Results 1 - 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/contrib/tensorrt/segment/
H A Dsegment.cc82 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 Dfunctionalize_control_flow.cc502 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 Dtf2xla_util.cc226 for (const string& in_edge : map_entry.second->input()) {
227 auto id = ParseTensorName(in_edge);
/external/tensorflow/tensorflow/contrib/nccl/kernels/
H A Dnccl_rewrite.cc142 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 Dconstant_folding.cc357 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 Ddirect_session.cc1047 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 Dscheduler.cc148 for (const Edge* in_edge : curr->in_edges()) {
151 Node* src = in_edge->src();
152 if (!in_edge->IsControlEdge() &&
H A Dmaster_session.cc874 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 Dencapsulate_subgraphs_pass.cc1668 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 Dgraph_partition.cc331 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 Dremote_fused_graph_execute_utils.cc772 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