aboutsummaryrefslogtreecommitdiff
path: root/src/core/CL/cl_kernels/pooling_layer.cl
diff options
context:
space:
mode:
authorAnthony Barbier <anthony.barbier@arm.com>2017-07-03 17:39:37 +0100
committerAnthony Barbier <anthony.barbier@arm.com>2018-09-17 14:15:39 +0100
commitac69aa137e360340fe9f148f019d93af6c3d8336 (patch)
tree88c2fb6ea8693c69b19d3b7b38fe26cd916303c8 /src/core/CL/cl_kernels/pooling_layer.cl
parent05da6dd102c988081c7d5eccb227f559f740ceef (diff)
downloadComputeLibrary-ac69aa137e360340fe9f148f019d93af6c3d8336.tar.gz
COMPMID-418 Add check and fix comments after preprocessor conditions
Change-Id: I1353fd652ee180e3931e58b4ce13d651a48c7e2c Reviewed-on: http://mpd-gerrit.cambridge.arm.com/79567 Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com> Reviewed-by: Moritz Pflanzer <moritz.pflanzer@arm.com>
Diffstat (limited to 'src/core/CL/cl_kernels/pooling_layer.cl')
-rw-r--r--src/core/CL/cl_kernels/pooling_layer.cl22
1 files changed, 11 insertions, 11 deletions
diff --git a/src/core/CL/cl_kernels/pooling_layer.cl b/src/core/CL/cl_kernels/pooling_layer.cl
index 6bdb174235..b7245203d4 100644
--- a/src/core/CL/cl_kernels/pooling_layer.cl
+++ b/src/core/CL/cl_kernels/pooling_layer.cl
@@ -23,11 +23,11 @@
*/
#include "helpers.h"
-#if defined POOL_AVG
+#ifdef POOL_AVG
#define POOL_OP(x, y) ((x) + (y))
-#else
+#else /* POOL_AVG */
#define POOL_OP(x, y) (fmax((x), (y)))
-#endif
+#endif /* POOL_AVG */
float calculate_avg_scale(const int pool_size, const int upper_bound_w, const int upper_bound_h,
const int pad_x, const int pad_y, const int stride_x, const int stride_y)
@@ -70,7 +70,7 @@ __kernel void pooling_layer_2(
#ifdef POOL_AVG
,
int2 max_dims, int2 strides, int2 paddings
-#endif
+#endif /* POOL_AVG */
)
{
// Get pixels pointer
@@ -90,7 +90,7 @@ __kernel void pooling_layer_2(
// Divide by pool region in case of average pooling
#ifdef POOL_AVG
res *= calculate_avg_scale(2, max_dims.x, max_dims.y, paddings.x, paddings.y, strides.x, strides.y);
-#endif
+#endif /* POOL_AVG */
// Store result
*(__global DATA_TYPE *)output.ptr = res;
@@ -127,7 +127,7 @@ __kernel void pooling_layer_3(
#ifdef POOL_AVG
,
int2 max_dims, int2 strides, int2 paddings
-#endif
+#endif /* POOL_AVG */
)
{
// Get pixels pointer
@@ -150,7 +150,7 @@ __kernel void pooling_layer_3(
// Divide by pool region in case of average pooling
#ifdef POOL_AVG
res *= calculate_avg_scale(3, max_dims.x, max_dims.y, paddings.x, paddings.y, strides.x, strides.y);
-#endif
+#endif /* POOL_AVG */
// Store result
*(__global DATA_TYPE *)output.ptr = res;
@@ -187,7 +187,7 @@ __kernel void pooling_layer_7(
#ifdef POOL_AVG
,
int2 max_dims, int2 strides, int2 paddings
-#endif
+#endif /* POOL_AVG */
)
{
// Get pixels pointer
@@ -221,9 +221,9 @@ __kernel void pooling_layer_7(
// Set last element
#ifdef POOL_AVG
data0.s7 = 0;
-#else
+#else /* POOL_AVG */
data0.s7 = data0.s6;
-#endif
+#endif /* POOL_AVG */
// Reduce result
VEC_DATA_TYPE(DATA_TYPE, 4)
@@ -235,7 +235,7 @@ __kernel void pooling_layer_7(
// Divide by pool region in case of average pooling
#ifdef POOL_AVG
res *= calculate_avg_scale(7, max_dims.x, max_dims.y, paddings.x, paddings.y, strides.x, strides.y);
-#endif
+#endif /* POOL_AVG */
// Store result
*(__global DATA_TYPE *)output.ptr = res;