1 files changed, 4 insertions, 4 deletions
diff --git a/src/armnn/OutputHandler.hpp b/src/armnn/OutputHandler.hpp
index 3fd2519ed5..d1cb2deea3 100644
--- a/src/armnn/OutputHandler.hpp
+++ b/src/armnn/OutputHandler.hpp
@@ -50,15 +50,15 @@ public:
 
     void SetData(std::unique_ptr<ITensorHandle> data) { m_TensorHandle = std::move(data); }
 
-    void SetAllocatedData() { m_AllocatedTensorHandle = std::move(m_TensorHandle); }
+    void SetAllocatedData();
 
-    void UseAllocatedData() { m_TensorHandle = std::move(m_AllocatedTensorHandle); }
+    void UseAllocatedData() { m_TensorHandle = m_AllocatedTensorHandle; }
 
     /// @brief Returns true if SetTensorInfo() has been called at least once on this.
     bool IsTensorInfoSet() const { return m_bTensorInfoSet; }
 private:
-    std::unique_ptr<ITensorHandle> m_TensorHandle;
-    std::unique_ptr<ITensorHandle> m_AllocatedTensorHandle;
+    std::shared_ptr<ITensorHandle> m_TensorHandle;
+    std::shared_ptr<ITensorHandle> m_AllocatedTensorHandle;
     TensorInfo m_TensorInfo;
     bool m_bTensorInfoSet = false;
 };