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/traits.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/traits.h')
-rw-r--r-- | arm_compute/core/NEON/wrapper/traits.h | 4 |
1 files changed, 4 insertions, 0 deletions
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<uint64_t, 2>{ using type = uint64x2_t; using tag_ template <> struct neon_vector<int64_t, 2>{ using type = int64x2_t; using tag_type = vector_128_tag; }; template <> struct neon_vector<float_t, 2>{ using type = float32x2_t; using tag_type = vector_64_tag; }; template <> struct neon_vector<float_t, 4>{ using type = float32x4_t; using tag_type = vector_128_tag; }; +#ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC +template <> struct neon_vector<float16_t, 4>{ using type = float16x4_t; using tag_type = vector_64_tag; }; +template <> struct neon_vector<float16_t, 8>{ 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 */ |