commit 41a729edf9facc6e901055e0cc84219f75670475
author Yair Schwarzbaum <yair.schwarzbaum@arm.com> Mon Nov 15 20:42:47 2021 +0200
committer Yair Schwarzbaum <yair.schwarzbaum@arm.com> Tue Mar 08 16:26:46 2022 +0000
tree 86ca99fa9bef53220874195a1f78aa2ae11c5ab8
parent 232c45253a84c16fc70eae6406cac5f4048efaca

Decouple fuseBatchNormalizationKernel

- Decouple data type for CPU implementation
    supported data types are: fp32, fp16

Resolves COMPMID-4613
Signed-off-by: Yair Schwarzbaum <yair.schwarzbaum@arm.com>
Change-Id: I8aff3ba2d446f64e4d182a866e3a3debc9ef613b
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/7175
Reviewed-by: Giorgio Arena <giorgio.arena@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>

Android.bp

diff --git a/Android.bp b/Android.bp
index c858b07..491dbca 100644
--- a/Android.bp
+++ b/Android.bp
@@ -468,6 +468,13 @@
         "src/cpu/kernels/elementwise_unary/generic/neon/integer.cpp",
         "src/cpu/kernels/floor/neon/fp16.cpp",
         "src/cpu/kernels/floor/neon/fp32.cpp",
+        "src/cpu/kernels/fuse_batch_normalization/generic/fp16.cpp",
+        "src/cpu/kernels/fuse_batch_normalization/generic/fp32.cpp",
+        "src/cpu/kernels/fuse_batch_normalization/generic/impl.cpp",
+        "src/cpu/kernels/fuse_batch_normalization/nchw/all.cpp",
+        "src/cpu/kernels/fuse_batch_normalization/nhwc/neon/fp16.cpp",
+        "src/cpu/kernels/fuse_batch_normalization/nhwc/neon/fp32.cpp",
+        "src/cpu/kernels/fuse_batch_normalization/nhwc/neon/impl.cpp",
         "src/cpu/kernels/gemm_matrix_add/generic/neon/fp16.cpp",
         "src/cpu/kernels/gemm_matrix_add/generic/neon/fp32.cpp",
         "src/cpu/kernels/gemm_matrix_add/generic/neon/impl.cpp",