From fb62908bd8148bd347bd204e881156f8ebf7835d Mon Sep 17 00:00:00 2001
From: Giorgio Arena <giorgio.arena@arm.com>
Date: Mon, 20 Aug 2018 18:03:27 +0100
Subject: COMPMID-1494 Optimise NEON im2col and weights reshape for NHWC

Change-Id: I99ebae61024a7bce9d17292a02c28626ae6c29d5
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/144872
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
---
 arm_compute/core/NEON/kernels/NEIm2ColKernel.h | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'arm_compute/core/NEON')

diff --git a/arm_compute/core/NEON/kernels/NEIm2ColKernel.h b/arm_compute/core/NEON/kernels/NEIm2ColKernel.h
index ec89f0f713..f76521f770 100644
--- a/arm_compute/core/NEON/kernels/NEIm2ColKernel.h
+++ b/arm_compute/core/NEON/kernels/NEIm2ColKernel.h
@@ -109,11 +109,11 @@ public:
     void run(const Window &window, const ThreadInfo &info) override;
 
 private:
-    /** Template function to run the im2col
+    /** Template function to run im2col
      *
      * @param[in] window Region on which to execute the kernel. (Must be a valid region of the window returned by window()).
      */
-    template <typename T, bool has_pads>
+    template <typename T, bool has_pads, bool is_nchw>
     void run_im2col(const Window &window);
 
     /** Common signature for all the specialised im2col functions
-- 
cgit v1.2.1