diff options
Diffstat (limited to 'src/armnn/layers/Convolution2dLayer.hpp')
-rw-r--r-- | src/armnn/layers/Convolution2dLayer.hpp | 17 |
1 files changed, 7 insertions, 10 deletions
diff --git a/src/armnn/layers/Convolution2dLayer.hpp b/src/armnn/layers/Convolution2dLayer.hpp index 02ae05f83b..f7e4dec72f 100644 --- a/src/armnn/layers/Convolution2dLayer.hpp +++ b/src/armnn/layers/Convolution2dLayer.hpp @@ -1,5 +1,5 @@ // -// Copyright © 2017 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2017,2022 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #pragma once @@ -16,13 +16,6 @@ class Convolution2dLayer : public LayerWithParameters<Convolution2dDescriptor> { public: - /// A unique pointer to store Weight values. - /// @Note: Deprecated. Removal date is 23.02. Weights are stored in ConstantLayers now. - std::shared_ptr<ConstTensorHandle> m_Weight; - /// A unique pointer to store Bias values. - /// @Note: Deprecated. Removal date is 23.02. Bias are stored in ConstantLayers now. - std::shared_ptr<ConstTensorHandle> m_Bias; - /// Makes a workload for the Convolution2d type. /// @param [in] graph The graph where this layer can be found. /// @param [in] factory The workload factory which will create the workload. @@ -48,6 +41,10 @@ public: void SerializeLayerParameters(ParameterStringifyFunction& fn) const override; + /// This layer does not have any data stored, weights and bias are now stored in constant layers. + /// We do not want to release the data in the constant layer, that is why we override with an empty function. + void ReleaseConstantData() override {} + protected: /// Constructor to create a Convolution2dLayer. /// @param [in] param Convolution2dDescriptor to configure the convolution2d operation. @@ -57,8 +54,8 @@ protected: /// Default destructor ~Convolution2dLayer() = default; - /// @Note Deprecated. GetConstantTensorsByRef is deprecated. m_Weights and m_Bias - /// should be connected to layer as Constant Layers instead." + /// Retrieve the handles to the constant values connected to the layer. + /// @return A vector of the constant tensors connected to the layer. ConstantTensors GetConstantTensorsByRef() override; }; |