/cmsis-nn-latest/Source/PadFunctions/ |
D | arm_pad_s8.c | 62 pre_pad->c + input_size->c + post_pad->c}; in arm_pad_s8() 64 const int32_t batch_block_size = output_size.h * output_size.w * output_size.c; in arm_pad_s8() 65 const int32_t row_block_size = output_size.w * output_size.c; in arm_pad_s8() 66 const int32_t col_block_size = output_size.c; in arm_pad_s8() 80 if (input_size->c == output_size.c) in arm_pad_s8() 82 arm_memcpy_s8(output, input, input_size->w * input_size->c); in arm_pad_s8() 83 output += input_size->w * input_size->c; in arm_pad_s8() 84 input += input_size->w * input_size->c; in arm_pad_s8() 91 arm_memset_s8(output, pad_value, pre_pad->c); in arm_pad_s8() 92 output += pre_pad->c; in arm_pad_s8() [all …]
|
/cmsis-nn-latest/Source/ConvolutionFunctions/ |
D | arm_depthwise_conv_get_buffer_sizes_s8.c | 49 …const cmsis_nn_dims filter_conv_dims = {filter_dims->c, filter_dims->h, filter_dims->w, filter_dim… in arm_deptwise_conv_s8_one_in_ch_get_buffer_size_mve() 59 size += filter_dims->c * filter_dims->h * filter_dims->w * filter_dims->n; in arm_deptwise_conv_s8_one_in_ch_get_buffer_size_mve() 72 return (input_dims->c * filter_dims->w * filter_dims->h) * sizeof(int16_t); in arm_depthwise_conv_s8_opt_get_buffer_size_dsp() 96 …if (input_dims->c == 1 && output_dims->c > CONVERT_DW_CONV_WITH_ONE_INPUT_CH_AND_OUTPUT_CH_ABOVE_T… in arm_depthwise_conv_wrapper_s8_get_buffer_size() 102 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() 125 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() 146 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() 152 …if (input_dims->c == 1 && output_dims->c > CONVERT_DW_CONV_WITH_ONE_INPUT_CH_AND_OUTPUT_CH_ABOVE_T… in arm_depthwise_conv_wrapper_s8_get_buffer_size_mve()
|
D | arm_transpose_conv_get_buffer_sizes_s8.c | 59 const bool reverse_conv_efficient = (input_dims->c > REVERSE_TCOL_EFFICIENT_THRESHOLD); in arm_transpose_conv_s8_get_buffer_size() 66 input_dims->c}; in arm_transpose_conv_s8_get_buffer_size() 73 out_dims->c; in arm_transpose_conv_s8_get_buffer_size() 85 const bool reverse_conv_efficient = (input_dims->c > REVERSE_TCOL_EFFICIENT_THRESHOLD); in arm_transpose_conv_s8_get_reverse_conv_buffer_size() 89 return input_dims->c * filter_dims->w * filter_dims->h * filter_dims->n; in arm_transpose_conv_s8_get_reverse_conv_buffer_size()
|
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_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_convolve_get_buffer_sizes_s8.c | 46 return (2 * input_dims->c) * (int32_t)sizeof(int16_t); in arm_convolve_1x1_s8_fast_get_buffer_size_dsp() 56 int32_t col_length = input_dims->c * filter_dims->w * filter_dims->h; in arm_convolve_s8_get_buffer_size_mve() 86 const int32_t pad_size_left = pad_x * input_dims->c; in arm_convolve_1_x_n_s8_get_buffer_size_mve() 87 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() 88 const int32_t num_elem_left = kernel_x * input_dims->c; in arm_convolve_1_x_n_s8_get_buffer_size_mve() 89 const int32_t num_elem_right = num_elem_left - input_dims->c; in arm_convolve_1_x_n_s8_get_buffer_size_mve() 100 const int32_t rhs_cols = filter_dims->w * filter_dims->h * input_dims->c; in arm_convolve_s8_get_buffer_size() 163 (conv_params->stride.w * input_dims->c % 4 == 0)) in arm_convolve_wrapper_s8_get_buffer_size() 193 (conv_params->stride.w * input_dims->c % 4 == 0)) in arm_convolve_wrapper_s8_get_buffer_size_mve() 222 (conv_params->stride.w * input_dims->c % 4 == 0)) in arm_convolve_wrapper_s8_get_buffer_size_dsp()
|
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() 110 else if (((filter_dims->h * filter_dims->w * input_dims->c) & 0x1) == 0) in arm_convolve_wrapper_s4()
|
/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 file(GLOB SRC_S32 "./*_s32*.c") 23 target_sources(cmsis-nn PRIVATE ${SRC_S4} ${SRC_S8} ${SRC_S16} ${SRC_S32} arm_nntables.c 24 arm_q7_to_q15_with_offset.c 25 arm_s8_to_s16_unordered_with_offset.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/Source/TransposeFunctions/ |
D | arm_transpose_s8.c | 52 const int32_t c = input_dims->c; in arm_transpose_s8_nhcw() local 58 const uint16_t src_cols = c; in arm_transpose_s8_nhcw() 135 const int32_t c = input_dims->c; in arm_transpose_s8_default() local 143 for (int32_t z = 0; z < c; z++) in arm_transpose_s8_default() 180 const int32_t c = input_dims->c; in arm_transpose_s8() local 182 in_strides[0] = h * w * c; in arm_transpose_s8() 183 in_strides[1] = w * c; in arm_transpose_s8() 184 in_strides[2] = c; in arm_transpose_s8() 226 out_strides[perm[0]] = output_dims->h * output_dims->w * output_dims->c; in arm_transpose_s8() 227 out_strides[perm[1]] = output_dims->w * output_dims->c; in arm_transpose_s8() [all …]
|
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_maximum_minimum_s8/ |
D | test_arm_maximum_s8.c | 55 input_1_dims.c = MAXIMUM_SCALAR_1_INT8_CHANNEL_1; in maximum_scalar_1_int8() 60 input_2_dims.c = MAXIMUM_SCALAR_1_INT8_CHANNEL_2; in maximum_scalar_1_int8() 65 output_dims.c = MAXIMUM_SCALAR_1_INT8_OUTPUT_CHANNEL; in maximum_scalar_1_int8() 103 input_1_dims.c = MAXIMUM_SCALAR_2_INT8_CHANNEL_1; in maximum_scalar_2_int8() 108 input_2_dims.c = MAXIMUM_SCALAR_2_INT8_CHANNEL_2; in maximum_scalar_2_int8() 113 output_dims.c = MAXIMUM_SCALAR_2_INT8_OUTPUT_CHANNEL; in maximum_scalar_2_int8() 150 input_1_dims.c = MAXIMUM_NO_BROADCAST_INT8_CHANNEL_1; in maximum_no_broadcast_int8() 155 input_2_dims.c = MAXIMUM_NO_BROADCAST_INT8_CHANNEL_2; in maximum_no_broadcast_int8() 160 output_dims.c = MAXIMUM_NO_BROADCAST_INT8_OUTPUT_CHANNEL; in maximum_no_broadcast_int8() 197 input_1_dims.c = MAXIMUM_BROADCAST_BATCH_INT8_CHANNEL_1; in maximum_broadcast_batch_int8() [all …]
|
D | test_arm_minimum_s8.c | 55 input_1_dims.c = MINIMUM_SCALAR_1_INT8_CHANNEL_1; in minimum_scalar_1_int8() 60 input_2_dims.c = MINIMUM_SCALAR_1_INT8_CHANNEL_2; in minimum_scalar_1_int8() 65 output_dims.c = MINIMUM_SCALAR_1_INT8_OUTPUT_CHANNEL; in minimum_scalar_1_int8() 103 input_1_dims.c = MINIMUM_SCALAR_2_INT8_CHANNEL_1; in minimum_scalar_2_int8() 108 input_2_dims.c = MINIMUM_SCALAR_2_INT8_CHANNEL_2; in minimum_scalar_2_int8() 113 output_dims.c = MINIMUM_SCALAR_2_INT8_OUTPUT_CHANNEL; in minimum_scalar_2_int8() 150 input_1_dims.c = MINIMUM_NO_BROADCAST_INT8_CHANNEL_1; in minimum_no_broadcast_int8() 155 input_2_dims.c = MINIMUM_NO_BROADCAST_INT8_CHANNEL_2; in minimum_no_broadcast_int8() 160 output_dims.c = MINIMUM_NO_BROADCAST_INT8_OUTPUT_CHANNEL; in minimum_no_broadcast_int8() 197 input_1_dims.c = MINIMUM_BROADCAST_BATCH_INT8_CHANNEL_1; in minimum_broadcast_batch_int8() [all …]
|
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_ds_cnn_s_s8/ |
D | test_arm_ds_cnn_s_s8.c | 58 input_dims.c = CONV_2D_1_IN_CH; in ds_cnn_s_s8_get_buffer_size() 62 filter_dims.c = CONV_2D_1_IN_CH; in ds_cnn_s_s8_get_buffer_size() 67 output_dims.c = CONV_2D_1_OUT_CH; in ds_cnn_s_s8_get_buffer_size() 93 input_dims.c = DEPTHWISE_CONV_2D_2_OUT_CH; in ds_cnn_s_s8_get_buffer_size() 97 output_dims.c = DEPTHWISE_CONV_2D_2_OUT_CH; in ds_cnn_s_s8_get_buffer_size() 153 in_out_dim_0.c = CONV_2D_1_IN_CH; in ds_cnn_s_s8_inference() 157 conv_filter_dims.c = CONV_2D_1_IN_CH; in ds_cnn_s_s8_inference() 162 in_out_dim_1.c = CONV_2D_1_OUT_CH; in ds_cnn_s_s8_inference() 163 bias_dims.c = CONV_2D_1_OUT_CH; in ds_cnn_s_s8_inference() 203 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_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/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/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/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/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()
|
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_fully_connected_s8/ |
D | test_arm_fully_connected_s8.c | 54 input_dims.c = FULLY_CONNECTED_IN_CH; in fully_connected_arm_fully_connected_s8() 56 filter_dims.c = FULLY_CONNECTED_OUT_CH; in fully_connected_arm_fully_connected_s8() 58 output_dims.c = FULLY_CONNECTED_OUT_CH; in fully_connected_arm_fully_connected_s8() 78 output_dims.c, in fully_connected_arm_fully_connected_s8() 129 input_dims.c = FULLY_CONNECTED_W_ZP_IN_CH; in fully_connected_w_zp_arm_fully_connected_s8() 131 filter_dims.c = FULLY_CONNECTED_W_ZP_OUT_CH; in fully_connected_w_zp_arm_fully_connected_s8() 133 output_dims.c = FULLY_CONNECTED_W_ZP_OUT_CH; in fully_connected_w_zp_arm_fully_connected_s8() 153 output_dims.c, in fully_connected_w_zp_arm_fully_connected_s8() 201 input_dims.c = FULLY_CONNECTED_MVE_0_IN_CH; in fully_connected_mve_0_arm_fully_connected_s8() 203 filter_dims.c = FULLY_CONNECTED_MVE_0_OUT_CH; in fully_connected_mve_0_arm_fully_connected_s8() [all …]
|
/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/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_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/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()
|