| /cmsis-nn-latest/Tests/UnitTest/RefactoredTestGen/JsonTemplates/ |
| D | lstm_s16.json | 15 batch_size,
285 batch_size,
306 batch_size,
387 batch_size,
|
| D | lstm_s16_tm.json | 16 batch_size,
285 batch_size,
306 batch_size,
388 batch_size,
|
| D | fully_connected_null_bias.json | 15 batch_size,
56 batch_size,
|
| D | fully_connected.json | 15 batch_size,
75 batch_size,
|
| /cmsis-nn-latest/Source/NNSupportFunctions/ |
| D | arm_nn_lstm_calculate_gate_s16.c | 55 memset(output, 0, params->hidden_size * params->batch_size * sizeof(int16_t)); in arm_nn_lstm_calculate_gate_s16()
65 params->batch_size, in arm_nn_lstm_calculate_gate_s16()
79 params->batch_size, in arm_nn_lstm_calculate_gate_s16()
83 …arm_nn_activation_s16(output, output, params->hidden_size * params->batch_size, 0, gate->activatio… in arm_nn_lstm_calculate_gate_s16()
|
| D | arm_nn_lstm_calculate_gate_s8_s16.c | 62 memset(output, 0, params->hidden_size * params->batch_size * sizeof(int16_t)); in arm_nn_lstm_calculate_gate_s8_s16()
72 params->batch_size, in arm_nn_lstm_calculate_gate_s8_s16()
85 params->batch_size, in arm_nn_lstm_calculate_gate_s8_s16()
89 …arm_nn_activation_s16(output, output, params->hidden_size * params->batch_size, 0, gate->activatio… in arm_nn_lstm_calculate_gate_s8_s16()
|
| D | arm_nn_lstm_step_s16.c | 74 params->hidden_size * params->batch_size); in arm_nn_lstm_step_s16()
91 params->hidden_size * params->batch_size); in arm_nn_lstm_step_s16()
97 …cell_state, hidden_temp, params->hidden_size * params->batch_size, params->cell_scale_power + 12, … in arm_nn_lstm_step_s16()
105 params->batch_size, in arm_nn_lstm_step_s16()
|
| D | arm_nn_lstm_step_s8.c | 73 params->hidden_size * params->batch_size); in arm_nn_lstm_step_s8()
89 params->hidden_size * params->batch_size); in arm_nn_lstm_step_s8()
95 …cell_state, hidden_temp, params->hidden_size * params->batch_size, params->cell_scale_power + 12, … in arm_nn_lstm_step_s8()
103 params->batch_size, in arm_nn_lstm_step_s8()
|
| /cmsis-nn-latest/Source/LSTMFunctions/ |
| D | arm_lstm_unidirectional_s16.c | 56 memset(buffers->cell_state, 0, params->batch_size * params->hidden_size * sizeof(int16_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()
|
| D | arm_lstm_unidirectional_s8.c | 56 memset(buffers->cell_state, 0, params->batch_size * params->hidden_size * sizeof(int16_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()
|
| /cmsis-nn-latest/Source/BasicMathFunctions/ |
| D | arm_elementwise_mul_s16_s8.c | 55 const int32_t batch_size, in arm_elementwise_mul_s16_s8() argument
59 for (int i = 0; i < batch_size; i++) in arm_elementwise_mul_s16_s8()
|
| D | arm_elementwise_mul_s16_batch_offset.c | 56 const int32_t batch_size, in arm_elementwise_mul_s16_batch_offset() argument
62 for (int i = 0; i < batch_size; i++) in arm_elementwise_mul_s16_batch_offset()
|
| /cmsis-nn-latest/Source/PoolingFunctions/ |
| D | arm_max_pool_s16.c | 167 const int32_t batch_size = input_x * input_y * channel_in; in arm_max_pool_s16() local
218 src += batch_size; in arm_max_pool_s16()
|
| D | arm_max_pool_s8.c | 184 const int32_t batch_size = input_x * input_y * channel_in; in arm_max_pool_s8() local
234 src += batch_size; in arm_max_pool_s8()
|
| D | arm_avgpool_s8.c | 272 const int32_t batch_size = input_x * input_y * ch_src; in arm_avgpool_s8() local
327 src += batch_size; in arm_avgpool_s8()
|
| /cmsis-nn-latest/Include/ |
| D | arm_nn_types.h | 218 int32_t batch_size; member
|
| D | arm_nnsupportfunctions.h | 1900 const int32_t batch_size,
1926 const int32_t batch_size,
|
| /cmsis-nn-latest/Tests/UnitTest/ |
| D | pooling_settings.py | 90 model.add(keras.layers.InputLayer(input_shape=input_shape[1:], batch_size=self.batches))
|
| D | fully_connected_settings.py | 242 batch_size=self.batches))
|
| D | conv_settings.py | 336 model.add(keras.layers.InputLayer(input_shape=input_shape[1:], batch_size=self.batches))
|
| D | lstm_settings.py | 146 batch_size=self.batches,
|
| /cmsis-nn-latest/Tests/UnitTest/RefactoredTestGen/Lib/ |
| D | op_conv.py | 76 … model.add(keras.layers.InputLayer(input_shape=input_shape[1:], batch_size=params["batch_size"]))
|
| D | op_lstm.py | 49 batch_size=params["batch_size"],
|
