aboutsummaryrefslogtreecommitdiff
path: root/arm_compute
diff options
context:
space:
mode:
authorMichalis Spyrou <michalis.spyrou@arm.com>2019-04-09 14:03:17 +0100
committerMichalis Spyrou <michalis.spyrou@arm.com>2019-04-11 09:16:00 +0000
commita50e702289af66944e860eafc7f3b32f6c5f30be (patch)
tree9803990dd7eed6686cbe01a3ca1b6ed90d555261 /arm_compute
parent8d94269d7985b9cee67e52581e2f58b6c99d7f0d (diff)
downloadComputeLibrary-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')
-rw-r--r--arm_compute/core/NEON/kernels/NEReorgLayerKernel.h23
-rw-r--r--arm_compute/core/NEON/kernels/NEStackLayerKernel.h14
-rw-r--r--arm_compute/core/NEON/kernels/NEWeightsReshapeKernel.h11
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__ */