/cmsis-nn-latest/Source/NNSupportFunctions/ |
D | CMakeLists.txt | 19 file(GLOB SRC_S4 "./*_s4*.c") 20 file(GLOB SRC_S8 "./*_s8*.c") 21 file(GLOB SRC_S16 "./*_s16*.c") 22 target_sources(cmsis-nn PRIVATE ${SRC_S4} ${SRC_S8} ${SRC_S16} arm_nntables.c 23 arm_q7_to_q15_with_offset.c 24 arm_nn_mat_mult_nt_t_s8_s32.c 25 arm_s8_to_s16_unordered_with_offset.c)
|
/cmsis-nn-latest/Source/ConvolutionFunctions/ |
D | arm_depthwise_conv_get_buffer_sizes_s8.c | 51 return (input_dims->c * filter_dims->w * filter_dims->h) * sizeof(int16_t); in arm_depthwise_conv_s8_opt_get_buffer_size_dsp() 74 if (input_dims->c == output_dims->c && input_dims->n == 1 && dw_conv_params->dilation.w == 1 && in arm_depthwise_conv_wrapper_s8_get_buffer_size() 97 if (input_dims->c == output_dims->c && input_dims->n == 1 && dw_conv_params->dilation.w == 1 && in arm_depthwise_conv_wrapper_s8_get_buffer_size_dsp() 118 if (input_dims->c == output_dims->c && input_dims->n == 1 && dw_conv_params->dilation.w == 1 && in arm_depthwise_conv_wrapper_s8_get_buffer_size_mve()
|
D | arm_convolve_wrapper_s8.c | 64 (input_dims->c == filter_dims->c)) in arm_convolve_wrapper_s8() 96 … ((conv_params->stride.w * input_dims->c) % 4 == 0) && (input_dims->c == filter_dims->c)) in arm_convolve_wrapper_s8()
|
D | arm_depthwise_conv_get_buffer_sizes_s4.c | 59 if (input_dims->c == output_dims->c && input_dims->n == 1 && dw_conv_params->dilation.w == 1 && in arm_depthwise_conv_wrapper_s4_get_buffer_size() 83 if (input_dims->c == output_dims->c && input_dims->n == 1 && dw_conv_params->dilation.w == 1 && in arm_depthwise_conv_wrapper_s4_get_buffer_size_mve()
|
D | arm_convolve_get_buffer_sizes_s8.c | 47 int32_t col_length = input_dims->c * filter_dims->w * filter_dims->h; in arm_convolve_s8_get_buffer_size_mve() 77 const int32_t pad_size_left = pad_x * input_dims->c; in arm_convolve_1_x_n_s8_get_buffer_size_mve() 78 const int32_t pad_size_right = asym_pad ? right_pad_num * input_dims->c : pad_size_left; in arm_convolve_1_x_n_s8_get_buffer_size_mve() 79 const int32_t num_elem_left = kernel_x * input_dims->c; in arm_convolve_1_x_n_s8_get_buffer_size_mve() 80 const int32_t num_elem_right = num_elem_left - input_dims->c; in arm_convolve_1_x_n_s8_get_buffer_size_mve() 91 const int32_t rhs_cols = filter_dims->w * filter_dims->h * input_dims->c; in arm_convolve_s8_get_buffer_size() 148 (conv_params->stride.w * input_dims->c % 4 == 0)) in arm_convolve_wrapper_s8_get_buffer_size() 178 (conv_params->stride.w * input_dims->c % 4 == 0)) in arm_convolve_wrapper_s8_get_buffer_size_mve()
|
D | arm_convolve_wrapper_s4.c | 95 … ((conv_params->stride.w * input_dims->c) % 4 == 0) && (input_dims->c == filter_dims->c)) in arm_convolve_wrapper_s4()
|
D | CMakeLists.txt | 19 file(GLOB SRC_S4 "./*_s4*.c") 20 file(GLOB SRC_S8 "./*_s8*.c") 21 file(GLOB SRC_S16 "./*_s16*.c")
|
/cmsis-nn-latest/Source/SoftmaxFunctions/ |
D | CMakeLists.txt | 2 # Copyright (c) 2019-2022 Arm Limited. 19 file(GLOB SRC "./*_s8.c") 20 target_sources(cmsis-nn PRIVATE ${SRC} arm_softmax_s8_s16.c 21 arm_softmax_s16.c 22 arm_nn_softmax_common_s8.c)
|
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_ds_cnn_s_s8/ |
D | test_arm_ds_cnn_s_s8.c | 59 input_dims.c = CONV_2D_1_IN_CH; in ds_cnn_s_s8_get_buffer_size() 63 filter_dims.c = CONV_2D_1_IN_CH; in ds_cnn_s_s8_get_buffer_size() 68 output_dims.c = CONV_2D_1_OUT_CH; in ds_cnn_s_s8_get_buffer_size() 94 input_dims.c = DEPTHWISE_CONV_2D_2_OUT_CH; in ds_cnn_s_s8_get_buffer_size() 98 output_dims.c = DEPTHWISE_CONV_2D_2_OUT_CH; in ds_cnn_s_s8_get_buffer_size() 151 in_out_dim_0.c = CONV_2D_1_IN_CH; in ds_cnn_s_s8_inference() 155 conv_filter_dims.c = CONV_2D_1_IN_CH; in ds_cnn_s_s8_inference() 160 in_out_dim_1.c = CONV_2D_1_OUT_CH; in ds_cnn_s_s8_inference() 161 bias_dims.c = CONV_2D_1_OUT_CH; in ds_cnn_s_s8_inference() 201 in_out_dim_0.c = DEPTHWISE_CONV_2D_2_OUT_CH; in ds_cnn_s_s8_inference() [all …]
|
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_convolve_1x1_s4_fast/ |
D | test_arm_convolve_1x1_s4_fast.c | 51 input_dims.c = KERNEL1X1_INT4_IN_CH; in kernel1x1_arm_convolve_1x1_s4_fast() 55 filter_dims.c = KERNEL1X1_INT4_IN_CH; in kernel1x1_arm_convolve_1x1_s4_fast() 59 output_dims.c = KERNEL1X1_INT4_OUT_CH; in kernel1x1_arm_convolve_1x1_s4_fast() 71 bias_dims.c = output_dims.c; in kernel1x1_arm_convolve_1x1_s4_fast() 125 input_dims.c = KERNEL1X1_INT4_2_IN_CH; in kernel1x1_2_arm_convolve_1x1_s4_fast() 129 filter_dims.c = KERNEL1X1_INT4_2_IN_CH; in kernel1x1_2_arm_convolve_1x1_s4_fast() 133 output_dims.c = KERNEL1X1_INT4_2_OUT_CH; in kernel1x1_2_arm_convolve_1x1_s4_fast() 145 bias_dims.c = output_dims.c; in kernel1x1_2_arm_convolve_1x1_s4_fast() 198 input_dims.c = KERNEL1X1_INT4_3_IN_CH; in kernel1x1_3_arm_convolve_1x1_s4_fast() 202 filter_dims.c = KERNEL1X1_INT4_3_IN_CH; in kernel1x1_3_arm_convolve_1x1_s4_fast() [all …]
|
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_convolve_1x1_s8_fast/ |
D | test_arm_convolve_1x1_s8_fast.c | 49 input_dims.c = KERNEL1X1_IN_CH; in kernel1x1_arm_convolve_1x1_s8_fast() 53 filter_dims.c = KERNEL1X1_IN_CH; in kernel1x1_arm_convolve_1x1_s8_fast() 57 output_dims.c = KERNEL1X1_OUT_CH; in kernel1x1_arm_convolve_1x1_s8_fast() 67 bias_dims.c = output_dims.c; in kernel1x1_arm_convolve_1x1_s8_fast() 121 input_dims.c = KERNEL1X1_STRIDE_X_IN_CH; in kernel1x1_stride_x_arm_convolve_1x1_s8() 126 filter_dims.c = KERNEL1X1_STRIDE_X_IN_CH; in kernel1x1_stride_x_arm_convolve_1x1_s8() 131 output_dims.c = KERNEL1X1_STRIDE_X_OUT_CH; in kernel1x1_stride_x_arm_convolve_1x1_s8() 136 bias_dims.c = output_dims.c; in kernel1x1_stride_x_arm_convolve_1x1_s8() 209 input_dims.c = KERNEL1X1_STRIDE_X_Y_IN_CH; in kernel1x1_stride_x_y_arm_convolve_1x1_s8() 214 filter_dims.c = KERNEL1X1_STRIDE_X_Y_IN_CH; in kernel1x1_stride_x_y_arm_convolve_1x1_s8() [all …]
|
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_ds_cnn_l_s8/ |
D | test_arm_ds_cnn_l_s8.c | 57 input_dims.c = CONV_2D_1_IN_CH; in ds_cnn_l_s8_get_buffer_size() 61 filter_dims.c = CONV_2D_1_IN_CH; in ds_cnn_l_s8_get_buffer_size() 66 output_dims.c = CONV_2D_1_OUT_CH; in ds_cnn_l_s8_get_buffer_size() 92 input_dims.c = DEPTHWISE_CONV_2D_2_OUT_CH; in ds_cnn_l_s8_get_buffer_size() 96 output_dims.c = DEPTHWISE_CONV_2D_2_OUT_CH; in ds_cnn_l_s8_get_buffer_size() 110 input_dims.c = DEPTHWISE_CONV_2D_4_IN_CH; in ds_cnn_l_s8_get_buffer_size() 118 output_dims.c = DEPTHWISE_CONV_2D_4_OUTPUT_H; in ds_cnn_l_s8_get_buffer_size() 170 in_out_dim_0.c = CONV_2D_1_IN_CH; in ds_cnn_l_s8_inference() 174 conv_filter_dims.c = CONV_2D_1_IN_CH; in ds_cnn_l_s8_inference() 179 in_out_dim_1.c = CONV_2D_1_OUT_CH; in ds_cnn_l_s8_inference() [all …]
|
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_fully_connected_s8/ |
D | test_arm_fully_connected_s8.c | 53 input_dims.c = FULLY_CONNECTED_IN_CH; in fully_connected_arm_fully_connected_s8() 55 filter_dims.c = FULLY_CONNECTED_OUT_CH; in fully_connected_arm_fully_connected_s8() 57 output_dims.c = FULLY_CONNECTED_OUT_CH; in fully_connected_arm_fully_connected_s8() 74 …TEST_ASSERT_EQUAL(expected, arm_vector_sum_s8(buf, filter_dims.n, output_dims.c, kernel_data, 1, N… in fully_connected_arm_fully_connected_s8() 121 input_dims.c = FULLY_CONNECTED_W_ZP_IN_CH; in fully_connected_w_zp_arm_fully_connected_s8() 123 filter_dims.c = FULLY_CONNECTED_W_ZP_OUT_CH; in fully_connected_w_zp_arm_fully_connected_s8() 125 output_dims.c = FULLY_CONNECTED_W_ZP_OUT_CH; in fully_connected_w_zp_arm_fully_connected_s8() 142 …TEST_ASSERT_EQUAL(expected, arm_vector_sum_s8(buf, filter_dims.n, output_dims.c, kernel_data, 1, N… in fully_connected_w_zp_arm_fully_connected_s8() 186 input_dims.c = FULLY_CONNECTED_MVE_0_IN_CH; in fully_connected_mve_0_arm_fully_connected_s8() 188 filter_dims.c = FULLY_CONNECTED_MVE_0_OUT_CH; in fully_connected_mve_0_arm_fully_connected_s8() [all …]
|
/cmsis-nn-latest/Tests/UnitTest/Corstone-300/ |
D | retarget.c | 68 int SER_PutChar(int c) { return UartPutc(c); } in SER_PutChar() argument 274 char c; in _read() local 279 c = SER_GetChar(); in _read() 280 if (c == 0x0D) in _read() 282 *ptr++ = c; in _read() 283 SER_PutChar(c); in _read()
|
/cmsis-nn-latest/Source/ActivationFunctions/ |
D | CMakeLists.txt | 18 file(GLOB SRC "./*_s8*.c") 19 file(GLOB SRC_S16 "./*_s16*.c") 20 target_sources(cmsis-nn PRIVATE ${SRC} ${SRC_S16} arm_relu_q7.c arm_relu_q15.c)
|
/cmsis-nn-latest/Source/FullyConnectedFunctions/ |
D | CMakeLists.txt | 19 file(GLOB SRC_S4 "./*_s4.c") 20 file(GLOB SRC_S8 "./*_s8.c") 21 file(GLOB SRC_S16 "./*_s16*.c") 22 file(GLOB SRC_S64 "./*_s64.c")
|
D | arm_fully_connected_s16.c | 78 output_dims->c, /* row_dim or output_depth */ in arm_fully_connected_s16() 82 output += output_dims->c; in arm_fully_connected_s16()
|
D | arm_fully_connected_s4.c | 80 output_dims->c, /* row_dim or output_depth */ in arm_fully_connected_s4() 85 output += output_dims->c; in arm_fully_connected_s4()
|
D | arm_fully_connected_get_buffer_sizes_s8.c | 50 return filter_dims->c * sizeof(int32_t); in arm_fully_connected_s8_get_buffer_size_mve()
|
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_fully_connected_s4/ |
D | test_arm_fully_connected_s4.c | 54 input_dims.c = FULLY_CONNECTED_INT4_IN_CH; in fully_connected_int4_arm_fully_connected_s4() 56 filter_dims.c = FULLY_CONNECTED_INT4_OUT_CH; in fully_connected_int4_arm_fully_connected_s4() 58 output_dims.c = FULLY_CONNECTED_INT4_OUT_CH; in fully_connected_int4_arm_fully_connected_s4() 117 input_dims.c = FULLY_CONNECTED_INT4_2_IN_CH; in fully_connected_int4_arm_fully_connected_s4_2() 119 filter_dims.c = FULLY_CONNECTED_INT4_2_OUT_CH; in fully_connected_int4_arm_fully_connected_s4_2() 121 output_dims.c = FULLY_CONNECTED_INT4_2_OUT_CH; in fully_connected_int4_arm_fully_connected_s4_2() 180 input_dims.c = FULLY_CONNECTED_INT4_3_IN_CH; in fully_connected_int4_arm_fully_connected_s4_3() 182 filter_dims.c = FULLY_CONNECTED_INT4_3_OUT_CH; in fully_connected_int4_arm_fully_connected_s4_3() 184 output_dims.c = FULLY_CONNECTED_INT4_3_OUT_CH; in fully_connected_int4_arm_fully_connected_s4_3() 243 input_dims.c = FULLY_CONNECTED_INT4_4_IN_CH; in fully_connected_int4_arm_fully_connected_s4_4() [all …]
|
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_depthwise_conv_s16/ |
D | test_arm_depthwise_conv_s16.c | 49 input_dims.c = DW_INT16XINT8_IN_CH; in dw_int16xint8_arm_depthwise_conv_s16() 54 output_dims.c = DW_INT16XINT8_OUT_CH; in dw_int16xint8_arm_depthwise_conv_s16() 146 input_dims.c = DW_INT16XINT8_DILATION_IN_CH; in dw_int16xint8_dilation_arm_depthwise_conv_s16() 151 output_dims.c = DW_INT16XINT8_DILATION_OUT_CH; in dw_int16xint8_dilation_arm_depthwise_conv_s16() 243 input_dims.c = DW_INT16XINT8_MULT4_IN_CH; in dw_int16xint8_mult4_arm_depthwise_conv_s16() 248 output_dims.c = DW_INT16XINT8_MULT4_OUT_CH; in dw_int16xint8_mult4_arm_depthwise_conv_s16() 328 input_dims.c = DW_INT16XINT8_MULT4_IN_CH; in arm_depthwise_conv_wrapper_s16_buffer() 334 output_dims.c = input_dims.c; in arm_depthwise_conv_wrapper_s16_buffer() 342 dw_conv_params.ch_mult = output_dims.c / input_dims.c; in arm_depthwise_conv_wrapper_s16_buffer() 352 input_dims.c = 513; in arm_depthwise_conv_wrapper_s16_buffer() [all …]
|
/cmsis-nn-latest/Source/PoolingFunctions/ |
D | CMakeLists.txt | 2 # Copyright (c) 2019-2022 Arm Limited. 19 file(GLOB SRC "./*_s8.c") 20 file(GLOB SRC_S16 "./*_s16.c")
|
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_fully_connected_s16/ |
D | test_arm_fully_connected_s16.c | 50 input_dims.c = FULLY_CONNECTED_INT16_IN_CH; in fully_connected_int16_arm_fully_connected_s16() 52 filter_dims.c = FULLY_CONNECTED_INT16_OUT_CH; in fully_connected_int16_arm_fully_connected_s16() 56 output_dims.c = FULLY_CONNECTED_INT16_OUT_CH; in fully_connected_int16_arm_fully_connected_s16() 115 input_dims.c = FULLY_CONNECTED_INT16_BIG_IN_CH; in fully_connected_int16_big_arm_fully_connected_s16() 117 filter_dims.c = FULLY_CONNECTED_INT16_BIG_OUT_CH; in fully_connected_int16_big_arm_fully_connected_s16() 121 output_dims.c = FULLY_CONNECTED_INT16_BIG_OUT_CH; in fully_connected_int16_big_arm_fully_connected_s16() 179 input_dims.c = FC_INT16_SLOW_IN_CH; in fc_int16_slow_arm_fully_connected_s16() 181 filter_dims.c = FC_INT16_SLOW_OUT_CH; in fc_int16_slow_arm_fully_connected_s16() 185 output_dims.c = FC_INT16_SLOW_OUT_CH; in fc_int16_slow_arm_fully_connected_s16()
|
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_max_pool_s8/ |
D | test_arm_max_pool_s8.c | 50 input_dims.c = MAXPOOLING_IN_CH; in maxpooling_arm_max_pool_s8() 55 output_dims.c = MAXPOOLING_OUT_CH; in maxpooling_arm_max_pool_s8() 91 input_dims.c = MAXPOOLING_1_IN_CH; in maxpooling_1_arm_max_pool_s8() 96 output_dims.c = MAXPOOLING_1_OUT_CH; in maxpooling_1_arm_max_pool_s8() 132 input_dims.c = MAXPOOLING_2_IN_CH; in maxpooling_2_arm_max_pool_s8() 137 output_dims.c = MAXPOOLING_2_OUT_CH; in maxpooling_2_arm_max_pool_s8() 173 input_dims.c = MAXPOOLING_3_IN_CH; in maxpooling_3_arm_max_pool_s8() 178 output_dims.c = MAXPOOLING_3_OUT_CH; in maxpooling_3_arm_max_pool_s8() 214 input_dims.c = MAXPOOLING_4_IN_CH; in maxpooling_4_arm_max_pool_s8() 219 output_dims.c = MAXPOOLING_4_OUT_CH; in maxpooling_4_arm_max_pool_s8() [all …]
|
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_max_pool_s16/ |
D | test_arm_max_pool_s16.c | 45 input_dims.c = MAXPOOL_INT16_IN_CH; in maxpool_int16_arm_max_pool_s16() 50 output_dims.c = MAXPOOL_INT16_OUT_CH; in maxpool_int16_arm_max_pool_s16() 86 input_dims.c = MAXPOOL_INT16_1_IN_CH; in maxpool_int16_1_arm_max_pool_s16() 91 output_dims.c = MAXPOOL_INT16_1_OUT_CH; in maxpool_int16_1_arm_max_pool_s16() 127 input_dims.c = MAXPOOL_INT16_2_IN_CH; in maxpool_int16_2_arm_max_pool_s16() 132 output_dims.c = MAXPOOL_INT16_2_OUT_CH; in maxpool_int16_2_arm_max_pool_s16() 168 input_dims.c = MAXPOOL_INT16_2_IN_CH; in maxpool_int16_param_fail_arm_max_pool_s16() 173 output_dims.c = MAXPOOL_INT16_2_OUT_CH; in maxpool_int16_param_fail_arm_max_pool_s16()
|