Lines Matching refs:params
51 const cmsis_nn_lstm_params *params, in arm_lstm_unidirectional_s16() argument
56 memset(buffers->cell_state, 0, params->batch_size * params->hidden_size * sizeof(int16_t)); in arm_lstm_unidirectional_s16()
57 if (params->time_major) in arm_lstm_unidirectional_s16()
60 for (int t = 0; t < params->time_steps; t++) in arm_lstm_unidirectional_s16()
62 const int16_t *data_in = input + (t * params->batch_size * params->input_size); in arm_lstm_unidirectional_s16()
63 int16_t *hidden_out = output + (t * params->batch_size * params->hidden_size); in arm_lstm_unidirectional_s16()
64 …_cmsis_nn_status status = arm_nn_lstm_step_s16(data_in, hidden_in, hidden_out, params, buffers, 1); in arm_lstm_unidirectional_s16()
76 for (int t = 0; t < params->time_steps; t++) in arm_lstm_unidirectional_s16()
78 const int16_t *data_in = input + (t * params->input_size); in arm_lstm_unidirectional_s16()
79 int16_t *hidden_out = output + (t * params->hidden_size); in arm_lstm_unidirectional_s16()
81 … arm_nn_lstm_step_s16(data_in, hidden_in, hidden_out, params, buffers, params->time_steps); in arm_lstm_unidirectional_s16()