Searched refs:output_shape (Results 1 – 3 of 3) sorted by relevance
| /cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_batch_matmul_s8/ |
| D | test_arm_batch_matmul_s8.c | 59 cmsis_nn_dims output_shape = {BATCH_MATMUL_1_S8_OUTPUT_BATCH, in batch_matmul_1_s8() local
69 int32_t buf_size = arm_fully_connected_s8_get_buffer_size(&output_shape); in batch_matmul_1_s8()
74 …m_params, &quant_params, &lhs_shape_nt, lhs_input, &rhs_shape_t, rhs_input, &output_shape, output); in batch_matmul_1_s8()
107 cmsis_nn_dims output_shape = {BATCH_MATMUL_2_S8_OUTPUT_BATCH, in batch_matmul_2_s8() local
117 int32_t buf_size = arm_fully_connected_s8_get_buffer_size(&output_shape); in batch_matmul_2_s8()
122 …_params, &quant_params, &lhs_shape_nt, lhs_input, &rhs_shape_nt, rhs_input, &output_shape, output); in batch_matmul_2_s8()
156 cmsis_nn_dims output_shape = {BATCH_MATMUL_3_S8_OUTPUT_BATCH, in batch_matmul_3_s8() local
166 int32_t buf_size = arm_fully_connected_s8_get_buffer_size(&output_shape); in batch_matmul_3_s8()
171 …mm_params, &quant_params, &lhs_shape_t, lhs_input, &rhs_shape_t, rhs_input, &output_shape, output); in batch_matmul_3_s8()
205 cmsis_nn_dims output_shape = {BATCH_MATMUL_4_S8_OUTPUT_BATCH, in batch_matmul_4_s8() local
[all …]
|
| /cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_batch_matmul_s16/ |
| D | test_arm_batch_matmul_s16.c | 59 cmsis_nn_dims output_shape = {BATCH_MATMUL_1_S16_OUTPUT_BATCH, in batch_matmul_1_s16() local
73 …m_params, &quant_params, &lhs_shape_nt, lhs_input, &rhs_shape_t, rhs_input, &output_shape, output); in batch_matmul_1_s16()
100 cmsis_nn_dims output_shape = {BATCH_MATMUL_2_S16_OUTPUT_BATCH, in batch_matmul_2_s16() local
114 …_params, &quant_params, &lhs_shape_nt, lhs_input, &rhs_shape_nt, rhs_input, &output_shape, output); in batch_matmul_2_s16()
142 cmsis_nn_dims output_shape = {BATCH_MATMUL_3_S16_OUTPUT_BATCH, in batch_matmul_3_s16() local
156 …mm_params, &quant_params, &lhs_shape_t, lhs_input, &rhs_shape_t, rhs_input, &output_shape, output); in batch_matmul_3_s16()
184 cmsis_nn_dims output_shape = {BATCH_MATMUL_4_S16_OUTPUT_BATCH, in batch_matmul_4_s16() local
198 …m_params, &quant_params, &lhs_shape_t, lhs_input, &rhs_shape_nt, rhs_input, &output_shape, output); in batch_matmul_4_s16()
225 cmsis_nn_dims output_shape = {BATCH_MATMUL_5_S16_OUTPUT_BATCH, in batch_matmul_5_s16() local
239 …_params, &quant_params, &lhs_shape_nt, lhs_input, &rhs_shape_nt, rhs_input, &output_shape, output); in batch_matmul_5_s16()
|
| /cmsis-nn-latest/Tests/UnitTest/ |
| D | model_extractor.py | 144 def shape_to_config(self, input_shape, filter_shape, output_shape, layer_name): argument
155 [self.y_output, self.x_output] = output_shape
163 if len(output_shape) == 2:
164 [self.y_output, self.x_output] = output_shape
166 [d, self.y_output, self.x_output, d1] = output_shape
217 output_shape = output_tensor['shape']
249 self.shape_to_config(input_shape, filter_shape, output_shape, builtin_name)
|
