From 149203bc23d5c84fe1326d9dea4730750fab6710 Mon Sep 17 00:00:00 2001 From: Dana Zlotnik Date: Wed, 26 Jan 2022 12:38:03 +0200 Subject: Port MaxUnpoolingLayer kernel and add KernelSelect vaidation test Resolves COMPMID-4958 Change-Id: Ibed5155f2e3ece46635f6ea9617bf11cefc402b1 Signed-off-by: Dana Zlotnik Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/7028 Tested-by: Arm Jenkins Reviewed-by: Giorgio Arena Comments-Addressed: Arm Jenkins --- arm_compute/runtime/NEON/functions/NEMaxUnpoolingLayer.h | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) (limited to 'arm_compute/runtime/NEON') diff --git a/arm_compute/runtime/NEON/functions/NEMaxUnpoolingLayer.h b/arm_compute/runtime/NEON/functions/NEMaxUnpoolingLayer.h index 41ea040457..2f77540e1e 100644 --- a/arm_compute/runtime/NEON/functions/NEMaxUnpoolingLayer.h +++ b/arm_compute/runtime/NEON/functions/NEMaxUnpoolingLayer.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2020-2021 Arm Limited. + * Copyright (c) 2020-2022 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -33,12 +33,10 @@ namespace arm_compute class ITensor; class ITensorInfo; class NEFill; -class NEMaxUnpoolingLayerKernel; /** Function to perform MaxUnpooling. This function calls the following kernels: * * -# @ref NEFill - * -# @ref NEMaxUnpoolingLayerKernel */ class NEMaxUnpoolingLayer : public IFunction { @@ -94,8 +92,9 @@ public: void run() override; private: - std::unique_ptr _fill_func; - std::unique_ptr _unpooling_layer_kernel; + std::unique_ptr _fill_func; + struct Impl; + std::unique_ptr _impl; }; } #endif /* ARM_COMPUTE_NEMAXUNPOOLINGLAYER_H */ -- cgit v1.2.1