diff options
author | Viet-Hoa Do <viet-hoa.do@arm.com> | 2022-10-07 15:58:05 +0100 |
---|---|---|
committer | Pablo Marquez Tello <pablo.tello@arm.com> | 2022-10-10 12:24:18 +0000 |
commit | 66704133478dcb2065e6313590187315de74ec5f (patch) | |
tree | 6f1af66aed2a2d03bf814a7cf7ae2165463ce0b3 /src/cpu/kernels/activation/list.h | |
parent | 2ffab6dc8afb3716c189a30f75c33b2f2c35a6d4 (diff) | |
download | ComputeLibrary-66704133478dcb2065e6313590187315de74ec5f.tar.gz |
Fix LUT-based activation layer
* Use the window instead of the tensor shape to determine the
number of elements in the x-dimension.
* Remove the LUT implementation in 32-bit build.
Resolves: COMPMID-5641
Signed-off-by: Viet-Hoa Do <viet-hoa.do@arm.com>
Change-Id: I0a79aa38d8f6a105ad01785bd94571f5a2ecb348
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/8380
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Pablo Marquez Tello <pablo.tello@arm.com>
Diffstat (limited to 'src/cpu/kernels/activation/list.h')
-rw-r--r-- | src/cpu/kernels/activation/list.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/cpu/kernels/activation/list.h b/src/cpu/kernels/activation/list.h index 3850d4de6b..c0a2446748 100644 --- a/src/cpu/kernels/activation/list.h +++ b/src/cpu/kernels/activation/list.h @@ -31,7 +31,10 @@ namespace cpu #define DECLARE_ACTIVATION_KERNEL(func_name) \ void func_name(const ITensor *src, ITensor *dst, const ActivationLayerInfo &act_info, const Window &window) +#ifdef __aarch64__ DECLARE_ACTIVATION_KERNEL(neon_q8_activation_lut); +#endif // __aarch64__ + DECLARE_ACTIVATION_KERNEL(neon_qasymm8_activation); DECLARE_ACTIVATION_KERNEL(sve2_qasymm8_activation); DECLARE_ACTIVATION_KERNEL(neon_qasymm8_signed_activation); |