Home
last modified time | relevance | path

Searched refs:ArgFeedRewrite (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/graph/
Dsubgraph.h120 class ArgFeedRewrite : public PruneRewrite {
122 ArgFeedRewrite(const string* endpoint_name, in ArgFeedRewrite() function
Dsubgraph.cc220 Status ArgFeedRewrite::AddNode(Graph* g, NodeBuilder::NodeOut feed_tensor, in AddNode()
304 feed_rewrites.emplace_back(new ArgFeedRewrite( in RewriteGraphForExecution()
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_execution_state.cc459 new subgraph::ArgFeedRewrite(&feed, device_info, i)); in PruneGraph()