diff options
author | Yair Schwarzbaum <yair.schwarzbaum@arm.com> | 2021-11-15 20:42:47 +0200 |
---|---|---|
committer | Yair Schwarzbaum <yair.schwarzbaum@arm.com> | 2022-03-08 16:26:46 +0000 |
commit | 41a729edf9facc6e901055e0cc84219f75670475 (patch) | |
tree | 86ca99fa9bef53220874195a1f78aa2ae11c5ab8 /filelist.json | |
parent | 232c45253a84c16fc70eae6406cac5f4048efaca (diff) | |
download | ComputeLibrary-41a729edf9facc6e901055e0cc84219f75670475.tar.gz |
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>
Diffstat (limited to 'filelist.json')
-rw-r--r-- | filelist.json | 17 |
1 files changed, 15 insertions, 2 deletions
diff --git a/filelist.json b/filelist.json index 112b1aac7d..b72b7b4591 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" ], |