From 8d5d78ba48358e5c511d4c625c17d99065763945 Mon Sep 17 00:00:00 2001 From: Sheri Zhang Date: Tue, 15 Dec 2020 20:25:31 +0000 Subject: COMPMID-3871: Create BatchNormalization SVE/SVE2 1. Decouple data type for NHWC 2. Add NHWC SVE support for BachNormalization Signed-off-by: Sheri Zhang Change-Id: I0383b969b555b429d9acebb4efa17ecba9429ea7 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/4755 Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins Reviewed-by: Michalis Spyrou --- Android.bp | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'Android.bp') diff --git a/Android.bp b/Android.bp index 9f7f447fe3..1032950f3e 100644 --- a/Android.bp +++ b/Android.bp @@ -368,6 +368,10 @@ cc_library_static { "src/core/NEON/kernels/arm_gemm/quantized.cpp", "src/core/NEON/kernels/arm_gemm/rowsum_indirect_s8.cpp", "src/core/NEON/kernels/arm_gemm/rowsum_indirect_u8.cpp", + "src/core/NEON/kernels/batchnormalization/impl/NEON/fp16.cpp", + "src/core/NEON/kernels/batchnormalization/impl/NEON/fp32.cpp", + "src/core/NEON/kernels/batchnormalization/impl/SVE/fp16.cpp", + "src/core/NEON/kernels/batchnormalization/impl/SVE/fp32.cpp", "src/core/NEON/kernels/convolution/common/padding.cpp", "src/core/NEON/kernels/convolution/common/qasymm8.cpp", "src/core/NEON/kernels/convolution/common/qsymm8.cpp", -- cgit v1.2.1