diff options
Diffstat (limited to 'arm_compute/runtime/common/LSTMParams.h')
| -rw-r--r-- | arm_compute/runtime/common/LSTMParams.h | 22 |
1 files changed, 14 insertions, 8 deletions
diff --git a/arm_compute/runtime/common/LSTMParams.h b/arm_compute/runtime/common/LSTMParams.h index ffb4ddd9d3..6800faf87f 100644 --- a/arm_compute/runtime/common/LSTMParams.h +++ b/arm_compute/runtime/common/LSTMParams.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2018-2020 Arm Limited. + * Copyright (c) 2018-2021 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -24,8 +24,6 @@ #ifndef ARM_COMPUTE_LSTMPARAMS_H #define ARM_COMPUTE_LSTMPARAMS_H -#include "arm_compute/core/IPyramid.h" -#include "arm_compute/core/PyramidInfo.h" #include "arm_compute/core/Types.h" #include "arm_compute/runtime/Tensor.h" @@ -81,7 +79,10 @@ public: * * @return Reference to this LSTMParams object */ - LSTMParams &set_cifg_params(const T *input_to_input_weights, const T *recurrent_to_input_weights, T *cell_to_input_weights, const T *input_gate_bias) + LSTMParams &set_cifg_params(const T *input_to_input_weights, + const T *recurrent_to_input_weights, + T *cell_to_input_weights, + const T *input_gate_bias) { _input_to_input_weights = input_to_input_weights; _recurrent_to_input_weights = recurrent_to_input_weights; @@ -127,8 +128,10 @@ public: * * @return Reference to this LSTMParams object */ - LSTMParams &set_layer_normalization_params(T *input_layer_norm_weights, T *forget_layer_norm_weights, - T *cell_layer_norm_weights, T *output_layer_norm_weights) + LSTMParams &set_layer_normalization_params(T *input_layer_norm_weights, + T *forget_layer_norm_weights, + T *cell_layer_norm_weights, + T *output_layer_norm_weights) { _input_layer_norm_weights = input_layer_norm_weights; _forget_layer_norm_weights = forget_layer_norm_weights; @@ -171,7 +174,10 @@ public: * * @return Reference to this LSTMParams object */ - LSTMParams &set_matmul_scale_params(float input_intermediate_scale, float forget_intermediate_scale, float cell_intermediate_scale, float output_intermediate_scale) + LSTMParams &set_matmul_scale_params(float input_intermediate_scale, + float forget_intermediate_scale, + float cell_intermediate_scale, + float output_intermediate_scale) { _input_intermediate_scale = input_intermediate_scale; _forget_intermediate_scale = forget_intermediate_scale; @@ -340,5 +346,5 @@ private: bool _has_cifg_opt; bool _use_layer_norm; }; -} +} // namespace arm_compute #endif /*ARM_COMPUTE_LSTMPARAMS_H */ |