diff options
author | Sheri Zhang <sheri.zhang@arm.com> | 2020-04-21 13:10:24 +0100 |
---|---|---|
committer | Sheri Zhang <sheri.zhang@arm.com> | 2020-04-26 21:31:26 +0000 |
commit | 3a35398ed6cc5d9c0f45f33dabb2bfbb017bcf60 (patch) | |
tree | aaa4a8949e288157ab92404d1745214529c0c69b /src/core/NEON/kernels/NEQLSTMLayerNormalizationKernel.cpp | |
parent | 31b49caa2ca9308a5ba62a598afc9d1982b4af18 (diff) | |
download | ComputeLibrary-3a35398ed6cc5d9c0f45f33dabb2bfbb017bcf60.tar.gz |
COMPMID-3240: Add support for layer normalization to CLQLSTMLayer
Signed-off-by: Sheri Zhang <sheri.zhang@arm.com>
Change-Id: I45359a4ddb46c059097a2d77c008f802e8f4c143
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3065
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Reviewed-by: Sang-Hoon Park <sang-hoon.park@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core/NEON/kernels/NEQLSTMLayerNormalizationKernel.cpp')
-rw-r--r-- | src/core/NEON/kernels/NEQLSTMLayerNormalizationKernel.cpp | 5 |
1 files changed, 1 insertions, 4 deletions
diff --git a/src/core/NEON/kernels/NEQLSTMLayerNormalizationKernel.cpp b/src/core/NEON/kernels/NEQLSTMLayerNormalizationKernel.cpp index e966c6bdba..29ffee867b 100644 --- a/src/core/NEON/kernels/NEQLSTMLayerNormalizationKernel.cpp +++ b/src/core/NEON/kernels/NEQLSTMLayerNormalizationKernel.cpp @@ -172,10 +172,7 @@ void NEQLSTMLayerNormalizationKernel::run(const Window &window, const ThreadInfo inline QuantizationInfo NEQLSTMLayerNormalizationKernel::compute_output_qinfo() { - const UniformQuantizationInfo iq_info = _input->info()->quantization_info().uniform(); - const UniformQuantizationInfo wq_info = _weight->info()->quantization_info().uniform(); - const float output_scale = (wq_info.scale * iq_info.scale) * 1024; - return QuantizationInfo(output_scale); + return QuantizationInfo(1.f / 4096); } inline std::pair<int64_t, int64_t> NEQLSTMLayerNormalizationKernel::sum_qsymm16(const int16_t *input_ptr) |