From 562bee584f3633167725af7915f50d07b0597f10 Mon Sep 17 00:00:00 2001 From: Teresa Charlin Date: Tue, 13 Apr 2021 17:44:15 +0100 Subject: Port CpuConvertFullyConnectedWeights to new API * Remove includes of NEConvertFullyConnectedWeightsKernel.h Resolves partially: COMPMID-4187 Signed-off-by: Teresa Charlin Change-Id: I1bf246546d3ef53edb4c5a8bc05a0db92d2d3bff Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5418 Tested-by: Arm Jenkins Reviewed-by: Michele Di Giorgio Comments-Addressed: Arm Jenkins --- .../runtime/NEON/functions/NEConvertFullyConnectedWeights.h | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) (limited to 'arm_compute') 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 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 _kernel; + struct Impl; + std::unique_ptr _impl; }; namespace weights_transformations { -/** Basic function to run @ref NEConvertFullyConnectedWeightsKernel. */ +/** Basic function to manage @ref NEConvertFullyConnectedWeights. */ class NEConvertFullyConnectedWeightsManaged : public ITransformWeights { public: -- cgit v1.2.1