27 #include "tests/datasets/LSTMLayerDataset.h"
32 #include "tests/validation/fixtures/LSTMLayerFixture.h"
125 ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU),
126 ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU),
127 ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU),
128 ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU),
129 ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU),
130 ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU),
131 ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU),
136 LSTMParams<ITensorInfo> lstm_params_info;
137 auto cell_bias_clone = cell_bias_info.clone();
138 lstm_params_info.set_peephole_params(cell_bias_clone.get(), cell_bias_clone.get())
139 .set_projection_params(&recurrent_weights_info, &projection_bias_info)
140 .set_cifg_params(&input_weights_info, &recurrent_weights_info, cell_bias_clone.get(), cell_bias_clone.get());
143 &input_weights_info.clone()->set_is_resizable(
false), &recurrent_weights_info.clone()->set_is_resizable(
false), &recurrent_weights_info.clone()->set_is_resizable(
false),
144 &recurrent_weights_info.clone()->set_is_resizable(
false), &cell_bias_info.clone()->set_is_resizable(
false), &cell_bias_info.clone()->set_is_resizable(
false),
145 &cell_bias_info.clone()->set_is_resizable(
false),
146 &
output_info.clone()->set_is_resizable(
false), &cell_state_info.clone()->set_is_resizable(
false),
147 &scratch_info.clone()->set_is_resizable(
false), &
output_info.clone()->set_is_resizable(
false), &cell_state_info.clone()->set_is_resizable(
false), &
output_info.clone()->set_is_resizable(
false),
153 template <
typename T>
154 using NELSTMLayerFixture = LSTMLayerValidationFixture<Tensor, Accessor, NELSTMLayer, LSTMParams<ITensor>, T>;
159 framework::dataset::
make("ProjectionOpt", {
true,
false })),
169 #ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC