diff options
Diffstat (limited to 'src/armnn/layers/ConstantLayer.hpp')
| -rw-r--r-- | src/armnn/layers/ConstantLayer.hpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/armnn/layers/ConstantLayer.hpp b/src/armnn/layers/ConstantLayer.hpp index ff4c03775f..ead8816684 100644 --- a/src/armnn/layers/ConstantLayer.hpp +++ b/src/armnn/layers/ConstantLayer.hpp @@ -9,7 +9,7 @@ namespace armnn { -class ScopedCpuTensorHandle; +class ScopedTensorHandle; /// A layer that the constant data can be bound to. class ConstantLayer : public Layer @@ -43,7 +43,7 @@ public: void ExecuteStrategy(IStrategy& strategy) const override; - std::shared_ptr<ConstCpuTensorHandle> m_LayerOutput; + std::shared_ptr<ConstTensorHandle> m_LayerOutput; protected: /// Constructor to create a ConstantLayer. |