aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorSang-Hoon Park <sang-hoon.park@arm.com>2020-05-19 17:30:35 +0100
committerManuel Bottini <manuel.bottini@arm.com>2020-05-20 11:16:03 +0100
commit820b433eb8429ce33fd3f3246f7dd92b75856e0b (patch)
tree655b084c36883d3d97ff8b74f696059554665afe
parent9329175a4d29f753b7b77b0bfdbec5438e174df2 (diff)
downloadComputeLibrary-820b433eb8429ce33fd3f3246f7dd92b75856e0b.tar.gz
COMPMID-3497: Fix build LSTMParams<TensorInfo>
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>
-rw-r--r--arm_compute/core/utils/misc/InfoHelpers.h7
1 files changed, 7 insertions, 0 deletions
diff --git a/arm_compute/core/utils/misc/InfoHelpers.h b/arm_compute/core/utils/misc/InfoHelpers.h
index 6ecda7a0dd..c6ee7c9031 100644
--- a/arm_compute/core/utils/misc/InfoHelpers.h
+++ b/arm_compute/core/utils/misc/InfoHelpers.h
@@ -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