aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/RefTensorHandleFactory.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/reference/RefTensorHandleFactory.hpp')
-rw-r--r--src/backends/reference/RefTensorHandleFactory.hpp7
1 files changed, 7 insertions, 0 deletions
diff --git a/src/backends/reference/RefTensorHandleFactory.hpp b/src/backends/reference/RefTensorHandleFactory.hpp
index 8ea02f58c4..7e572d7d1b 100644
--- a/src/backends/reference/RefTensorHandleFactory.hpp
+++ b/src/backends/reference/RefTensorHandleFactory.hpp
@@ -33,6 +33,13 @@ public:
std::unique_ptr<ITensorHandle> CreateTensorHandle(const TensorInfo& tensorInfo,
DataLayout dataLayout) const override;
+ std::unique_ptr<ITensorHandle> CreateTensorHandle(const TensorInfo& tensorInfo,
+ const bool IsMemoryManaged) const override;
+
+ std::unique_ptr<ITensorHandle> CreateTensorHandle(const TensorInfo& tensorInfo,
+ DataLayout dataLayout,
+ const bool IsMemoryManaged) const override;
+
static const FactoryId& GetIdStatic();
const FactoryId& GetId() const override;