diff options
Diffstat (limited to 'arm_compute/runtime/common/LSTMParams.h')
| -rw-r--r-- | arm_compute/runtime/common/LSTMParams.h | 52 |
1 files changed, 29 insertions, 23 deletions
diff --git a/arm_compute/runtime/common/LSTMParams.h b/arm_compute/runtime/common/LSTMParams.h index 5e4a76afae..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, const 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; @@ -111,7 +112,7 @@ public: * * @return Reference to this LSTMParams object */ - LSTMParams &set_peephole_params(const T *cell_to_forget_weights, const T *cell_to_output_weights) + LSTMParams &set_peephole_params(T *cell_to_forget_weights, T *cell_to_output_weights) { _cell_to_forget_weights = cell_to_forget_weights; _cell_to_output_weights = cell_to_output_weights; @@ -127,8 +128,10 @@ public: * * @return Reference to this LSTMParams object */ - LSTMParams &set_layer_normalization_params(const T *input_layer_norm_weights, const T *forget_layer_norm_weights, - const T *cell_layer_norm_weights, const 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; @@ -204,7 +210,7 @@ public: return _recurrent_to_input_weights; } - const T *cell_to_input_weights() const + T *cell_to_input_weights() const { return _cell_to_input_weights; } @@ -214,12 +220,12 @@ public: return _input_gate_bias; } - const T *cell_to_forget_weights() const + T *cell_to_forget_weights() const { return _cell_to_forget_weights; } - const T *cell_to_output_weights() const + T *cell_to_output_weights() const { return _cell_to_output_weights; } @@ -234,22 +240,22 @@ public: return _projection_bias; } - const T *input_layer_norm_weights() const + T *input_layer_norm_weights() const { return _input_layer_norm_weights; } - const T *forget_layer_norm_weights() const + T *forget_layer_norm_weights() const { return _forget_layer_norm_weights; } - const T *cell_layer_norm_weights() const + T *cell_layer_norm_weights() const { return _cell_layer_norm_weights; } - const T *output_layer_norm_weights() const + T *output_layer_norm_weights() const { return _output_layer_norm_weights; } @@ -317,16 +323,16 @@ public: private: const T *_input_to_input_weights; const T *_recurrent_to_input_weights; - const T *_cell_to_input_weights; + T *_cell_to_input_weights; const T *_input_gate_bias; - const T *_cell_to_forget_weights; - const T *_cell_to_output_weights; + T *_cell_to_forget_weights; + T *_cell_to_output_weights; const T *_projection_weights; const T *_projection_bias; - const T *_input_layer_norm_weights; - const T *_forget_layer_norm_weights; - const T *_cell_layer_norm_weights; - const T *_output_layer_norm_weights; + T *_input_layer_norm_weights; + T *_forget_layer_norm_weights; + T *_cell_layer_norm_weights; + T *_output_layer_norm_weights; float _cell_clip; float _projection_clip; float _input_intermediate_scale; @@ -340,5 +346,5 @@ private: bool _has_cifg_opt; bool _use_layer_norm; }; -} +} // namespace arm_compute #endif /*ARM_COMPUTE_LSTMPARAMS_H */ |