diff options
author | Sheri Zhang <sheri.zhang@arm.com> | 2020-12-15 20:25:31 +0000 |
---|---|---|
committer | Sheri Zhang <sheri.zhang@arm.com> | 2020-12-24 17:19:35 +0000 |
commit | 8d5d78ba48358e5c511d4c625c17d99065763945 (patch) | |
tree | 446b0d851a36c08af7423e8254699f6b24dd6f4d /Android.bp | |
parent | 410e21e88db9d98c8144cd93047e506ecd0b7ab4 (diff) | |
download | ComputeLibrary-8d5d78ba48358e5c511d4c625c17d99065763945.tar.gz |
COMPMID-3871: Create BatchNormalization SVE/SVE2
1. Decouple data type for NHWC
2. Add NHWC SVE support for BachNormalization
Signed-off-by: Sheri Zhang <sheri.zhang@arm.com>
Change-Id: I0383b969b555b429d9acebb4efa17ecba9429ea7
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/4755
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michalis Spyrou <michalis.spyrou@arm.com>
Diffstat (limited to 'Android.bp')
-rw-r--r-- | Android.bp | 4 |
1 files changed, 4 insertions, 0 deletions
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", |