aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/layers/ConstantLayer.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/layers/ConstantLayer.hpp')
-rw-r--r--src/armnn/layers/ConstantLayer.hpp2
1 files changed, 2 insertions, 0 deletions
diff --git a/src/armnn/layers/ConstantLayer.hpp b/src/armnn/layers/ConstantLayer.hpp
index c3316b78b1..d045952e7e 100644
--- a/src/armnn/layers/ConstantLayer.hpp
+++ b/src/armnn/layers/ConstantLayer.hpp
@@ -39,6 +39,8 @@ public:
/// Free up the constant source data stored by the layer.
void ReleaseConstantData() override {};
+ void Accept(ILayerVisitor& visitor) const override;
+
std::unique_ptr<ScopedCpuTensorHandle> m_LayerOutput;
protected:
/// Constructor to create a ConstantLayer.