diff options
Diffstat (limited to 'compiler/luci-interpreter/src/loader/GraphLoader.h')
-rw-r--r-- | compiler/luci-interpreter/src/loader/GraphLoader.h | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/compiler/luci-interpreter/src/loader/GraphLoader.h b/compiler/luci-interpreter/src/loader/GraphLoader.h index 89c5bcad7..fe066ecf8 100644 --- a/compiler/luci-interpreter/src/loader/GraphLoader.h +++ b/compiler/luci-interpreter/src/loader/GraphLoader.h @@ -19,6 +19,7 @@ #include "core/RuntimeGraph.h" #include "loader/RuntimeToIR.h" +#include "luci_interpreter/MemoryManager.h" #include <loco/IR/Graph.h> @@ -32,7 +33,8 @@ class GraphLoader public: GraphLoader(const loco::Graph *graph, RuntimeGraph *runtime_graph, RuntimeToIR &runtime_to_ir, const std::unordered_map<const loco::Graph *, RuntimeGraph *> &graph_to_runtime_graph, - std::unordered_map<const loco::Node *, Tensor *> &node_to_tensor); + std::unordered_map<const loco::Node *, Tensor *> &node_to_tensor, + IMemoryManager *memory_manager); void loadTensors(); void initInputOutputTensors() const; @@ -42,6 +44,7 @@ private: const loco::Graph *_graph; RuntimeGraph *_runtime_graph; RuntimeToIR &_runtime_to_ir; + IMemoryManager *_memory_manager; const std::unordered_map<const loco::Graph *, RuntimeGraph *> &_graph_to_runtime_graph; std::unordered_map<const loco::Node *, Tensor *> &_node_to_tensor; |