diff options
Diffstat (limited to 'src/armnn/layers/ConstantLayer.hpp')
-rw-r--r-- | src/armnn/layers/ConstantLayer.hpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/armnn/layers/ConstantLayer.hpp b/src/armnn/layers/ConstantLayer.hpp index 9d91551df9..ff4c03775f 100644 --- a/src/armnn/layers/ConstantLayer.hpp +++ b/src/armnn/layers/ConstantLayer.hpp @@ -43,7 +43,7 @@ public: void ExecuteStrategy(IStrategy& strategy) const override; - std::unique_ptr<ScopedCpuTensorHandle> m_LayerOutput; + std::shared_ptr<ConstCpuTensorHandle> m_LayerOutput; protected: /// Constructor to create a ConstantLayer. |