diff options
| author | Dana Zlotnik <dana.zlotnik@arm.com> | 2021-12-30 13:24:52 +0200 |
|---|---|---|
| committer | Dana Zlotnik <dana.zlotnik@arm.com> | 2022-01-06 12:41:04 +0000 |
| commit | 143310c6ec5e7c9f6035e9357bf0faa4020f2412 (patch) | |
| tree | 1085ac6269bccd23108de219f046fc052590d90f /src/core/NEON/kernels/NEMaxUnpoolingLayerKernel.h | |
| parent | b99e54e05a4e9fed3bae1172c874af87e712526b (diff) | |
| download | ComputeLibrary-143310c6ec5e7c9f6035e9357bf0faa4020f2412.tar.gz | |
Decouple NEMaxUnpoolingLayerKernel
Resolves COMPMID-4619
Change-Id: I9c43dcd3fb3a688e1c0ccc858a02376741381ba7
Signed-off-by: Dana Zlotnik <dana.zlotnik@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6874
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Giorgio Arena <giorgio.arena@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core/NEON/kernels/NEMaxUnpoolingLayerKernel.h')
| -rw-r--r-- | src/core/NEON/kernels/NEMaxUnpoolingLayerKernel.h | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/src/core/NEON/kernels/NEMaxUnpoolingLayerKernel.h b/src/core/NEON/kernels/NEMaxUnpoolingLayerKernel.h index ecc116e585..f7f9a31f6d 100644 --- a/src/core/NEON/kernels/NEMaxUnpoolingLayerKernel.h +++ b/src/core/NEON/kernels/NEMaxUnpoolingLayerKernel.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2020-2021 Arm Limited. + * Copyright (c) 2020-2022 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -88,10 +88,9 @@ private: using UnpoolingFunction = void (NEMaxUnpoolingLayerKernel::*)(const Window &window); private: - UnpoolingFunction _func; - const ITensor *_input; - ITensor *_output; - const ITensor *_indices; + const ITensor *_input; + ITensor *_output; + const ITensor *_indices; }; } // namespace arm_compute #endif /*ARM_COMPUTE_NEMAXUNPOOLINGLAYERKERNEL_H */ |