COMPMID-3497: Fix build LSTMParams<TensorInfo>
authorSang-Hoon Park <sang-hoon.park@arm.com>
Tue, 19 May 2020 16:30:35 +0000 (17:30 +0100)
committerManuel Bottini <manuel.bottini@arm.com>
Wed, 20 May 2020 10:16:03 +0000 (11:16 +0100)
Copy of missing parameters is added to the build function.

Change-Id: I06d665729a61f2f01dcfab21c6e6d55359fc81cb
Signed-off-by: Sang-Hoon Park <sang-hoon.park@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3226
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Reviewed-by: James Conroy <james.conroy@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
arm_compute/core/utils/misc/InfoHelpers.h

index 6ecda7a0dd299ce66b024957d987d085abffef3b..c6ee7c9031ad7e96da1ab37edaedc536d284e9cc 100644 (file)
@@ -107,6 +107,13 @@ inline void build_lstm_params_tensor_info(const LSTMParams<T>     &lstm_params,
 
         lstm_params_info->set_layer_normalization_params(input_info, forget_info, cell_info, output_info);
     }
+
+    lstm_params_info->set_matmul_scale_params(lstm_params.input_intermediate_scale(),
+                                              lstm_params.forget_intermediate_scale(),
+                                              lstm_params.cell_intermediate_scale(),
+                                              lstm_params.output_intermediate_scale());
+
+    lstm_params_info->set_hidden_state_params(lstm_params.hidden_state_zero(), lstm_params.hidden_state_scale());
 }
 } // namespace info_helpers
 } // namespace utils