diff options
| author | Sang-Hoon Park <sang-hoon.park@arm.com> | 2020-10-20 15:33:31 +0100 |
|---|---|---|
| committer | TeresaARM <teresa.charlinreyes@arm.com> | 2020-10-20 21:40:38 +0000 |
| commit | 9f8937568c65863e2bc080a1f586068153289b83 (patch) | |
| tree | eec39c91785a7c0a442495bdb95ad40ce81c56c8 /src/runtime/CL/functions/CLQLSTMLayer.cpp | |
| parent | 9bf1ba117b39c87f5de6e3766970613cb4c434a2 (diff) | |
| download | ComputeLibrary-9f8937568c65863e2bc080a1f586068153289b83.tar.gz | |
COMPMID-3876: Fix wrong validation logic in QLSTM
Change-Id: Ie0ea5b009bead53817e76fb06213b332e7712c9c
Signed-off-by: Sang-Hoon Park <sang-hoon.park@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/4215
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/runtime/CL/functions/CLQLSTMLayer.cpp')
| -rw-r--r-- | src/runtime/CL/functions/CLQLSTMLayer.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/runtime/CL/functions/CLQLSTMLayer.cpp b/src/runtime/CL/functions/CLQLSTMLayer.cpp index c493471667..2d21d210e4 100644 --- a/src/runtime/CL/functions/CLQLSTMLayer.cpp +++ b/src/runtime/CL/functions/CLQLSTMLayer.cpp @@ -834,7 +834,8 @@ Status CLQLSTMLayer::validate(const ITensorInfo *input, ARM_COMPUTE_RETURN_ON_ERROR(CLPixelWiseMultiplication::validate(&output_gate_info, &input_gate_info, &hidden_mul_res, 1.f, ConvertPolicy::SATURATE, RoundingPolicy::TO_ZERO)); const float hidden_state_scale = std::pow(2, -15) / lstm_params.hidden_state_scale() * std::pow(2, -15); ARM_COMPUTE_RETURN_ON_ERROR(quantization::calculate_quantized_multiplier(hidden_state_scale, &gemmlowp_info.gemmlowp_multiplier, &gemmlowp_info.gemmlowp_shift, /* ignore_epsilon */ true)); - gemmlowp_info.gemmlowp_offset = lstm_params.hidden_state_zero(); + gemmlowp_info.gemmlowp_offset = lstm_params.hidden_state_zero(); + gemmlowp_info.output_data_type = hidden_out_info.data_type(); ARM_COMPUTE_RETURN_ON_ERROR(CLGEMMLowpOutputStage::validate(&hidden_mul_res, nullptr, &hidden_out_info, gemmlowp_info)); const bool projection_tensor_copy_required = num_units != output_size; |