commit 76faef88284e6fd51f53b23063374d3d3a884e4f
author Gian Marco <gianmarco.iodice@arm.com> Mon Jan 29 12:15:32 2018 +0000
committer Anthony Barbier <anthony.barbier@arm.com> Fri Nov 02 16:45:42 2018 +0000
tree d29dfaa733edc7928125ade473b278e1e4923b6e
parent b28f329d0cdc2e5f2f72fd3ecc1913c6b913aee8

COMPMID-855 - Optimizing im2col on OpenCL (DCHW)

Introduced optimizations for 1x1, 3x3, 5x5 and 11x11

Change-Id: Ibb7f7a9fbec01a7684746ed8513634078126e452
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/118107
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Michalis Spyrou <michalis.spyrou@arm.com>

arm_compute/core/CL/kernels/CLIm2ColKernel.h

diff --git a/arm_compute/core/CL/kernels/CLIm2ColKernel.h b/arm_compute/core/CL/kernels/CLIm2ColKernel.h
index 88de1ba..e38e7e8 100644
--- a/arm_compute/core/CL/kernels/CLIm2ColKernel.h
+++ b/arm_compute/core/CL/kernels/CLIm2ColKernel.h
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2017 ARM Limited.
+ * Copyright (c) 2017-2018 ARM Limited.
  *
  * SPDX-License-Identifier: MIT
  *
@@ -25,11 +25,11 @@
 #define __ARM_COMPUTE_CLIM2COLKERNEL_H__
 
 #include "arm_compute/core/CL/ICLKernel.h"
+#include "arm_compute/core/Size2D.h"
 
 namespace arm_compute
 {
 class ICLTensor;
-class Size2D;
 
 /** Interface for the im2col reshape kernel.
  *
@@ -117,6 +117,7 @@
     std::pair<unsigned int, unsigned int> _convolved_dims;
     unsigned int   _num_elems_processed_per_iteration;
     Im2ColFunction _run_func;
+    Size2D         _kernel_dims;
 };
 } // namespace arm_compute
 #endif /*__ARM_COMPUTE_CLIM2COLKERNEL_H__ */