diff options
author | Michalis Spyrou <michalis.spyrou@arm.com> | 2019-04-09 14:03:17 +0100 |
---|---|---|
committer | Michalis Spyrou <michalis.spyrou@arm.com> | 2019-04-11 09:16:00 +0000 |
commit | a50e702289af66944e860eafc7f3b32f6c5f30be (patch) | |
tree | 9803990dd7eed6686cbe01a3ca1b6ed90d555261 /arm_compute/core/NEON/kernels | |
parent | 8d94269d7985b9cee67e52581e2f58b6c99d7f0d (diff) | |
download | ComputeLibrary-a50e702289af66944e860eafc7f3b32f6c5f30be.tar.gz |
COMPMID-2012: Remove unnecessary templates from NEON kernels
Change-Id: I2deb26188c7de7c6ad10d2f51f83e729fed7e5e2
Signed-off-by: Michalis Spyrou <michalis.spyrou@arm.com>
Reviewed-on: https://review.mlplatform.org/c/961
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/core/NEON/kernels')
-rw-r--r-- | arm_compute/core/NEON/kernels/NEReorgLayerKernel.h | 23 | ||||
-rw-r--r-- | arm_compute/core/NEON/kernels/NEStackLayerKernel.h | 14 | ||||
-rw-r--r-- | arm_compute/core/NEON/kernels/NEWeightsReshapeKernel.h | 11 |
3 files changed, 8 insertions, 40 deletions
diff --git a/arm_compute/core/NEON/kernels/NEReorgLayerKernel.h b/arm_compute/core/NEON/kernels/NEReorgLayerKernel.h index 7e0fb4350d..076af4fd1c 100644 --- a/arm_compute/core/NEON/kernels/NEReorgLayerKernel.h +++ b/arm_compute/core/NEON/kernels/NEReorgLayerKernel.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2018 ARM Limited. + * Copyright (c) 2018-2019 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -75,24 +75,9 @@ public: void run(const Window &window, const ThreadInfo &info) override; private: - /** Template function to run the reorg - * - * @param[in] window Region on which to execute the kernel. (Must be a valid region of the window returned by window()). - */ - template <typename T> - void run_reorg(const Window &window); - - /** Common signature for all the specialised reorg functions - * - * @param[in] window Region on which to execute the kernel. - */ - using ReorgFunctionPtr = void (NEReorgLayerKernel::*)(const Window &window); - -private: - ReorgFunctionPtr _func; - const ITensor *_input; - ITensor *_output; - int32_t _stride; + const ITensor *_input; + ITensor *_output; + int32_t _stride; }; } // namespace arm_compute #endif /*__ARM_COMPUTE_NEREORGLAYERKERNEL_H__ */ diff --git a/arm_compute/core/NEON/kernels/NEStackLayerKernel.h b/arm_compute/core/NEON/kernels/NEStackLayerKernel.h index 3a9e81fa94..42a0539c9f 100644 --- a/arm_compute/core/NEON/kernels/NEStackLayerKernel.h +++ b/arm_compute/core/NEON/kernels/NEStackLayerKernel.h @@ -84,24 +84,10 @@ public: void run(const Window &window, const ThreadInfo &info) override; private: - /** Template function to run the stack - * - * @param[in] window Region on which to execute the kernel. (Must be a valid region of the window returned by window()). - */ - template <typename T> - void run_stack(const Window &window); - - /** Common signature for all the specialised stack functions - * - * @param[in] window Region on which to execute the kernel. - */ - using StackFunctionPtr = void (NEStackLayerKernel::*)(const Window &window); - const ITensor *_input; ITensor *_output; unsigned int _axis; unsigned int _idx_input; - StackFunctionPtr _func; }; } // namespace arm_compute #endif /* __ARM_COMPUTE_NESTACKLAYERKERNEL_H__ */ diff --git a/arm_compute/core/NEON/kernels/NEWeightsReshapeKernel.h b/arm_compute/core/NEON/kernels/NEWeightsReshapeKernel.h index 21f36f6c2b..bba18a8fa8 100644 --- a/arm_compute/core/NEON/kernels/NEWeightsReshapeKernel.h +++ b/arm_compute/core/NEON/kernels/NEWeightsReshapeKernel.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017-2018 ARM Limited. + * Copyright (c) 2017-2019 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -99,12 +99,9 @@ public: void run(const Window &window, const ThreadInfo &info) override; private: - using WeightsReshapeKernel = void(const ITensor *input, const ITensor *bias, ITensor *output, const Window &window); - - WeightsReshapeKernel *_func; - const ITensor *_input; - const ITensor *_bias; - ITensor *_output; + const ITensor *_input; + const ITensor *_bias; + ITensor *_output; }; } // namespace arm_compute #endif /*__ARM_COMPUTE_NEWEIGHTSRESHAPEKERNEL_H__ */ |