diff options
Diffstat (limited to 'src/armnn/layers/FullyConnectedLayer.hpp')
| -rw-r--r-- | src/armnn/layers/FullyConnectedLayer.hpp | 12 |
1 files changed, 5 insertions, 7 deletions
diff --git a/src/armnn/layers/FullyConnectedLayer.hpp b/src/armnn/layers/FullyConnectedLayer.hpp index 07f4a936f9..f3ca696b62 100644 --- a/src/armnn/layers/FullyConnectedLayer.hpp +++ b/src/armnn/layers/FullyConnectedLayer.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 @@ -15,12 +15,6 @@ class ScopedTensorHandle; class FullyConnectedLayer : public LayerWithParameters<FullyConnectedDescriptor> { 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 FullyConnected type. /// @param [in] graph The graph where this layer can be found. @@ -45,6 +39,10 @@ public: void ExecuteStrategy(IStrategy& strategy) 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 FullyConnectedLayer. /// @param [in] param FullyConnectedDescriptor to configure the fully connected operation. |