Searched refs:RetvalFetchRewrite (Results 1 – 3 of 3) sorted by relevance
141 class RetvalFetchRewrite : public PruneRewrite {143 RetvalFetchRewrite(const string* endpoint_name, in RetvalFetchRewrite() function
257 Status RetvalFetchRewrite::AddNode(Graph* g, NodeBuilder::NodeOut fetch_tensor, in AddNode()318 fetch_rewrites.emplace_back(new RetvalFetchRewrite( in RewriteGraphForExecution()
479 new subgraph::RetvalFetchRewrite(&fetch, device_info, i)); in PruneGraph()