Fix build error
* Build error when using data_layout_support=nhwc
* Some kernels need to be guarded by ENABLE_NCHW_KERNELS
Change-Id: I9fb6cf0e204531f81b0dff3572a1740ba94cde0e
Signed-off-by: Pablo Marquez Tello <pablo.tello@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/10460
Reviewed-by: Viet-Hoa Do <viet-hoa.do@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
diff --git a/src/cpu/kernels/pool2d/neon/fp16.cpp b/src/cpu/kernels/pool2d/neon/fp16.cpp
index 95ff7b7..9d24d79 100644
--- a/src/cpu/kernels/pool2d/neon/fp16.cpp
+++ b/src/cpu/kernels/pool2d/neon/fp16.cpp
@@ -37,6 +37,8 @@
{
namespace cpu
{
+#ifdef ENABLE_NCHW_KERNELS
+
namespace
{
float16x4_t
@@ -148,6 +150,7 @@
},
in, out);
}
+#endif // ENABLE_NCHW_KERNELS
void pooling2_f16_maxpool_indices(const ITensor *src,
ITensor *dst0,
@@ -278,6 +281,7 @@
},
in, out, indices);
}
+#ifdef ENABLE_NCHW_KERNELS
void pooling2_fp16_neon_nchw(const ITensor *src,
ITensor *dst0,
@@ -461,6 +465,7 @@
},
in, out);
}
+#endif // ENABLE_NCHW_KERNELS
void poolingMxN_fp16_neon_nhwc(const ITensor *src,
ITensor *dst0,