From 41a729edf9facc6e901055e0cc84219f75670475 Mon Sep 17 00:00:00 2001 From: Yair Schwarzbaum Date: Mon, 15 Nov 2021 20:42:47 +0200 Subject: Decouple fuseBatchNormalizationKernel - Decouple data type for CPU implementation supported data types are: fp32, fp16 Resolves COMPMID-4613 Signed-off-by: Yair Schwarzbaum Change-Id: I8aff3ba2d446f64e4d182a866e3a3debc9ef613b Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/7175 Reviewed-by: Giorgio Arena Tested-by: Arm Jenkins Comments-Addressed: Arm Jenkins --- Android.bp | 7 +++++++ 1 file changed, 7 insertions(+) (limited to 'Android.bp') diff --git a/Android.bp b/Android.bp index c858b07230..491dbca165 100644 --- a/Android.bp +++ b/Android.bp @@ -468,6 +468,13 @@ cc_library_static { "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", -- cgit v1.2.1