From aaba4c626bcc6365e0108130633ce43fafe9da45 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Wed, 22 Aug 2018 16:20:21 +0100 Subject: COMPMID-1188: Add support for activation in NEBatchNormalization. Change-Id: I1e206574dac6433218db6e138adb7bf5f66a536d Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/145222 Tested-by: Jenkins Reviewed-by: Anthony Barbier --- arm_compute/core/NEON/wrapper/traits.h | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'arm_compute/core/NEON/wrapper/traits.h') diff --git a/arm_compute/core/NEON/wrapper/traits.h b/arm_compute/core/NEON/wrapper/traits.h index 495ddbb1af..5cd6086c0c 100644 --- a/arm_compute/core/NEON/wrapper/traits.h +++ b/arm_compute/core/NEON/wrapper/traits.h @@ -62,6 +62,10 @@ template <> struct neon_vector{ using type = uint64x2_t; using tag_ template <> struct neon_vector{ using type = int64x2_t; using tag_type = vector_128_tag; }; template <> struct neon_vector{ using type = float32x2_t; using tag_type = vector_64_tag; }; template <> struct neon_vector{ using type = float32x4_t; using tag_type = vector_128_tag; }; +#ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC +template <> struct neon_vector{ using type = float16x4_t; using tag_type = vector_64_tag; }; +template <> struct neon_vector{ using type = float16x8_t; using tag_type = vector_128_tag; }; +#endif // __ARM_FEATURE_FP16_VECTOR_ARITHMETIC #endif /* DOXYGEN_SKIP_THIS */ /** Helper type template to get the type of a neon vector */ -- cgit v1.2.1