/cmsis-nn-latest/Source/ConvolutionFunctions/ |
D | arm_convolve_wrapper_s4.c | 51 const cmsis_nn_conv_params *conv_params, in arm_convolve_wrapper_s4() argument 62 if ((conv_params->padding.w == 0) && (conv_params->padding.h == 0) && (filter_dims->w == 1) && in arm_convolve_wrapper_s4() 63 (filter_dims->h == 1) && (conv_params->dilation.w == 1 && conv_params->dilation.h == 1)) in arm_convolve_wrapper_s4() 65 if ((conv_params->stride.w == 1) && (conv_params->stride.h == 1)) in arm_convolve_wrapper_s4() 68 conv_params, in arm_convolve_wrapper_s4() 82 conv_params, in arm_convolve_wrapper_s4() 94 else if ((input_dims->h == 1) && conv_params->dilation.w == 1 && (filter_dims->h == 1) && in arm_convolve_wrapper_s4() 95 … ((conv_params->stride.w * input_dims->c) % 4 == 0) && (input_dims->c == filter_dims->c)) in arm_convolve_wrapper_s4() 98 conv_params, in arm_convolve_wrapper_s4() 112 conv_params, in arm_convolve_wrapper_s4()
|
D | arm_convolve_get_buffer_sizes_s4.c | 55 …C_INLINE int32_t arm_convolve_1_x_n_s4_get_buffer_size_mve(const cmsis_nn_conv_params *conv_params, in arm_convolve_1_x_n_s4_get_buffer_size_mve() argument 61 const int32_t pad_x = conv_params->padding.w; in arm_convolve_1_x_n_s4_get_buffer_size_mve() 64 const int32_t stride_x = conv_params->stride.w; in arm_convolve_1_x_n_s4_get_buffer_size_mve() 86 int32_t arm_convolve_1_x_n_s4_get_buffer_size(const cmsis_nn_conv_params *conv_params, in arm_convolve_1_x_n_s4_get_buffer_size() argument 92 (void)conv_params; in arm_convolve_1_x_n_s4_get_buffer_size() 97 …return arm_convolve_1_x_n_s4_get_buffer_size_mve(conv_params, input_dims, filter_dims, output_dims… in arm_convolve_1_x_n_s4_get_buffer_size() 114 int32_t arm_convolve_wrapper_s4_get_buffer_size(const cmsis_nn_conv_params *conv_params, in arm_convolve_wrapper_s4_get_buffer_size() argument 120 …return arm_convolve_wrapper_s8_get_buffer_size_mve(conv_params, input_dims, filter_dims, output_di… in arm_convolve_wrapper_s4_get_buffer_size() 123 if ((conv_params->padding.w == 0) && (conv_params->padding.h == 0) && (filter_dims->w == 1) && in arm_convolve_wrapper_s4_get_buffer_size() 124 (filter_dims->h == 1) && (conv_params->dilation.w == 1 && conv_params->dilation.h == 1)) in arm_convolve_wrapper_s4_get_buffer_size() [all …]
|
D | arm_convolve_get_buffer_sizes_s8.c | 55 …C_INLINE int32_t arm_convolve_1_x_n_s8_get_buffer_size_mve(const cmsis_nn_conv_params *conv_params, in arm_convolve_1_x_n_s8_get_buffer_size_mve() argument 61 const int32_t pad_x = conv_params->padding.w; in arm_convolve_1_x_n_s8_get_buffer_size_mve() 64 const int32_t stride_x = conv_params->stride.w; in arm_convolve_1_x_n_s8_get_buffer_size_mve() 98 int32_t arm_convolve_1_x_n_s8_get_buffer_size(const cmsis_nn_conv_params *conv_params, in arm_convolve_1_x_n_s8_get_buffer_size() argument 104 (void)conv_params; in arm_convolve_1_x_n_s8_get_buffer_size() 109 …return arm_convolve_1_x_n_s8_get_buffer_size_mve(conv_params, input_dims, filter_dims, output_dims… in arm_convolve_1_x_n_s8_get_buffer_size() 126 int32_t arm_convolve_wrapper_s8_get_buffer_size(const cmsis_nn_conv_params *conv_params, in arm_convolve_wrapper_s8_get_buffer_size() argument 132 …return arm_convolve_wrapper_s8_get_buffer_size_mve(conv_params, input_dims, filter_dims, output_di… in arm_convolve_wrapper_s8_get_buffer_size() 135 if ((conv_params->padding.w == 0) && (conv_params->padding.h == 0) && (filter_dims->w == 1) && in arm_convolve_wrapper_s8_get_buffer_size() 136 (filter_dims->h == 1) && (conv_params->dilation.w == 1 && conv_params->dilation.h == 1)) in arm_convolve_wrapper_s8_get_buffer_size() [all …]
|
D | arm_convolve_wrapper_s8.c | 51 const cmsis_nn_conv_params *conv_params, in arm_convolve_wrapper_s8() argument 62 if ((conv_params->padding.w == 0) && (conv_params->padding.h == 0) && (filter_dims->w == 1) && in arm_convolve_wrapper_s8() 63 (filter_dims->h == 1) && (conv_params->dilation.w == 1 && conv_params->dilation.h == 1) && in arm_convolve_wrapper_s8() 66 if ((conv_params->stride.w == 1) && (conv_params->stride.h == 1)) in arm_convolve_wrapper_s8() 69 conv_params, in arm_convolve_wrapper_s8() 83 conv_params, in arm_convolve_wrapper_s8() 95 else if ((input_dims->h == 1) && conv_params->dilation.w == 1 && (filter_dims->h == 1) && 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() 99 conv_params, in arm_convolve_wrapper_s8() 113 conv_params, in arm_convolve_wrapper_s8()
|
D | arm_convolve_1x1_s4_fast.c | 51 const cmsis_nn_conv_params *conv_params, in arm_convolve_1x1_s4_fast() argument 62 if (conv_params->padding.w != 0 || conv_params->padding.h != 0 || conv_params->stride.w != 1 || in arm_convolve_1x1_s4_fast() 63 conv_params->stride.h != 1) in arm_convolve_1x1_s4_fast() 85 conv_params->input_offset, in arm_convolve_1x1_s4_fast() 86 conv_params->output_offset, in arm_convolve_1x1_s4_fast() 87 conv_params->activation.min, in arm_convolve_1x1_s4_fast() 88 conv_params->activation.max, in arm_convolve_1x1_s4_fast()
|
D | arm_convolve_1x1_s8_fast.c | 51 const cmsis_nn_conv_params *conv_params, in arm_convolve_1x1_s8_fast() argument 62 if (conv_params->padding.w != 0 || conv_params->padding.h != 0 || conv_params->stride.w != 1 || in arm_convolve_1x1_s8_fast() 63 conv_params->stride.h != 1) in arm_convolve_1x1_s8_fast() 85 conv_params->input_offset, in arm_convolve_1x1_s8_fast() 86 conv_params->output_offset, in arm_convolve_1x1_s8_fast() 87 conv_params->activation.min, in arm_convolve_1x1_s8_fast() 88 conv_params->activation.max, in arm_convolve_1x1_s8_fast()
|
D | arm_convolve_1_x_n_s8.c | 49 const cmsis_nn_conv_params *conv_params, in arm_convolve_1_x_n_s8() argument 63 …if ((input_dims->h != 1) || conv_params->dilation.w != 1 || ctx->buf == NULL || conv_params->strid… in arm_convolve_1_x_n_s8() 64 (conv_params->stride.w * input_dims->c % 4 != 0)) in arm_convolve_1_x_n_s8() 76 const int32_t pad_x = conv_params->padding.w; in arm_convolve_1_x_n_s8() 77 const int32_t stride_x = conv_params->stride.w; in arm_convolve_1_x_n_s8() 102 conv_params, in arm_convolve_1_x_n_s8() 123 …arm_memset_s8(im2col, (int8_t)-conv_params->input_offset, sizeof(int8_t) * (uint32_t)pad_size_left… in arm_convolve_1_x_n_s8() 136 conv_params->input_offset, in arm_convolve_1_x_n_s8() 137 conv_params->output_offset, in arm_convolve_1_x_n_s8() 138 conv_params->activation.min, in arm_convolve_1_x_n_s8() [all …]
|
D | arm_convolve_1x1_s4.c | 51 const cmsis_nn_conv_params *conv_params, in arm_convolve_1x1_s4() argument 65 if (conv_params->padding.w != 0 || conv_params->padding.h != 0) in arm_convolve_1x1_s4() 73 const int32_t stride_w = conv_params->stride.w; in arm_convolve_1x1_s4() 74 const int32_t input_inc = input_dims->w * conv_params->stride.h * rhs_cols; in arm_convolve_1x1_s4() 95 conv_params->input_offset, in arm_convolve_1x1_s4() 96 conv_params->output_offset, in arm_convolve_1x1_s4() 97 conv_params->activation.min, in arm_convolve_1x1_s4() 98 conv_params->activation.max, in arm_convolve_1x1_s4()
|
D | arm_convolve_1x1_s8.c | 51 const cmsis_nn_conv_params *conv_params, in arm_convolve_1x1_s8() argument 65 if (conv_params->padding.w != 0 || conv_params->padding.h != 0) in arm_convolve_1x1_s8() 73 const int32_t stride_w = conv_params->stride.w; in arm_convolve_1x1_s8() 74 const int32_t input_inc = input_dims->w * conv_params->stride.h * rhs_cols; in arm_convolve_1x1_s8() 95 conv_params->input_offset, in arm_convolve_1x1_s8() 96 conv_params->output_offset, in arm_convolve_1x1_s8() 97 conv_params->activation.min, in arm_convolve_1x1_s8() 98 conv_params->activation.max, in arm_convolve_1x1_s8()
|
D | arm_convolve_get_buffer_sizes_s16.c | 70 int32_t arm_convolve_wrapper_s16_get_buffer_size(const cmsis_nn_conv_params *conv_params, in arm_convolve_wrapper_s16_get_buffer_size() argument 75 (void)conv_params; in arm_convolve_wrapper_s16_get_buffer_size() 81 int32_t arm_convolve_wrapper_s16_get_buffer_size_dsp(const cmsis_nn_conv_params *conv_params, in arm_convolve_wrapper_s16_get_buffer_size_dsp() argument 86 …return arm_convolve_wrapper_s16_get_buffer_size(conv_params, input_dims, filter_dims, output_dims); in arm_convolve_wrapper_s16_get_buffer_size_dsp() 89 int32_t arm_convolve_wrapper_s16_get_buffer_size_mve(const cmsis_nn_conv_params *conv_params, in arm_convolve_wrapper_s16_get_buffer_size_mve() argument 94 (void)conv_params; in arm_convolve_wrapper_s16_get_buffer_size_mve()
|
D | arm_convolve_1_x_n_s4.c | 50 const cmsis_nn_conv_params *conv_params, in arm_convolve_1_x_n_s4() argument 62 …int32_t buffer_size = arm_convolve_1_x_n_s4_get_buffer_size(conv_params, input_dims, filter_dims, … in arm_convolve_1_x_n_s4() 64 …if ((input_dims->h != 1) || conv_params->dilation.w != 1 || (buffer_size != 0 && ctx->buf == NULL)… in arm_convolve_1_x_n_s4() 65 conv_params->stride.w == 0 || (conv_params->stride.w * input_dims->c % 4 != 0)) in arm_convolve_1_x_n_s4() 78 const uint16_t pad_x = conv_params->padding.w; in arm_convolve_1_x_n_s4() 79 const uint16_t stride_x = conv_params->stride.w; in arm_convolve_1_x_n_s4() 119 conv_params, in arm_convolve_1_x_n_s4() 145 conv_params->input_offset, in arm_convolve_1_x_n_s4() 146 conv_params->output_offset, in arm_convolve_1_x_n_s4() 147 conv_params->activation.min, in arm_convolve_1_x_n_s4() [all …]
|
D | arm_convolve_wrapper_s16.c | 51 const cmsis_nn_conv_params *conv_params, in arm_convolve_wrapper_s16() argument 63 conv_params, in arm_convolve_wrapper_s16()
|
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_convolve_1x1_s4_fast/ |
D | test_arm_convolve_1x1_s4_fast.c | 38 cmsis_nn_conv_params conv_params; in kernel1x1_arm_convolve_1x1_s4_fast() local 61 conv_params.padding.h = KERNEL1X1_INT4_PAD_Y; in kernel1x1_arm_convolve_1x1_s4_fast() 62 conv_params.padding.w = KERNEL1X1_INT4_PAD_X; in kernel1x1_arm_convolve_1x1_s4_fast() 63 conv_params.stride.h = KERNEL1X1_INT4_STRIDE_Y; in kernel1x1_arm_convolve_1x1_s4_fast() 64 conv_params.stride.w = KERNEL1X1_INT4_STRIDE_X; in kernel1x1_arm_convolve_1x1_s4_fast() 65 conv_params.dilation.w = KERNEL1X1_INT4_DILATION_X; in kernel1x1_arm_convolve_1x1_s4_fast() 66 conv_params.dilation.h = KERNEL1X1_INT4_DILATION_Y; in kernel1x1_arm_convolve_1x1_s4_fast() 73 conv_params.input_offset = KERNEL1X1_INT4_INPUT_OFFSET; in kernel1x1_arm_convolve_1x1_s4_fast() 74 conv_params.output_offset = KERNEL1X1_INT4_OUTPUT_OFFSET; in kernel1x1_arm_convolve_1x1_s4_fast() 75 conv_params.activation.min = KERNEL1X1_INT4_OUT_ACTIVATION_MIN; in kernel1x1_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 | 36 cmsis_nn_conv_params conv_params; in kernel1x1_arm_convolve_1x1_s8_fast() local 59 conv_params.padding.h = KERNEL1X1_PAD_Y; in kernel1x1_arm_convolve_1x1_s8_fast() 60 conv_params.padding.w = KERNEL1X1_PAD_X; in kernel1x1_arm_convolve_1x1_s8_fast() 61 conv_params.stride.h = KERNEL1X1_STRIDE_Y; in kernel1x1_arm_convolve_1x1_s8_fast() 62 conv_params.stride.w = KERNEL1X1_STRIDE_X; in kernel1x1_arm_convolve_1x1_s8_fast() 69 conv_params.input_offset = KERNEL1X1_INPUT_OFFSET; in kernel1x1_arm_convolve_1x1_s8_fast() 70 conv_params.output_offset = KERNEL1X1_OUTPUT_OFFSET; in kernel1x1_arm_convolve_1x1_s8_fast() 71 conv_params.activation.min = KERNEL1X1_OUT_ACTIVATION_MIN; in kernel1x1_arm_convolve_1x1_s8_fast() 72 conv_params.activation.max = KERNEL1X1_OUT_ACTIVATION_MAX; in kernel1x1_arm_convolve_1x1_s8_fast() 81 &conv_params, in kernel1x1_arm_convolve_1x1_s8_fast() [all …]
|
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_convolve_s16/ |
D | test_arm_convolve_s16.c | 45 cmsis_nn_conv_params conv_params; in int16xint8_arm_convolve_s16() local 69 conv_params.padding.w = INT16XINT8_PAD_X; in int16xint8_arm_convolve_s16() 70 conv_params.padding.h = INT16XINT8_PAD_Y; in int16xint8_arm_convolve_s16() 71 conv_params.stride.w = INT16XINT8_STRIDE_X; in int16xint8_arm_convolve_s16() 72 conv_params.stride.h = INT16XINT8_STRIDE_Y; in int16xint8_arm_convolve_s16() 73 conv_params.dilation.w = INT16XINT8_DILATION_X; in int16xint8_arm_convolve_s16() 74 conv_params.dilation.h = INT16XINT8_DILATION_Y; in int16xint8_arm_convolve_s16() 76 conv_params.input_offset = 0; in int16xint8_arm_convolve_s16() 77 conv_params.output_offset = 0; in int16xint8_arm_convolve_s16() 78 conv_params.activation.min = INT16XINT8_OUT_ACTIVATION_MIN; in int16xint8_arm_convolve_s16() [all …]
|
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_convolve_s8/ |
D | test_arm_convolve_s8.c | 45 cmsis_nn_conv_params conv_params; in basic_arm_convolve_s8() local 69 conv_params.padding.w = BASIC_PAD_X; in basic_arm_convolve_s8() 70 conv_params.padding.h = BASIC_PAD_Y; in basic_arm_convolve_s8() 71 conv_params.stride.w = BASIC_STRIDE_X; in basic_arm_convolve_s8() 72 conv_params.stride.h = BASIC_STRIDE_Y; in basic_arm_convolve_s8() 73 conv_params.dilation.w = BASIC_DILATION_X; in basic_arm_convolve_s8() 74 conv_params.dilation.h = BASIC_DILATION_Y; in basic_arm_convolve_s8() 76 conv_params.input_offset = BASIC_INPUT_OFFSET; in basic_arm_convolve_s8() 77 conv_params.output_offset = BASIC_OUTPUT_OFFSET; in basic_arm_convolve_s8() 78 conv_params.activation.min = BASIC_OUT_ACTIVATION_MIN; in basic_arm_convolve_s8() [all …]
|
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_convolve_1_x_n_s8/ |
D | test_arm_convolve_1_x_n_s8.c | 41 cmsis_nn_conv_params conv_params; in conv_1_x_n_1_arm_convolve_s8() local 65 conv_params.padding.w = CONV_1_X_N_1_PAD_X; in conv_1_x_n_1_arm_convolve_s8() 66 conv_params.padding.h = CONV_1_X_N_1_PAD_Y; in conv_1_x_n_1_arm_convolve_s8() 67 conv_params.stride.w = CONV_1_X_N_1_STRIDE_X; in conv_1_x_n_1_arm_convolve_s8() 68 conv_params.stride.h = CONV_1_X_N_1_STRIDE_Y; in conv_1_x_n_1_arm_convolve_s8() 69 conv_params.dilation.w = CONV_1_X_N_1_DILATION_X; in conv_1_x_n_1_arm_convolve_s8() 70 conv_params.dilation.h = CONV_1_X_N_1_DILATION_Y; in conv_1_x_n_1_arm_convolve_s8() 72 conv_params.input_offset = CONV_1_X_N_1_INPUT_OFFSET; in conv_1_x_n_1_arm_convolve_s8() 73 conv_params.output_offset = CONV_1_X_N_1_OUTPUT_OFFSET; in conv_1_x_n_1_arm_convolve_s8() 74 conv_params.activation.min = CONV_1_X_N_1_OUT_ACTIVATION_MIN; in conv_1_x_n_1_arm_convolve_s8() [all …]
|
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_convolve_s4/ |
D | test_arm_convolve_s4.c | 51 cmsis_nn_conv_params conv_params; in basic_arm_convolve_s4() local 74 conv_params.padding.w = BASIC_INT4_PAD_X; in basic_arm_convolve_s4() 75 conv_params.padding.h = BASIC_INT4_PAD_Y; in basic_arm_convolve_s4() 76 conv_params.stride.w = BASIC_INT4_STRIDE_X; in basic_arm_convolve_s4() 77 conv_params.stride.h = BASIC_INT4_STRIDE_Y; in basic_arm_convolve_s4() 78 conv_params.dilation.w = BASIC_INT4_DILATION_X; in basic_arm_convolve_s4() 79 conv_params.dilation.h = BASIC_INT4_DILATION_Y; in basic_arm_convolve_s4() 81 conv_params.input_offset = BASIC_INT4_INPUT_OFFSET; in basic_arm_convolve_s4() 82 conv_params.output_offset = BASIC_INT4_OUTPUT_OFFSET; in basic_arm_convolve_s4() 83 conv_params.activation.min = BASIC_INT4_OUT_ACTIVATION_MIN; in basic_arm_convolve_s4() [all …]
|
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_grouped_convolve_s8/ |
D | test_arm_grouped_convolve_s8.c | 36 cmsis_nn_conv_params conv_params; in grouped_conv_arm_grouped_convolve_1_s8() local 60 conv_params.padding.w = GROUPED_CONV_1_PAD_X; in grouped_conv_arm_grouped_convolve_1_s8() 61 conv_params.padding.h = GROUPED_CONV_1_PAD_Y; in grouped_conv_arm_grouped_convolve_1_s8() 62 conv_params.stride.w = GROUPED_CONV_1_STRIDE_X; in grouped_conv_arm_grouped_convolve_1_s8() 63 conv_params.stride.h = GROUPED_CONV_1_STRIDE_Y; in grouped_conv_arm_grouped_convolve_1_s8() 64 conv_params.dilation.w = GROUPED_CONV_1_DILATION_X; in grouped_conv_arm_grouped_convolve_1_s8() 65 conv_params.dilation.h = GROUPED_CONV_1_DILATION_Y; in grouped_conv_arm_grouped_convolve_1_s8() 67 conv_params.input_offset = GROUPED_CONV_1_INPUT_OFFSET; in grouped_conv_arm_grouped_convolve_1_s8() 68 conv_params.output_offset = GROUPED_CONV_1_OUTPUT_OFFSET; in grouped_conv_arm_grouped_convolve_1_s8() 69 conv_params.activation.min = GROUPED_CONV_1_OUT_ACTIVATION_MIN; in grouped_conv_arm_grouped_convolve_1_s8() [all …]
|
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_ds_cnn_s_s8/ |
D | test_arm_ds_cnn_s_s8.c | 43 cmsis_nn_conv_params conv_params; in ds_cnn_s_s8_get_buffer_size() local 49 conv_params.padding.h = CONV_2D_1_PAD_H; in ds_cnn_s_s8_get_buffer_size() 50 conv_params.padding.w = CONV_2D_1_PAD_W; in ds_cnn_s_s8_get_buffer_size() 51 conv_params.stride.h = CONV_2D_1_STRIDE_H; in ds_cnn_s_s8_get_buffer_size() 52 conv_params.stride.w = CONV_2D_1_STRIDE_W; in ds_cnn_s_s8_get_buffer_size() 53 conv_params.dilation.h = CONV_2D_1_DILATION_H; in ds_cnn_s_s8_get_buffer_size() 54 conv_params.dilation.w = CONV_2D_1_DILATION_W; in ds_cnn_s_s8_get_buffer_size() 70 …int32_t size = arm_convolve_wrapper_s8_get_buffer_size(&conv_params, &input_dims, &filter_dims, &o… in ds_cnn_s_s8_get_buffer_size() 125 cmsis_nn_conv_params conv_params; in ds_cnn_s_s8_inference() local 133 conv_params.padding.h = CONV_2D_1_PAD_H; in ds_cnn_s_s8_inference() [all …]
|
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_ds_cnn_l_s8/ |
D | test_arm_ds_cnn_l_s8.c | 41 cmsis_nn_conv_params conv_params; in ds_cnn_l_s8_get_buffer_size() local 47 conv_params.padding.h = CONV_2D_1_PAD_H; in ds_cnn_l_s8_get_buffer_size() 48 conv_params.padding.w = CONV_2D_1_PAD_W; in ds_cnn_l_s8_get_buffer_size() 49 conv_params.stride.h = CONV_2D_1_STRIDE_H; in ds_cnn_l_s8_get_buffer_size() 50 conv_params.stride.w = CONV_2D_1_STRIDE_W; in ds_cnn_l_s8_get_buffer_size() 51 conv_params.dilation.h = CONV_2D_1_DILATION_H; in ds_cnn_l_s8_get_buffer_size() 52 conv_params.dilation.w = CONV_2D_1_DILATION_W; in ds_cnn_l_s8_get_buffer_size() 68 …int32_t size = arm_convolve_wrapper_s8_get_buffer_size(&conv_params, &input_dims, &filter_dims, &o… in ds_cnn_l_s8_get_buffer_size() 144 cmsis_nn_conv_params conv_params; in ds_cnn_l_s8_inference() local 152 conv_params.padding.h = CONV_2D_1_PAD_H; in ds_cnn_l_s8_inference() [all …]
|
/cmsis-nn-latest/Source/NNSupportFunctions/ |
D | arm_nn_mat_mul_core_1x_s8.c | 52 const cmsis_nn_conv_params *conv_params, in arm_nn_mat_mul_core_1x_s8() argument 61 const int32_t out_offset = conv_params->output_offset; in arm_nn_mat_mul_core_1x_s8() 62 const int32_t out_activation_min = conv_params->activation.min; in arm_nn_mat_mul_core_1x_s8() 63 const int32_t out_activation_max = conv_params->activation.max; in arm_nn_mat_mul_core_1x_s8() 96 sum_tmp *= conv_params->input_offset; in arm_nn_mat_mul_core_1x_s8() 130 acc_n0 += conv_params->output_offset; in arm_nn_mat_mul_core_1x_s8() 131 acc_n0 = MAX(acc_n0, conv_params->activation.min); in arm_nn_mat_mul_core_1x_s8() 132 acc_n0 = MIN(acc_n0, conv_params->activation.max); in arm_nn_mat_mul_core_1x_s8() 143 (void)conv_params; in arm_nn_mat_mul_core_1x_s8()
|
D | arm_nn_mat_mul_core_1x_s4.c | 52 const cmsis_nn_conv_params *conv_params, in arm_nn_mat_mul_core_1x_s4() argument 61 const int32_t out_offset = conv_params->output_offset; in arm_nn_mat_mul_core_1x_s4() 62 const int32_t out_activation_min = conv_params->activation.min; in arm_nn_mat_mul_core_1x_s4() 63 const int32_t out_activation_max = conv_params->activation.max; in arm_nn_mat_mul_core_1x_s4() 91 sum_tmp *= conv_params->input_offset; in arm_nn_mat_mul_core_1x_s4() 126 acc_n0 += conv_params->output_offset; in arm_nn_mat_mul_core_1x_s4() 127 acc_n0 = MAX(acc_n0, conv_params->activation.min); in arm_nn_mat_mul_core_1x_s4() 128 acc_n0 = MIN(acc_n0, conv_params->activation.max); in arm_nn_mat_mul_core_1x_s4() 139 (void)conv_params; in arm_nn_mat_mul_core_1x_s4()
|
D | arm_nn_mat_mul_core_4x_s8.c | 52 const cmsis_nn_conv_params *conv_params, in arm_nn_mat_mul_core_4x_s8() argument 115 sum_tmp *= conv_params->input_offset; in arm_nn_mat_mul_core_4x_s8() 123 res = vaddq_n_s32(res, conv_params->output_offset); in arm_nn_mat_mul_core_4x_s8() 125 res = vmaxq_s32(res, vdupq_n_s32(conv_params->activation.min)); in arm_nn_mat_mul_core_4x_s8() 126 res = vminq_s32(res, vdupq_n_s32(conv_params->activation.max)); in arm_nn_mat_mul_core_4x_s8() 140 (void)conv_params; in arm_nn_mat_mul_core_4x_s8()
|
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_depthwise_conv_s16/ |
D | test_arm_depthwise_conv_s16.c | 367 cmsis_nn_dw_conv_params conv_params; in buffer_size_mve_arm_depthwise_conv_s16() local 382 conv_params.padding.w = DW_INT16XINT8_MULT4_PAD_X; in buffer_size_mve_arm_depthwise_conv_s16() 383 conv_params.padding.h = DW_INT16XINT8_MULT4_PAD_Y; in buffer_size_mve_arm_depthwise_conv_s16() 384 conv_params.stride.w = DW_INT16XINT8_MULT4_STRIDE_X; in buffer_size_mve_arm_depthwise_conv_s16() 385 conv_params.stride.h = DW_INT16XINT8_MULT4_STRIDE_Y; in buffer_size_mve_arm_depthwise_conv_s16() 386 conv_params.dilation.w = DW_INT16XINT8_MULT4_DILATION_X; in buffer_size_mve_arm_depthwise_conv_s16() 387 conv_params.dilation.h = DW_INT16XINT8_MULT4_DILATION_Y; in buffer_size_mve_arm_depthwise_conv_s16() 388 conv_params.ch_mult = DW_INT16XINT8_MULT4_CH_MULT; in buffer_size_mve_arm_depthwise_conv_s16() 389 conv_params.input_offset = DW_INT16XINT8_MULT4_INPUT_OFFSET; in buffer_size_mve_arm_depthwise_conv_s16() 390 conv_params.output_offset = DW_INT16XINT8_MULT4_OUTPUT_OFFSET; in buffer_size_mve_arm_depthwise_conv_s16() [all …]
|