Searched refs:batch_offset (Results 1 – 9 of 9) sorted by relevance
| /cmsis-nn-3.7.0/Source/NNSupportFunctions/ |
| D | arm_nn_lstm_step_s16.c | 51 const int32_t batch_offset) in arm_nn_lstm_step_s16() argument
61 …tm_calculate_gate_s16(data_in, hidden_in, ¶ms->forget_gate, params, forget_gate, batch_offset); in arm_nn_lstm_step_s16()
76 …lstm_calculate_gate_s16(data_in, hidden_in, ¶ms->input_gate, params, input_gate, batch_offset); in arm_nn_lstm_step_s16()
78 …n_lstm_calculate_gate_s16(data_in, hidden_in, ¶ms->cell_gate, params, cell_gate, batch_offset); in arm_nn_lstm_step_s16()
93 …tm_calculate_gate_s16(data_in, hidden_in, ¶ms->output_gate, params, output_gate, batch_offset); in arm_nn_lstm_step_s16()
106 batch_offset); in arm_nn_lstm_step_s16()
|
| D | arm_nn_lstm_step_s8.c | 50 const int32_t batch_offset) in arm_nn_lstm_step_s8() argument
60 …calculate_gate_s8_s16(data_in, hidden_in, ¶ms->forget_gate, params, forget_gate, batch_offset); in arm_nn_lstm_step_s8()
75 …m_calculate_gate_s8_s16(data_in, hidden_in, ¶ms->input_gate, params, input_gate, batch_offset); in arm_nn_lstm_step_s8()
76 …stm_calculate_gate_s8_s16(data_in, hidden_in, ¶ms->cell_gate, params, cell_gate, batch_offset); in arm_nn_lstm_step_s8()
91 …calculate_gate_s8_s16(data_in, hidden_in, ¶ms->output_gate, params, output_gate, batch_offset); in arm_nn_lstm_step_s8()
104 batch_offset); in arm_nn_lstm_step_s8()
|
| D | arm_nn_lstm_calculate_gate_s16.c | 52 const int32_t batch_offset) in arm_nn_lstm_calculate_gate_s16() argument
66 batch_offset); in arm_nn_lstm_calculate_gate_s16()
80 batch_offset); in arm_nn_lstm_calculate_gate_s16()
|
| D | arm_nn_lstm_calculate_gate_s8_s16.c | 59 const int32_t batch_offset) in arm_nn_lstm_calculate_gate_s8_s16() argument
73 batch_offset); in arm_nn_lstm_calculate_gate_s8_s16()
86 batch_offset); in arm_nn_lstm_calculate_gate_s8_s16()
|
| D | arm_nn_vec_mat_mul_result_acc_s16.c | 57 const int32_t batch_offset) in arm_nn_vec_mat_mul_result_acc_s16() argument
354 lhs += rhs_cols * batch_offset; in arm_nn_vec_mat_mul_result_acc_s16()
|
| D | arm_nn_vec_mat_mul_result_acc_s8_s16.c | 52 const int32_t batch_offset) in arm_nn_vec_mat_mul_result_acc_s8_s16() argument
343 lhs += rhs_cols * batch_offset; in arm_nn_vec_mat_mul_result_acc_s8_s16()
|
| /cmsis-nn-3.7.0/Source/BasicMathFunctions/ |
| D | arm_elementwise_mul_s16_s8.c | 56 const int32_t batch_offset) in arm_elementwise_mul_s16_s8() argument
130 output += (batch_offset - 1) * block_size; in arm_elementwise_mul_s16_s8()
|
| D | arm_elementwise_mul_s16_batch_offset.c | 57 const int32_t batch_offset) in arm_elementwise_mul_s16_batch_offset() argument
159 output += (batch_offset - 1) * block_size; in arm_elementwise_mul_s16_batch_offset()
|
| /cmsis-nn-3.7.0/Include/ |
| D | arm_nnsupportfunctions.h | 1749 const int32_t batch_offset);
1773 const int32_t batch_offset);
1794 const int32_t batch_offset);
1815 const int32_t batch_offset);
1845 const int32_t batch_offset);
1875 const int32_t batch_offset);
1901 const int32_t batch_offset);
1927 const int32_t batch_offset);
|
