aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/core/NEON/kernels/NEQLSTMLayerNormalizationKernel.h
diff options
context:
space:
mode:
Diffstat (limited to 'arm_compute/core/NEON/kernels/NEQLSTMLayerNormalizationKernel.h')
-rw-r--r--arm_compute/core/NEON/kernels/NEQLSTMLayerNormalizationKernel.h2
1 files changed, 2 insertions, 0 deletions
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 */