diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2017-08-31 14:21:36 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:35:24 +0000 |
| commit | cdf51455df8835e9e3bfd3e31ed389146af9a573 (patch) | |
| tree | 31b0bf9302decbf8b1063f46373e3d26a9ca1409 /arm_compute/core/NEON/NEMath.h | |
| parent | 29088d517a2a9f249fe5cc851e0c97de3d4cc917 (diff) | |
| download | ComputeLibrary-cdf51455df8835e9e3bfd3e31ed389146af9a573.tar.gz | |
COMPMID-515: L2 Pooling for FP32/FP16 in CL.
Change-Id: I43641fa672f5905ca62edd1f63fc93e0cf7ea382
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/85963
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Diffstat (limited to 'arm_compute/core/NEON/NEMath.h')
| -rw-r--r-- | arm_compute/core/NEON/NEMath.h | 41 |
1 files changed, 37 insertions, 4 deletions
diff --git a/arm_compute/core/NEON/NEMath.h b/arm_compute/core/NEON/NEMath.h index 523649c65d..ba65926802 100644 --- a/arm_compute/core/NEON/NEMath.h +++ b/arm_compute/core/NEON/NEMath.h @@ -42,17 +42,23 @@ float32x4_t vfloorq_f32(float32x4_t val); * * @return The calculated inverse square root. */ -float32x4_t vinvsqrtq_f32(float32x4_t x); +float32x2_t vinvsqrt_f32(float32x2_t x); -#ifdef ARM_COMPUTE_ENABLE_FP16 /** Calculate inverse square root. * * @param[in] x Input value. * * @return The calculated inverse square root. */ -float16x8_t vinvsqrtq_f16(float16x8_t x); -#endif /* ARM_COMPUTE_ENABLE_FP16 */ +float32x4_t vinvsqrtq_f32(float32x4_t x); + +/** Calculate reciprocal. + * + * @param[in] x Input value. + * + * @return The calculated reciprocal. + */ +float32x2_t vinv_f32(float32x2_t x); /** Calculate reciprocal. * @@ -122,6 +128,31 @@ float32x4_t vpowq_f32(float32x4_t val, float32x4_t n); * @return The calculated Hyperbolic Tangent. */ float16x8_t vtanhq_f16(float16x8_t val); + +/** Calculate reciprocal. + * + * @param[in] x Input value. + * + * @return The calculated reciprocal. + */ +float16x4_t vinv_f16(float16x4_t x); + +/** Calculate reciprocal. + * + * @param[in] x Input value. + * + * @return The calculated reciprocal. + */ +float16x8_t vinvq_f16(float16x8_t x); + +/** Calculate inverse square root. + * + * @param[in] x Input value. + * + * @return The calculated inverse square root. + */ +float16x4_t vinvsqrt_f16(float16x4_t x); + /** Calculate inverse square root. * * @param[in] x Input value. @@ -129,6 +160,7 @@ float16x8_t vtanhq_f16(float16x8_t val); * @return The calculated inverse square root. */ float16x8_t vinvsqrtq_f16(float16x8_t x); + /** Calculate exponential * * @param[in] x Input vector value in F16 format. @@ -136,6 +168,7 @@ float16x8_t vinvsqrtq_f16(float16x8_t x); * @return The calculated exponent. */ float16x8_t vexpq_f16(float16x8_t x); + /** Calculate n power of a number. * * pow(x,n) = e^(n*log(x)) |