diff options
| author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2021-04-13 17:44:15 +0100 |
|---|---|---|
| committer | TeresaARM <teresa.charlinreyes@arm.com> | 2021-04-20 13:20:59 +0000 |
| commit | 562bee584f3633167725af7915f50d07b0597f10 (patch) | |
| tree | cc0a5658cfd82f3d1552103472253734e1f437cd /arm_compute | |
| parent | ada6cbc057ff725e57d301a99a1816ce602485b9 (diff) | |
| download | ComputeLibrary-562bee584f3633167725af7915f50d07b0597f10.tar.gz | |
Port CpuConvertFullyConnectedWeights to new API
* Remove includes of NEConvertFullyConnectedWeightsKernel.h
Resolves partially: COMPMID-4187
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I1bf246546d3ef53edb4c5a8bc05a0db92d2d3bff
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5418
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute')
| -rw-r--r-- | arm_compute/runtime/NEON/functions/NEConvertFullyConnectedWeights.h | 12 |
1 files changed, 5 insertions, 7 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEConvertFullyConnectedWeights.h b/arm_compute/runtime/NEON/functions/NEConvertFullyConnectedWeights.h index 984e8d68c0..f9ce66db13 100644 --- a/arm_compute/runtime/NEON/functions/NEConvertFullyConnectedWeights.h +++ b/arm_compute/runtime/NEON/functions/NEConvertFullyConnectedWeights.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2018-2020 Arm Limited. + * Copyright (c) 2018-2021 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -26,17 +26,14 @@ #include "arm_compute/runtime/IFunction.h" #include "arm_compute/runtime/ITransformWeights.h" -#include "arm_compute/runtime/NEON/NEScheduler.h" #include "arm_compute/runtime/Tensor.h" -#include <memory> namespace arm_compute { // Forward declarations class ITensor; -class NEConvertFullyConnectedWeightsKernel; -/** Basic function to run @ref NEConvertFullyConnectedWeightsKernel. */ +/** Basic function to run @ref cpu::kernels::CpuConvertFullyConnectedWeightsKernel. */ class NEConvertFullyConnectedWeights : public IFunction { public: @@ -75,12 +72,13 @@ public: void run() override; private: - std::unique_ptr<NEConvertFullyConnectedWeightsKernel> _kernel; + struct Impl; + std::unique_ptr<Impl> _impl; }; namespace weights_transformations { -/** Basic function to run @ref NEConvertFullyConnectedWeightsKernel. */ +/** Basic function to manage @ref NEConvertFullyConnectedWeights. */ class NEConvertFullyConnectedWeightsManaged : public ITransformWeights { public: |