diff options
| author | Mike Kelly <mike.kelly@arm.com> | 2022-11-25 13:55:24 +0000 |
|---|---|---|
| committer | mike.kelly <mike.kelly@arm.com> | 2022-12-12 15:58:21 +0000 |
| commit | ec67a0f08e0f96a5aebf3cac65331c67f6649f5e (patch) | |
| tree | 94146a1f43c74d89d83fd5da54688ae0fc19cf85 /src/backends/neon/workloads/NeonFullyConnectedWorkload.hpp | |
| parent | 5383767a7a759c867235ab66bd71f88281e3bd06 (diff) | |
| download | armnn-ec67a0f08e0f96a5aebf3cac65331c67f6649f5e.tar.gz | |
IVGCVSW-7209 Remove deprecated code due to be removed in 23.02
* Removed weights and bias from Convolution, DepthwiseConv & FullyConnected
layers
* Removed the weight and bias ConstTensorHandles from the QueueDescriptors
* Updated Workloads to take tensors from WorkloadInfo rather than the
QueueDescriptors
* Removed unused RedirectMembersToConstantInputs optimization and tests.
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Change-Id: I9ffcdc4a1c0dff725539dd69fc435b700bd98a56
Diffstat (limited to 'src/backends/neon/workloads/NeonFullyConnectedWorkload.hpp')
| -rw-r--r-- | src/backends/neon/workloads/NeonFullyConnectedWorkload.hpp | 11 |
1 files changed, 7 insertions, 4 deletions
diff --git a/src/backends/neon/workloads/NeonFullyConnectedWorkload.hpp b/src/backends/neon/workloads/NeonFullyConnectedWorkload.hpp index 944731d7bd..d5bb932c6e 100644 --- a/src/backends/neon/workloads/NeonFullyConnectedWorkload.hpp +++ b/src/backends/neon/workloads/NeonFullyConnectedWorkload.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 // @@ -29,13 +29,16 @@ class NeonFullyConnectedWorkload : public NeonBaseWorkload<FullyConnectedQueueDe public: NeonFullyConnectedWorkload(const FullyConnectedQueueDescriptor& descriptor, const WorkloadInfo& info, std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager); + virtual void Execute() const override; private: std::unique_ptr<arm_compute::IFunction> m_FullyConnectedLayer; - std::unique_ptr<arm_compute::Tensor> m_WeightsTensor; - std::unique_ptr<arm_compute::Tensor> m_BiasesTensor; - + mutable std::unique_ptr<arm_compute::Tensor> m_WeightsTensor; + mutable std::unique_ptr<arm_compute::Tensor> m_BiasesTensor; + TensorInfo m_WeightsTensorInfo; + TensorInfo m_BiasesTensorInfo; + mutable bool prepared = false; }; } //namespace armnn |