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>
diff --git a/filelist.json b/filelist.json
index 112b1aa..b72b7b4 100644
--- a/filelist.json
+++ b/filelist.json
@@ -901,8 +901,21 @@
"src/runtime/NEON/functions/NEFuseBatchNormalization.cpp"
],
"neon": {
- "fp16": [ "src/core/NEON/kernels/batchnormalization/impl/NEON/fp16.cpp" ],
- "fp32": [ "src/core/NEON/kernels/batchnormalization/impl/NEON/fp32.cpp" ]
+ "common": [
+ "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/impl.cpp"
+ ],
+ "fp16": [
+ "src/cpu/kernels/fuse_batch_normalization/generic/fp16.cpp",
+ "src/core/NEON/kernels/batchnormalization/impl/NEON/fp16.cpp",
+ "src/cpu/kernels/fuse_batch_normalization/nhwc/neon/fp16.cpp"
+ ],
+ "fp32": [
+ "src/cpu/kernels/fuse_batch_normalization/generic/fp32.cpp",
+ "src/core/NEON/kernels/batchnormalization/impl/NEON/fp32.cpp",
+ "src/cpu/kernels/fuse_batch_normalization/nhwc/neon/fp32.cpp"
+ ]
},
"sve": {
"fp16": [ "src/core/NEON/kernels/batchnormalization/impl/SVE/fp16.cpp" ],