Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/graph/
Dsubgraph.cc124 Graph* g, const std::vector<std::unique_ptr<PruneRewrite>>& fetch_rewrites, in FetchOutputs() argument
128 out_fetch_nodes->reserve(fetch_rewrites.size()); in FetchOutputs()
129 for (size_t i = 0; i < fetch_rewrites.size(); ++i) { in FetchOutputs()
130 const string& t = fetch_rewrites[i]->endpoint_name(); in FetchOutputs()
162 fetch_rewrites[i]->AddNode(g, {n, id.second}, &fetch_node)); in FetchOutputs()
314 std::vector<std::unique_ptr<PruneRewrite>> fetch_rewrites; in RewriteGraphForExecution() local
315 fetch_rewrites.reserve(fetch_outputs.size()); in RewriteGraphForExecution()
318 fetch_rewrites.emplace_back(new RetvalFetchRewrite( in RewriteGraphForExecution()
323 fetch_rewrites.emplace_back( in RewriteGraphForExecution()
328 return RewriteGraphForExecution(g, feed_rewrites, fetch_rewrites, in RewriteGraphForExecution()
[all …]
Dsubgraph.h111 const std::vector<std::unique_ptr<PruneRewrite>>& fetch_rewrites,
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_execution_state.cc446 std::vector<std::unique_ptr<subgraph::PruneRewrite>> fetch_rewrites; in PruneGraph() local
447 fetch_rewrites.reserve(options.callable_options.fetch_size()); in PruneGraph()
478 fetch_rewrites.emplace_back( in PruneGraph()
499 fetch_rewrites.emplace_back( in PruneGraph()
535 graph, feed_rewrites, fetch_rewrites, target_node_names, in PruneGraph()