aboutsummaryrefslogtreecommitdiff
path: root/arm_compute
diff options
context:
space:
mode:
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__ */