diff options
author | Pablo Tello <pablo.tello@arm.com> | 2017-07-03 16:25:09 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-09-17 14:15:39 +0100 |
commit | df24618b53cffed1c574e11e9fd4ba7740f8c009 (patch) | |
tree | 1f1145bca27c5dd0ca63538c2e8cdadd2b0a03cf /arm_compute/core/NEON/NEMath.h | |
parent | d1b0ecc206e3858327503888c4a46842ec1808e9 (diff) | |
download | ComputeLibrary-df24618b53cffed1c574e11e9fd4ba7740f8c009.tar.gz |
COMPMID-421: Added FP16 suppot to NENormalizationLayer and NEPixelWiseMultiplication.
Change-Id: If174f8071502fc5cc94b27cd44a9b1d5e451a9e2
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/79553
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Diffstat (limited to 'arm_compute/core/NEON/NEMath.h')
-rw-r--r-- | arm_compute/core/NEON/NEMath.h | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/arm_compute/core/NEON/NEMath.h b/arm_compute/core/NEON/NEMath.h index bb8a330c1e..8dd9d609e7 100644 --- a/arm_compute/core/NEON/NEMath.h +++ b/arm_compute/core/NEON/NEMath.h @@ -91,6 +91,26 @@ float32x4_t vtanhq_f32(float32x4_t val); * @return The calculated power. */ float32x4_t vpowq_f32(float32x4_t val, float32x4_t n); + +#ifdef ARM_COMPUTE_ENABLE_FP16 +/** Calculate exponential + * + * @param[in] x Input vector value in F16 format. + * + * @return The calculated exponent. + */ +float16x8_t vexpq_f16(float16x8_t x); +/** Calculate n power of a number. + * + * pow(x,n) = e^(n*log(x)) + * + * @param[in] val Input vector value in F16 format. + * @param[in] n Powers to raise the input to. + * + * @return The calculated power. + */ +float16x8_t vpowq_f16(float16x8_t val, float16x8_t n); +#endif /* ARM_COMPUTE_ENABLE_FP16 */ } #include "arm_compute/core/NEON/NEMath.inl" #endif /* __ARM_COMPUTE_NEMATH_H__ */ |