Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/graph/
Dsubgraph.h50 class PruneRewrite {
53 PruneRewrite(const string* endpoint_name, const DeviceAttributes* device_info) in PruneRewrite() function
55 virtual ~PruneRewrite() {} in ~PruneRewrite()
110 Graph* g, const std::vector<std::unique_ptr<PruneRewrite>>& feed_rewrites,
111 const std::vector<std::unique_ptr<PruneRewrite>>& fetch_rewrites,
120 class ArgFeedRewrite : public PruneRewrite {
124 : PruneRewrite(endpoint_name, device_info), arg_index_(arg_index) {} in ArgFeedRewrite()
133 class RecvFeedRewrite : public PruneRewrite {
135 using PruneRewrite::PruneRewrite;
141 class RetvalFetchRewrite : public PruneRewrite {
[all …]
Dsubgraph.cc59 Graph* g, const std::vector<std::unique_ptr<PruneRewrite>>& feed_rewrites, in FeedInputs()
124 Graph* g, const std::vector<std::unique_ptr<PruneRewrite>>& fetch_rewrites, in FetchOutputs()
300 std::vector<std::unique_ptr<PruneRewrite>> feed_rewrites; in RewriteGraphForExecution()
314 std::vector<std::unique_ptr<PruneRewrite>> fetch_rewrites; in RewriteGraphForExecution()
340 Graph* g, const std::vector<std::unique_ptr<PruneRewrite>>& feed_rewrites, in RewriteGraphForExecution()
341 const std::vector<std::unique_ptr<PruneRewrite>>& fetch_rewrites, in RewriteGraphForExecution()
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_execution_state.cc251 class TensorConnectionPruneRewrite : public subgraph::PruneRewrite {
255 : subgraph::PruneRewrite(endpoint_name, nullptr /* device_info */), in TensorConnectionPruneRewrite()
444 std::vector<std::unique_ptr<subgraph::PruneRewrite>> feed_rewrites; in PruneGraph()
446 std::vector<std::unique_ptr<subgraph::PruneRewrite>> fetch_rewrites; in PruneGraph()