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