1 files changed, 8 insertions, 29 deletions
diff --git a/arm_compute/core/NEON/kernels/NEUpsampleLayerKernel.h b/arm_compute/core/NEON/kernels/NEUpsampleLayerKernel.h
index 50c3e342e3..1ea3f974e7 100644
--- a/arm_compute/core/NEON/kernels/NEUpsampleLayerKernel.h
+++ b/arm_compute/core/NEON/kernels/NEUpsampleLayerKernel.h
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2018-2019 ARM Limited.
+ * Copyright (c) 2018-2020 ARM Limited.
  *
  * SPDX-License-Identifier: MIT
  *
@@ -74,40 +74,19 @@ public:
     void run(const Window &window, const ThreadInfo &info) override;
 
 private:
-    /** Function to run upsample layer for FP32 (NCHW)
+    /** Function to run upsample layer (NCHW)
      *
      * @param[in] window Region on which to execute the kernel. (Must be a valid region of the window returned by window()).
      */
-    void upsample_f32_nchw(const Window &window);
-    /** Function to run upsample layer for FP32 (NHWC)
+    template <typename T, int S>
+    void upsample_nchw(const Window &window);
+    /** Function to run upsample layer (NHWC)
      *
      * @param[in] window Region on which to execute the kernel. (Must be a valid region of the window returned by window()).
      */
-    void upsample_f32_nhwc(const Window &window);
-    /** Function to run upsample layer for FP16 (NCHW)
-     *
-     * @param[in] window Region on which to execute the kernel. (Must be a valid region of the window returned by window()).
-     */
-    void upsample_f16_nchw(const Window &window);
-    /** Function to run upsample layer for FP16 (NHWC)
-     *
-     * @param[in] window Region on which to execute the kernel. (Must be a valid region of the window returned by window()).
-     */
-    void upsample_f16_nhwc(const Window &window);
-    /** Function to run upsample layer for QASYMM8 (NCHW)
-     *
-     * @param[in] window Region on which to execute the kernel. (Must be a valid region of the window returned by window()).
-     */
-    void upsample_qasymm8_nchw(const Window &window);
-    /** Function to run upsample layer for QASYMM8 (NHWC)
-     *
-     * @param[in] window Region on which to execute the kernel. (Must be a valid region of the window returned by window()).
-     */
-    void upsample_qasymm8_nhwc(const Window &window);
-    /** Common signature for all the upsample layer functions
-     *
-     * @param[in] window Region on which to execute the kernel.
-     */
+    template <typename T, int S>
+    void upsample_nhwc(const Window &window);
+
     using UpsampleFunctionPtr = void (NEUpsampleLayerKernel::*)(const Window &window);
 
 private: