Searched refs:feed_node (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/graph/ |
| D | subgraph.cc | 77 Node* feed_node; in FeedInputs() local
79 feed_rewrites[i]->AddNode(g, {n, id.second}, &feed_node)); in FeedInputs()
82 (*name_index)[feed_node->name()] = feed_node; in FeedInputs()
85 g->AddControlEdge(g->source_node(), feed_node, true); in FeedInputs()
108 g->AddEdge(feed_node, 0, e->dst(), e->dst_input()); in FeedInputs()
113 g->AddControlEdge(feed_node, e->dst(), true); in FeedInputs()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | graph_compiler_util.cc | 74 const Node* feed_node = node_it->second; in AddArgNodes() local
85 .Attr("T", BaseType(feed_node->output_type(output_index))) in AddArgNodes()
99 for (const Edge* edge : feed_node->out_edges()) { in AddArgNodes()
|
| D | tf2xla_util.cc | 560 TF_ASSIGN_OR_RETURN(Node * feed_node, g.AddNode(gd.node(0))); in AddPlaceholdersForFeeds()
562 if (info.feed->id().output_index() < feed_node->num_outputs()) { in AddPlaceholdersForFeeds()
564 BaseType(feed_node->output_type(info.feed->id().output_index())); in AddPlaceholdersForFeeds()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item_builder.cc | 302 for (const auto& feed_node : cfg.feed_nodes) { in GrapplerItemFromMetaGraphDef() local
303 const string feed_name = NodeName(feed_node); in GrapplerItemFromMetaGraphDef()
|