Add if defined block and remove ambiguous comments from BatchToSpaceND CL kernel...

This patch adds if defined block and removes ambiguous comment from BatchToSpaceND CL Kernel.

Signed-off-by: prasannar <prasanna.r@samsung.com>

diff --git a/libs/ARMComputeEx/src/core/CL/cl_kernels/batch_to_space_nd.cl b/libs/ARMComputeEx/src/core/CL/cl_kernels/batch_to_space_nd.cl
index e4c6e28..66da326 100644 (file)
--- a/libs/ARMComputeEx/src/core/CL/cl_kernels/batch_to_space_nd.cl
+++ b/libs/ARMComputeEx/src/core/CL/cl_kernels/batch_to_space_nd.cl
@@ -16,7 +16,8 @@
  */
 #include "helpers.h"
 
-/** Perform space to depth rearrangement of tensor
+#if defined(DATA_TYPE) && defined(DEPTH_OUT) && defined(BLOCK_SIZE0) && defined(BLOCK_SIZE1)
+/** Perform batch to space rearrangement of tensor
  *
  * @attention Data type can be passed using the -DDATA_TYPE compile flag, e.g. -DDATA_TYPE=float
  * @attention Output tensor batch should be given as a preprocessor argument using -DBATCH_OUT=size. e.g. -DBATCH_OUT=16
@@ -66,3 +67,4 @@ __kernel void batch_to_space_nd(
 
     *((__global DATA_TYPE *)out.ptr) = *((__global DATA_TYPE *)tensor4D_offset(&in, in_index[0], in_index[1], in_index[2], in_index[3]));
  }
+#endif // defined(DATA_TYPE) && defined(DEPTH_IN) && defined(BLOCK_SIZE0) && defined(BLOCK_SIZE1)