From 9230e2789e421021804bc3a65cf47df4749b0765 Mon Sep 17 00:00:00 2001 From: Sang-Hoon Park Date: Sat, 18 Apr 2020 00:46:34 +0100 Subject: COMPMID-3241: Add Layer Normalization to NEQLSTMLayer - Add output quantization calculation to Layer Normalization - Add members for Layer Normalization to NEQLSTMLayer - Add configure/validate/run of Layer Normalization to NEQLSTMLayer Change-Id: I278c8e0edbb21212f3afa4d4a336df0f1a4c1bfb Signed-off-by: Sang-Hoon Park Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3059 Tested-by: Arm Jenkins Reviewed-by: Michele Di Giorgio Comments-Addressed: Arm Jenkins --- arm_compute/core/NEON/kernels/NEQLSTMLayerNormalizationKernel.h | 2 ++ 1 file changed, 2 insertions(+) (limited to 'arm_compute/core/NEON/kernels/NEQLSTMLayerNormalizationKernel.h') diff --git a/arm_compute/core/NEON/kernels/NEQLSTMLayerNormalizationKernel.h b/arm_compute/core/NEON/kernels/NEQLSTMLayerNormalizationKernel.h index 631de66cc2..f5e8da7feb 100644 --- a/arm_compute/core/NEON/kernels/NEQLSTMLayerNormalizationKernel.h +++ b/arm_compute/core/NEON/kernels/NEQLSTMLayerNormalizationKernel.h @@ -130,6 +130,8 @@ private: const int16_t *weight_ptr, const int32_t *bias_ptr, int32_t mean, int32_t inv_std_mul, int32_t inv_std_shift); + /** Function to compute output quantization information */ + QuantizationInfo compute_output_qinfo(); }; } // namespace arm_compute #endif /* ARM_COMPUTE_NEQLSTMLAYERNORMALIZATIONKERNEL_H */ -- cgit v1.2.1