diff options
Diffstat (limited to 'src/armnn/layers/Convolution2dLayer.hpp')
-rw-r--r-- | src/armnn/layers/Convolution2dLayer.hpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/armnn/layers/Convolution2dLayer.hpp b/src/armnn/layers/Convolution2dLayer.hpp index 05a26da82a..0e85b33355 100644 --- a/src/armnn/layers/Convolution2dLayer.hpp +++ b/src/armnn/layers/Convolution2dLayer.hpp @@ -15,6 +15,7 @@ class ScopedCpuTensorHandle; class Convolution2dLayer : public LayerWithParameters<Convolution2dDescriptor> { public: + /// A unique pointer to store Weight values. std::unique_ptr<ScopedCpuTensorHandle> m_Weight; /// A unique pointer to store Bias values. @@ -43,6 +44,8 @@ public: void Accept(ILayerVisitor& visitor) const override; + void SerializeLayerParameters(ParameterStringifyFunction& fn) const override; + protected: /// Constructor to create a Convolution2dLayer. /// @param [in] param Convolution2dDescriptor to configure the convolution2d operation. |