Searched refs:n_batch (Results 1 – 4 of 4) sorted by relevance
| /cmsis-nn-3.6.0-3.5.0/Source/NNSupportFunctions/ |
| D | arm_nn_lstm_step_s8_s16.c | 54 const int n_batch, in arm_nn_lstm_step_s8_s16() argument
63 const int32_t n_block = n_batch * n_cell; in arm_nn_lstm_step_s8_s16()
74 n_batch, in arm_nn_lstm_step_s8_s16()
90 n_batch, in arm_nn_lstm_step_s8_s16()
106 n_batch, in arm_nn_lstm_step_s8_s16()
130 n_batch, in arm_nn_lstm_step_s8_s16()
138 arm_nn_lstm_update_output_s8_s16(n_batch, in arm_nn_lstm_step_s8_s16()
148 arm_memcpy_s8(output, output_state, n_batch * n_output * sizeof(int8_t)); in arm_nn_lstm_step_s8_s16()
|
| D | arm_nn_lstm_calculate_gate_s8_s16.c | 63 const int32_t n_batch, in arm_nn_lstm_calculate_gate_s8_s16() argument
70 const int32_t n_block = n_batch * n_cell; in arm_nn_lstm_calculate_gate_s8_s16()
82 n_batch); in arm_nn_lstm_calculate_gate_s8_s16()
93 n_batch); in arm_nn_lstm_calculate_gate_s8_s16()
|
| D | arm_nn_lstm_update_output_s8_s16.c | 47 void arm_nn_lstm_update_output_s8_s16(const int n_batch, in arm_nn_lstm_update_output_s8_s16() argument
57 const int32_t size = n_batch * n_cell; in arm_nn_lstm_update_output_s8_s16()
|
| /cmsis-nn-3.6.0-3.5.0/Include/ |
| D | arm_nnsupportfunctions.h | 1190 const int n_batch,
1225 const int32_t n_batch,
1261 void arm_nn_lstm_update_output_s8_s16(const int n_batch,
|
