aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/runtime/common/LSTMParams.h
diff options
context:
space:
mode:
authorSang-Hoon Park <sang-hoon.park@arm.com>2020-04-18 01:40:57 +0100
committerSang-Hoon Park <sang-hoon.park@arm.com>2020-04-20 14:53:23 +0000
commit30b46a660629ccd5bf715f63b90d6d7655416e2c (patch)
tree9d30cb7d4950d4f21d0033c9d88c842c587f5a51 /arm_compute/runtime/common/LSTMParams.h
parenteb65f6da695ac0d3e495817145cceb1c4de4f048 (diff)
downloadComputeLibrary-30b46a660629ccd5bf715f63b90d6d7655416e2c.tar.gz
COMPMID-3241: Fix hidden scale in NEQLSTMLayer
- Fix wrong data types in LSTMParams - Add logic to ignore epsilon for quantization multiplier computation - Ignore epsilon for hidden gate scale computation Change-Id: Ia0b2f523b1c2ad325f3523439a8eea051d81958c Signed-off-by: Sang-Hoon Park <sang-hoon.park@arm.com> Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3058 Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com> Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/runtime/common/LSTMParams.h')
-rw-r--r--arm_compute/runtime/common/LSTMParams.h8
1 files changed, 4 insertions, 4 deletions
diff --git a/arm_compute/runtime/common/LSTMParams.h b/arm_compute/runtime/common/LSTMParams.h
index e21ddd7af1..5e4a76afae 100644
--- a/arm_compute/runtime/common/LSTMParams.h
+++ b/arm_compute/runtime/common/LSTMParams.h
@@ -58,8 +58,8 @@ public:
_forget_intermediate_scale(0.0f),
_cell_intermediate_scale(0.0f),
_output_intermediate_scale(0.0f),
- _hidden_state_zero(0.0f),
- _hidden_state_scale(0),
+ _hidden_state_zero(0),
+ _hidden_state_scale(0.0f),
_has_peephole_opt(false),
_has_projection(false),
_has_cifg_opt(true),
@@ -333,8 +333,8 @@ private:
float _forget_intermediate_scale;
float _cell_intermediate_scale;
float _output_intermediate_scale;
- float _hidden_state_zero;
- int32_t _hidden_state_scale;
+ int32_t _hidden_state_zero;
+ float _hidden_state_scale;
bool _has_peephole_opt;
bool _has_projection;
bool _has_cifg_opt;