diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-08-22 16:20:21 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:54:54 +0000 |
| commit | aaba4c626bcc6365e0108130633ce43fafe9da45 (patch) | |
| tree | 768ee62e90b9bf39dfbfe3a6551a45071e83c6c3 /arm_compute/core/NEON/wrapper/intrinsics/max.h | |
| parent | 630d941836e32fadca62549c177cb57d4436b44b (diff) | |
| download | ComputeLibrary-aaba4c626bcc6365e0108130633ce43fafe9da45.tar.gz | |
COMPMID-1188: Add support for activation in NEBatchNormalization.
Change-Id: I1e206574dac6433218db6e138adb7bf5f66a536d
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/145222
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'arm_compute/core/NEON/wrapper/intrinsics/max.h')
| -rw-r--r-- | arm_compute/core/NEON/wrapper/intrinsics/max.h | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/arm_compute/core/NEON/wrapper/intrinsics/max.h b/arm_compute/core/NEON/wrapper/intrinsics/max.h index 1a8e95de87..05ed051c62 100644 --- a/arm_compute/core/NEON/wrapper/intrinsics/max.h +++ b/arm_compute/core/NEON/wrapper/intrinsics/max.h @@ -43,6 +43,9 @@ VMAX_IMPL(int16_t, int16x4_t, vmax, s16) VMAX_IMPL(uint32_t, uint32x2_t, vmax, u32) VMAX_IMPL(int32_t, int32x2_t, vmax, s32) VMAX_IMPL(float, float32x2_t, vmax, f32) +#ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC +VMAX_IMPL(float16_t, float16x4_t, vmax, f16) +#endif // __ARM_FEATURE_FP16_VECTOR_ARITHMETIC VMAX_IMPL(uint8_t, uint8x16_t, vmaxq, u8) VMAX_IMPL(int8_t, int8x16_t, vmaxq, s8) @@ -51,6 +54,9 @@ VMAX_IMPL(int16_t, int16x8_t, vmaxq, s16) VMAX_IMPL(uint32_t, uint32x4_t, vmaxq, u32) VMAX_IMPL(int32_t, int32x4_t, vmaxq, s32) VMAX_IMPL(float, float32x4_t, vmaxq, f32) +#ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC +VMAX_IMPL(float16_t, float16x8_t, vmaxq, f16) +#endif // __ARM_FEATURE_FP16_VECTOR_ARITHMETIC #undef VMAX_IMPL } // namespace wrapper |