commit 8fda1cb6f4142133fff045a6f9c18778757c316c
author Pablo Tello <pablo.tello@arm.com> Wed Jul 05 15:20:38 2017 +0100
committer Anthony Barbier <anthony.barbier@arm.com> Mon Sep 17 14:16:42 2018 +0100
tree 3f0ad562b24cc3c76e8a745cb59cd584b664ec57
parent 8df3fafde3dcf131def3471db8e8b1a1c34b354b

COMPMID-421: Added FP16 support in BatchNormalizationLayer.

Change-Id: I7142e0e8466ef79e016ae56d285e8e9291573e52
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/79814
Reviewed-by: Moritz Pflanzer <moritz.pflanzer@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>

arm_compute/core/NEON/NEMath.h

diff --git a/arm_compute/core/NEON/NEMath.h b/arm_compute/core/NEON/NEMath.h
index b467a60..39f0c3b 100644
--- a/arm_compute/core/NEON/NEMath.h
+++ b/arm_compute/core/NEON/NEMath.h
@@ -36,6 +36,16 @@
  */
 float32x4_t vinvsqrtq_f32(float32x4_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 */
+
 /** Calculate reciprocal.
  *
  * @param[in] x Input value.