diff options
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NEConvertFullyConnectedWeights.h')
-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: |