Home
last modified time | relevance | path

Searched refs:dw_conv_params (Results 1 – 18 of 18) sorted by relevance

/cmsis-nn-3.6.0-3.5.0/Source/ConvolutionFunctions/
Darm_depthwise_conv_wrapper_s8.c50 const cmsis_nn_dw_conv_params *dw_conv_params, in arm_depthwise_conv_wrapper_s8() argument
62 if (1 == dw_conv_params->ch_mult && input_dims->n == 1 && dw_conv_params->dilation.w == 1 && in arm_depthwise_conv_wrapper_s8()
63 dw_conv_params->dilation.h == 1) in arm_depthwise_conv_wrapper_s8()
66 if (filter_dims->w == 3 && filter_dims->h == 3 && dw_conv_params->padding.h <= 1 && in arm_depthwise_conv_wrapper_s8()
67 dw_conv_params->padding.w <= 1) in arm_depthwise_conv_wrapper_s8()
70 dw_conv_params, in arm_depthwise_conv_wrapper_s8()
85 dw_conv_params, in arm_depthwise_conv_wrapper_s8()
100 dw_conv_params, in arm_depthwise_conv_wrapper_s8()
Darm_depthwise_conv_get_buffer_sizes_s8.c69 int32_t arm_depthwise_conv_wrapper_s8_get_buffer_size(const cmsis_nn_dw_conv_params *dw_conv_params, in arm_depthwise_conv_wrapper_s8_get_buffer_size() argument
76 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()
77 dw_conv_params->dilation.h == 1) in arm_depthwise_conv_wrapper_s8_get_buffer_size()
80 if (filter_dims->w == 3 && filter_dims->h == 3 && dw_conv_params->padding.h <= 1 && in arm_depthwise_conv_wrapper_s8_get_buffer_size()
81 dw_conv_params->padding.w <= 1) in arm_depthwise_conv_wrapper_s8_get_buffer_size()
92 …_t arm_depthwise_conv_wrapper_s8_get_buffer_size_dsp(const cmsis_nn_dw_conv_params *dw_conv_params, in arm_depthwise_conv_wrapper_s8_get_buffer_size_dsp() argument
99 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()
100 dw_conv_params->dilation.h == 1) in arm_depthwise_conv_wrapper_s8_get_buffer_size_dsp()
102 if (filter_dims->w == 3 && filter_dims->h == 3 && dw_conv_params->padding.h <= 1 && in arm_depthwise_conv_wrapper_s8_get_buffer_size_dsp()
103 dw_conv_params->padding.w <= 1) in arm_depthwise_conv_wrapper_s8_get_buffer_size_dsp()
[all …]
Darm_depthwise_conv_wrapper_s16.c51 const cmsis_nn_dw_conv_params *dw_conv_params, in arm_depthwise_conv_wrapper_s16() argument
64 if (USE_FAST_DW_CONV_S16_FUNCTION(dw_conv_params, filter_dims, input_dims)) in arm_depthwise_conv_wrapper_s16()
67 dw_conv_params, in arm_depthwise_conv_wrapper_s16()
81 dw_conv_params, in arm_depthwise_conv_wrapper_s16()
Darm_depthwise_conv_get_buffer_sizes_s16.c70 …t32_t arm_depthwise_conv_wrapper_s16_get_buffer_size(const cmsis_nn_dw_conv_params *dw_conv_params, in arm_depthwise_conv_wrapper_s16_get_buffer_size() argument
79 if (USE_FAST_DW_CONV_S16_FUNCTION(dw_conv_params, filter_dims, input_dims)) in arm_depthwise_conv_wrapper_s16_get_buffer_size()
87 …t arm_depthwise_conv_wrapper_s16_get_buffer_size_mve(const cmsis_nn_dw_conv_params *dw_conv_params, in arm_depthwise_conv_wrapper_s16_get_buffer_size_mve() argument
96 if (USE_FAST_DW_CONV_S16_FUNCTION(dw_conv_params, filter_dims, input_dims)) in arm_depthwise_conv_wrapper_s16_get_buffer_size_mve()
104 …t arm_depthwise_conv_wrapper_s16_get_buffer_size_dsp(const cmsis_nn_dw_conv_params *dw_conv_params, in arm_depthwise_conv_wrapper_s16_get_buffer_size_dsp() argument
113 if (USE_FAST_DW_CONV_S16_FUNCTION(dw_conv_params, filter_dims, input_dims)) in arm_depthwise_conv_wrapper_s16_get_buffer_size_dsp()
Darm_depthwise_conv_s8.c274 const cmsis_nn_dw_conv_params *dw_conv_params, in arm_depthwise_conv_s8() argument
285 const uint16_t dilation_x = dw_conv_params->dilation.w; in arm_depthwise_conv_s8()
286 const uint16_t dilation_y = dw_conv_params->dilation.h; in arm_depthwise_conv_s8()
291 if (dw_conv_params->ch_mult % 4 == 0 && input_dims->n == 1 && dw_conv_params->dilation.w == 1 && in arm_depthwise_conv_s8()
292 dw_conv_params->dilation.h == 1) in arm_depthwise_conv_s8()
300 dw_conv_params->ch_mult, in arm_depthwise_conv_s8()
303 dw_conv_params->padding.w, in arm_depthwise_conv_s8()
304 dw_conv_params->padding.h, in arm_depthwise_conv_s8()
305 dw_conv_params->stride.w, in arm_depthwise_conv_s8()
306 dw_conv_params->stride.h, in arm_depthwise_conv_s8()
[all …]
Darm_depthwise_conv_s16.c245 const cmsis_nn_dw_conv_params *dw_conv_params, in arm_depthwise_conv_s16() argument
256 const uint16_t dilation_x = dw_conv_params->dilation.w; in arm_depthwise_conv_s16()
257 const uint16_t dilation_y = dw_conv_params->dilation.h; in arm_depthwise_conv_s16()
268 dw_conv_params->ch_mult, in arm_depthwise_conv_s16()
271 dw_conv_params->padding.w, in arm_depthwise_conv_s16()
272 dw_conv_params->padding.h, in arm_depthwise_conv_s16()
273 dw_conv_params->stride.w, in arm_depthwise_conv_s16()
274 dw_conv_params->stride.h, in arm_depthwise_conv_s16()
281 dw_conv_params->activation.min, in arm_depthwise_conv_s16()
282 dw_conv_params->activation.max, in arm_depthwise_conv_s16()
Darm_depthwise_conv_3x3_s8.c53 const cmsis_nn_dw_conv_params *dw_conv_params, in arm_depthwise_conv_3x3_s8() argument
71 const int32_t pad_x = dw_conv_params->padding.w; in arm_depthwise_conv_3x3_s8()
72 const int32_t pad_y = dw_conv_params->padding.h; in arm_depthwise_conv_3x3_s8()
73 const int32_t stride_x = dw_conv_params->stride.w; in arm_depthwise_conv_3x3_s8()
74 const int32_t stride_y = dw_conv_params->stride.h; in arm_depthwise_conv_3x3_s8()
79 const int32_t output_offset = dw_conv_params->output_offset; in arm_depthwise_conv_3x3_s8()
80 const int32_t input_offset = dw_conv_params->input_offset; in arm_depthwise_conv_3x3_s8()
81 const int32_t output_activation_min = dw_conv_params->activation.min; in arm_depthwise_conv_3x3_s8()
82 const int32_t output_activation_max = dw_conv_params->activation.max; in arm_depthwise_conv_3x3_s8()
Darm_depthwise_conv_s8_opt.c52 const cmsis_nn_dw_conv_params *dw_conv_params, in arm_depthwise_conv_s8_opt() argument
82 const int32_t pad_x = dw_conv_params->padding.w; in arm_depthwise_conv_s8_opt()
83 const int32_t pad_y = dw_conv_params->padding.h; in arm_depthwise_conv_s8_opt()
84 const int32_t stride_x = dw_conv_params->stride.w; in arm_depthwise_conv_s8_opt()
85 const int32_t stride_y = dw_conv_params->stride.h; in arm_depthwise_conv_s8_opt()
90 const int32_t output_offset = dw_conv_params->output_offset; in arm_depthwise_conv_s8_opt()
91 const int32_t input_offset = dw_conv_params->input_offset; in arm_depthwise_conv_s8_opt()
92 const int32_t output_activation_min = dw_conv_params->activation.min; in arm_depthwise_conv_s8_opt()
93 const int32_t output_activation_max = dw_conv_params->activation.max; in arm_depthwise_conv_s8_opt()
408 dw_conv_params, in arm_depthwise_conv_s8_opt()
Darm_depthwise_conv_fast_s16.c52 const cmsis_nn_dw_conv_params *dw_conv_params, in arm_depthwise_conv_fast_s16() argument
89 const int32_t pad_x = dw_conv_params->padding.w; in arm_depthwise_conv_fast_s16()
90 const int32_t pad_y = dw_conv_params->padding.h; in arm_depthwise_conv_fast_s16()
91 const int32_t stride_x = dw_conv_params->stride.w; in arm_depthwise_conv_fast_s16()
92 const int32_t stride_y = dw_conv_params->stride.h; in arm_depthwise_conv_fast_s16()
97 const int32_t output_activation_min = dw_conv_params->activation.min; in arm_depthwise_conv_fast_s16()
98 const int32_t output_activation_max = dw_conv_params->activation.max; in arm_depthwise_conv_fast_s16()
433 dw_conv_params, in arm_depthwise_conv_fast_s16()
/cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_depthwise_conv_fast_s16/
Dtest_arm_depthwise_conv_fast_s16.c40 cmsis_nn_dw_conv_params dw_conv_params; in dw_int16xint8_fast_arm_depthwise_conv_fast_s16() local
63 dw_conv_params.padding.w = DW_INT16XINT8_FAST_PAD_X; in dw_int16xint8_fast_arm_depthwise_conv_fast_s16()
64 dw_conv_params.padding.h = DW_INT16XINT8_FAST_PAD_Y; in dw_int16xint8_fast_arm_depthwise_conv_fast_s16()
65 dw_conv_params.stride.w = DW_INT16XINT8_FAST_STRIDE_X; in dw_int16xint8_fast_arm_depthwise_conv_fast_s16()
66 dw_conv_params.stride.h = DW_INT16XINT8_FAST_STRIDE_Y; in dw_int16xint8_fast_arm_depthwise_conv_fast_s16()
67 dw_conv_params.dilation.w = DW_INT16XINT8_FAST_DILATION_X; in dw_int16xint8_fast_arm_depthwise_conv_fast_s16()
68 dw_conv_params.dilation.h = DW_INT16XINT8_FAST_DILATION_Y; in dw_int16xint8_fast_arm_depthwise_conv_fast_s16()
70 dw_conv_params.ch_mult = DW_INT16XINT8_FAST_CH_MULT; in dw_int16xint8_fast_arm_depthwise_conv_fast_s16()
72 dw_conv_params.input_offset = DW_INT16XINT8_FAST_INPUT_OFFSET; in dw_int16xint8_fast_arm_depthwise_conv_fast_s16()
73 dw_conv_params.output_offset = DW_INT16XINT8_FAST_OUTPUT_OFFSET; in dw_int16xint8_fast_arm_depthwise_conv_fast_s16()
[all …]
/cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_depthwise_conv_s16/
Dtest_arm_depthwise_conv_s16.c33 cmsis_nn_dw_conv_params dw_conv_params; in dw_int16xint8_arm_depthwise_conv_s16() local
56 dw_conv_params.padding.w = DW_INT16XINT8_PAD_X; in dw_int16xint8_arm_depthwise_conv_s16()
57 dw_conv_params.padding.h = DW_INT16XINT8_PAD_Y; in dw_int16xint8_arm_depthwise_conv_s16()
58 dw_conv_params.stride.w = DW_INT16XINT8_STRIDE_X; in dw_int16xint8_arm_depthwise_conv_s16()
59 dw_conv_params.stride.h = DW_INT16XINT8_STRIDE_Y; in dw_int16xint8_arm_depthwise_conv_s16()
60 dw_conv_params.dilation.w = DW_INT16XINT8_DILATION_X; in dw_int16xint8_arm_depthwise_conv_s16()
61 dw_conv_params.dilation.h = DW_INT16XINT8_DILATION_Y; in dw_int16xint8_arm_depthwise_conv_s16()
63 dw_conv_params.ch_mult = DW_INT16XINT8_CH_MULT; in dw_int16xint8_arm_depthwise_conv_s16()
65 dw_conv_params.input_offset = DW_INT16XINT8_INPUT_OFFSET; in dw_int16xint8_arm_depthwise_conv_s16()
66 dw_conv_params.output_offset = DW_INT16XINT8_OUTPUT_OFFSET; in dw_int16xint8_arm_depthwise_conv_s16()
[all …]
/cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_depthwise_conv_3x3_s8/
Dtest_arm_depthwise_conv_3x3_s8.c35 cmsis_nn_dw_conv_params dw_conv_params; in depthwise_kernel_3x3_arm_depthwise_conv_3x3_s8() local
57 dw_conv_params.padding.w = DEPTHWISE_KERNEL_3X3_PAD_X; in depthwise_kernel_3x3_arm_depthwise_conv_3x3_s8()
58 dw_conv_params.padding.h = DEPTHWISE_KERNEL_3X3_PAD_Y; in depthwise_kernel_3x3_arm_depthwise_conv_3x3_s8()
59 dw_conv_params.stride.w = DEPTHWISE_KERNEL_3X3_STRIDE_X; in depthwise_kernel_3x3_arm_depthwise_conv_3x3_s8()
60 dw_conv_params.stride.h = DEPTHWISE_KERNEL_3X3_STRIDE_Y; in depthwise_kernel_3x3_arm_depthwise_conv_3x3_s8()
61 dw_conv_params.dilation.w = DEPTHWISE_KERNEL_3X3_DILATION_X; in depthwise_kernel_3x3_arm_depthwise_conv_3x3_s8()
62 dw_conv_params.dilation.h = DEPTHWISE_KERNEL_3X3_DILATION_Y; in depthwise_kernel_3x3_arm_depthwise_conv_3x3_s8()
64 dw_conv_params.ch_mult = DEPTHWISE_KERNEL_3X3_CH_MULT; in depthwise_kernel_3x3_arm_depthwise_conv_3x3_s8()
66 dw_conv_params.input_offset = DEPTHWISE_KERNEL_3X3_INPUT_OFFSET; in depthwise_kernel_3x3_arm_depthwise_conv_3x3_s8()
67 dw_conv_params.output_offset = DEPTHWISE_KERNEL_3X3_OUTPUT_OFFSET; in depthwise_kernel_3x3_arm_depthwise_conv_3x3_s8()
[all …]
/cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_depthwise_conv_s8/
Dtest_arm_depthwise_conv_s8.c35 cmsis_nn_dw_conv_params dw_conv_params; in depthwise_2_arm_depthwise_conv_s8() local
57 dw_conv_params.padding.w = DEPTHWISE_2_PAD_X; in depthwise_2_arm_depthwise_conv_s8()
58 dw_conv_params.padding.h = DEPTHWISE_2_PAD_Y; in depthwise_2_arm_depthwise_conv_s8()
59 dw_conv_params.stride.w = DEPTHWISE_2_STRIDE_X; in depthwise_2_arm_depthwise_conv_s8()
60 dw_conv_params.stride.h = DEPTHWISE_2_STRIDE_Y; in depthwise_2_arm_depthwise_conv_s8()
61 dw_conv_params.dilation.w = DEPTHWISE_2_DILATION_X; in depthwise_2_arm_depthwise_conv_s8()
62 dw_conv_params.dilation.h = DEPTHWISE_2_DILATION_Y; in depthwise_2_arm_depthwise_conv_s8()
64 dw_conv_params.ch_mult = DEPTHWISE_2_CH_MULT; in depthwise_2_arm_depthwise_conv_s8()
66 dw_conv_params.input_offset = DEPTHWISE_2_INPUT_OFFSET; in depthwise_2_arm_depthwise_conv_s8()
67 dw_conv_params.output_offset = DEPTHWISE_2_OUTPUT_OFFSET; in depthwise_2_arm_depthwise_conv_s8()
[all …]
/cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_depthwise_conv_s8_opt/
Dtest_arm_depthwise_conv_s8_opt.c38 cmsis_nn_dw_conv_params dw_conv_params; in basic_arm_depthwise_conv_s8_opt() local
59 dw_conv_params.padding.w = BASIC_PAD_X; in basic_arm_depthwise_conv_s8_opt()
60 dw_conv_params.padding.h = BASIC_PAD_Y; in basic_arm_depthwise_conv_s8_opt()
61 dw_conv_params.stride.w = BASIC_STRIDE_X; in basic_arm_depthwise_conv_s8_opt()
62 dw_conv_params.stride.h = BASIC_STRIDE_Y; in basic_arm_depthwise_conv_s8_opt()
63 dw_conv_params.dilation.w = BASIC_DILATION_X; in basic_arm_depthwise_conv_s8_opt()
64 dw_conv_params.dilation.h = BASIC_DILATION_Y; in basic_arm_depthwise_conv_s8_opt()
66 dw_conv_params.ch_mult = 1; in basic_arm_depthwise_conv_s8_opt()
68 dw_conv_params.input_offset = BASIC_INPUT_OFFSET; in basic_arm_depthwise_conv_s8_opt()
69 dw_conv_params.output_offset = BASIC_OUTPUT_OFFSET; in basic_arm_depthwise_conv_s8_opt()
[all …]
/cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_ds_cnn_l_s8/
Dtest_arm_ds_cnn_l_s8.c72 cmsis_nn_dw_conv_params dw_conv_params; in ds_cnn_l_s8_get_buffer_size() local
73 dw_conv_params.activation.min = DEPTHWISE_CONV_2D_2_OUT_ACTIVATION_MIN; in ds_cnn_l_s8_get_buffer_size()
74 dw_conv_params.activation.max = DEPTHWISE_CONV_2D_2_OUT_ACTIVATION_MAX; in ds_cnn_l_s8_get_buffer_size()
75 dw_conv_params.ch_mult = 1; in ds_cnn_l_s8_get_buffer_size()
76 dw_conv_params.dilation.h = DEPTHWISE_CONV_2D_2_DILATION_H; in ds_cnn_l_s8_get_buffer_size()
77 dw_conv_params.dilation.w = DEPTHWISE_CONV_2D_2_DILATION_W; in ds_cnn_l_s8_get_buffer_size()
78 dw_conv_params.input_offset = DEPTHWISE_CONV_2D_2_INPUT_OFFSET; in ds_cnn_l_s8_get_buffer_size()
79 dw_conv_params.output_offset = DEPTHWISE_CONV_2D_2_OUTPUT_OFFSET; in ds_cnn_l_s8_get_buffer_size()
80 dw_conv_params.padding.h = DEPTHWISE_CONV_2D_2_PAD_H; in ds_cnn_l_s8_get_buffer_size()
81 dw_conv_params.padding.w = DEPTHWISE_CONV_2D_2_PAD_W; in ds_cnn_l_s8_get_buffer_size()
[all …]
/cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_ds_cnn_s_s8/
Dtest_arm_ds_cnn_s_s8.c73 cmsis_nn_dw_conv_params dw_conv_params; in ds_cnn_s_s8_get_buffer_size() local
74 dw_conv_params.activation.min = DEPTHWISE_CONV_2D_2_OUT_ACTIVATION_MIN; in ds_cnn_s_s8_get_buffer_size()
75 dw_conv_params.activation.max = DEPTHWISE_CONV_2D_2_OUT_ACTIVATION_MAX; in ds_cnn_s_s8_get_buffer_size()
76 dw_conv_params.ch_mult = 1; in ds_cnn_s_s8_get_buffer_size()
77 dw_conv_params.dilation.h = DEPTHWISE_CONV_2D_2_DILATION_H; in ds_cnn_s_s8_get_buffer_size()
78 dw_conv_params.dilation.w = DEPTHWISE_CONV_2D_2_DILATION_W; in ds_cnn_s_s8_get_buffer_size()
79 dw_conv_params.input_offset = DEPTHWISE_CONV_2D_2_INPUT_OFFSET; in ds_cnn_s_s8_get_buffer_size()
80 dw_conv_params.output_offset = DEPTHWISE_CONV_2D_2_OUTPUT_OFFSET; in ds_cnn_s_s8_get_buffer_size()
81 dw_conv_params.padding.h = DEPTHWISE_CONV_2D_2_PAD_H; in ds_cnn_s_s8_get_buffer_size()
82 dw_conv_params.padding.w = DEPTHWISE_CONV_2D_2_PAD_W; in ds_cnn_s_s8_get_buffer_size()
[all …]
/cmsis-nn-3.6.0-3.5.0/Include/
Darm_nnfunctions.h649 const cmsis_nn_dw_conv_params *dw_conv_params,
673 int32_t arm_depthwise_conv_wrapper_s8_get_buffer_size(const cmsis_nn_dw_conv_params *dw_conv_params,
686 …_t arm_depthwise_conv_wrapper_s8_get_buffer_size_dsp(const cmsis_nn_dw_conv_params *dw_conv_params,
699 …_t arm_depthwise_conv_wrapper_s8_get_buffer_size_mve(const cmsis_nn_dw_conv_params *dw_conv_params,
734 const cmsis_nn_dw_conv_params *dw_conv_params,
775 const cmsis_nn_dw_conv_params *dw_conv_params,
820 const cmsis_nn_dw_conv_params *dw_conv_params,
844 …t32_t arm_depthwise_conv_wrapper_s16_get_buffer_size(const cmsis_nn_dw_conv_params *dw_conv_params,
857 …t arm_depthwise_conv_wrapper_s16_get_buffer_size_dsp(const cmsis_nn_dw_conv_params *dw_conv_params,
870 …t arm_depthwise_conv_wrapper_s16_get_buffer_size_mve(const cmsis_nn_dw_conv_params *dw_conv_params,
[all …]
Darm_nnsupportfunctions.h43 #define USE_FAST_DW_CONV_S16_FUNCTION(dw_conv_params, filter_dims, input_dims) … argument
44 …(dw_conv_params->ch_mult == 1 && dw_conv_params->dilation.w == 1 && dw_conv_params->dilation.h == …