diff options
| author | Viet-Hoa Do <viet-hoa.do@arm.com> | 2022-06-21 15:56:15 +0100 |
|---|---|---|
| committer | Viet-Hoa Do <viet-hoa.do@arm.com> | 2022-06-29 13:30:06 +0000 |
| commit | b042e39060901b44e615b923b5723c04d9b42a95 (patch) | |
| tree | e23fd9b89c753f9731e1e8ec4a0d9ca468f9f683 /src/cpu/kernels/activation/list.h | |
| parent | 13f96d0a5efc140785a6de58bff9b24b80dd0cfd (diff) | |
| download | ComputeLibrary-b042e39060901b44e615b923b5723c04d9b42a95.tar.gz | |
Add LUT-based leaky relu for QASYMM8 on CPU
* Add LUT generation function for Leaky ReLU.
* Some additional changes in the existing LUT implementation:
+ Bring back the NEON implementation of hard swish for 32-bit
build. Library size of 64-bit build is not affected.
+ Add some extra #ifdef to remove unnecessary code in 32-bit
build.
Resolves: COMPMID-5386
Signed-off-by: Viet-Hoa Do <viet-hoa.do@arm.com>
Change-Id: I1ea49611cc922765ee741e31138c888401d33e9b
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/7845
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gunes Bayir <gunes.bayir@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/cpu/kernels/activation/list.h')
| -rw-r--r-- | src/cpu/kernels/activation/list.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/cpu/kernels/activation/list.h b/src/cpu/kernels/activation/list.h index 7220d6cce1..b2322a6477 100644 --- a/src/cpu/kernels/activation/list.h +++ b/src/cpu/kernels/activation/list.h @@ -32,7 +32,7 @@ namespace cpu void func_name(const ITensor *src, ITensor *dst, const ActivationLayerInfo &act_info, const Window &window) DECLARE_ACTIVATION_KERNEL(neon_qasymm8_activation); -DECLARE_ACTIVATION_KERNEL(neon_qasymm8_hardswish_lut); +DECLARE_ACTIVATION_KERNEL(neon_qasymm8_activation_lut); DECLARE_ACTIVATION_KERNEL(sve2_qasymm8_activation); DECLARE_ACTIVATION_KERNEL(neon_qasymm8_signed_activation); DECLARE_ACTIVATION_KERNEL(sve2_qasymm8_signed_activation); |