Home
last modified time | relevance | path

Searched refs:input_dims (Results 1 – 25 of 68) sorted by relevance

123

/cmsis-nn-latest/Source/ConvolutionFunctions/
Darm_depthwise_conv_get_buffer_sizes_s16.c42 …TIC_INLINE int32_t arm_depthwise_conv_fast_s16_get_buffer_size_mve(const cmsis_nn_dims *input_dims, in arm_depthwise_conv_fast_s16_get_buffer_size_mve() argument
46 return 4 * input_dims->c * filter_dims->w * filter_dims->h * sizeof(int16_t) + 8; in arm_depthwise_conv_fast_s16_get_buffer_size_mve()
49 …TIC_INLINE int32_t arm_depthwise_conv_fast_s16_get_buffer_size_dsp(const cmsis_nn_dims *input_dims, in arm_depthwise_conv_fast_s16_get_buffer_size_dsp() argument
52 return input_dims->c * filter_dims->w * filter_dims->h * sizeof(int16_t); in arm_depthwise_conv_fast_s16_get_buffer_size_dsp()
55 int32_t arm_depthwise_conv_fast_s16_get_buffer_size(const cmsis_nn_dims *input_dims, const cmsis_nn… in arm_depthwise_conv_fast_s16_get_buffer_size() argument
59 return arm_depthwise_conv_fast_s16_get_buffer_size_mve(input_dims, filter_dims); in arm_depthwise_conv_fast_s16_get_buffer_size()
61 return arm_depthwise_conv_fast_s16_get_buffer_size_dsp(input_dims, filter_dims); in arm_depthwise_conv_fast_s16_get_buffer_size()
64 (void)input_dims; in arm_depthwise_conv_fast_s16_get_buffer_size()
71 const cmsis_nn_dims *input_dims, 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()
[all …]
Darm_depthwise_conv_get_buffer_sizes_s4.c43 int32_t arm_depthwise_conv_s4_opt_get_buffer_size(const cmsis_nn_dims *input_dims, const cmsis_nn_d… in arm_depthwise_conv_s4_opt_get_buffer_size() argument
46 return arm_depthwise_conv_s8_opt_get_buffer_size_mve(input_dims, filter_dims); in arm_depthwise_conv_s4_opt_get_buffer_size()
48 return arm_depthwise_conv_s8_opt_get_buffer_size_dsp(input_dims, filter_dims); in arm_depthwise_conv_s4_opt_get_buffer_size()
53 const cmsis_nn_dims *input_dims, in arm_depthwise_conv_wrapper_s4_get_buffer_size() argument
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()
62 size = arm_depthwise_conv_s4_opt_get_buffer_size(input_dims, filter_dims); in arm_depthwise_conv_wrapper_s4_get_buffer_size()
69 const cmsis_nn_dims *input_dims, in arm_depthwise_conv_wrapper_s4_get_buffer_size_dsp() argument
73 …return arm_depthwise_conv_wrapper_s4_get_buffer_size(dw_conv_params, input_dims, filter_dims, outp… in arm_depthwise_conv_wrapper_s4_get_buffer_size_dsp()
77 const cmsis_nn_dims *input_dims, in arm_depthwise_conv_wrapper_s4_get_buffer_size_mve() argument
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()
[all …]
Darm_depthwise_conv_get_buffer_sizes_s8.c43 int32_t arm_depthwise_conv_s8_opt_get_buffer_size_mve(const cmsis_nn_dims *input_dims, const cmsis_… in arm_depthwise_conv_s8_opt_get_buffer_size_mve() argument
45 (void)input_dims; in arm_depthwise_conv_s8_opt_get_buffer_size_mve()
49 int32_t arm_depthwise_conv_s8_opt_get_buffer_size_dsp(const cmsis_nn_dims *input_dims, const cmsis_… in arm_depthwise_conv_s8_opt_get_buffer_size_dsp() argument
51 return (input_dims->c * filter_dims->w * filter_dims->h) * sizeof(int16_t); in arm_depthwise_conv_s8_opt_get_buffer_size_dsp()
54 int32_t arm_depthwise_conv_s8_opt_get_buffer_size(const cmsis_nn_dims *input_dims, const cmsis_nn_d… in arm_depthwise_conv_s8_opt_get_buffer_size() argument
57 return arm_depthwise_conv_s8_opt_get_buffer_size_mve(input_dims, filter_dims); in arm_depthwise_conv_s8_opt_get_buffer_size()
59 return arm_depthwise_conv_s8_opt_get_buffer_size_dsp(input_dims, filter_dims); in arm_depthwise_conv_s8_opt_get_buffer_size()
61 (void)input_dims; in arm_depthwise_conv_s8_opt_get_buffer_size()
68 const cmsis_nn_dims *input_dims, in arm_depthwise_conv_wrapper_s8_get_buffer_size() argument
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()
[all …]
Darm_convolve_get_buffer_sizes_s8.c44 __STATIC_INLINE int32_t arm_convolve_s8_get_buffer_size_mve(const cmsis_nn_dims *input_dims, in arm_convolve_s8_get_buffer_size_mve() argument
47 int32_t col_length = input_dims->c * filter_dims->w * filter_dims->h; in arm_convolve_s8_get_buffer_size_mve()
56 const cmsis_nn_dims *input_dims, in arm_convolve_1_x_n_s8_get_buffer_size_mve() argument
60 const int32_t input_x = input_dims->w; in arm_convolve_1_x_n_s8_get_buffer_size_mve()
74 return arm_convolve_s8_get_buffer_size_mve(input_dims, filter_dims); in arm_convolve_1_x_n_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()
86 int32_t arm_convolve_s8_get_buffer_size(const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filte… in arm_convolve_s8_get_buffer_size() argument
[all …]
Darm_convolve_get_buffer_sizes_s16.c43 __STATIC_INLINE int32_t arm_convolve_s16_get_buffer_size_mve(const cmsis_nn_dims *input_dims, in arm_convolve_s16_get_buffer_size_mve() argument
46 int32_t col_length = input_dims->c * filter_dims->w * filter_dims->h; in arm_convolve_s16_get_buffer_size_mve()
54 int32_t arm_convolve_s16_get_buffer_size(const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filt… in arm_convolve_s16_get_buffer_size() argument
57 return arm_convolve_s16_get_buffer_size_mve(input_dims, filter_dims); in arm_convolve_s16_get_buffer_size()
59 return (2 * input_dims->c * filter_dims->w * filter_dims->h) * (int32_t)sizeof(int16_t); in arm_convolve_s16_get_buffer_size()
71 const cmsis_nn_dims *input_dims, in arm_convolve_wrapper_s16_get_buffer_size() argument
78 return arm_convolve_s16_get_buffer_size(input_dims, filter_dims); in arm_convolve_wrapper_s16_get_buffer_size()
82 const cmsis_nn_dims *input_dims, 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()
90 const cmsis_nn_dims *input_dims, in arm_convolve_wrapper_s16_get_buffer_size_mve() argument
[all …]
Darm_transpose_conv_get_buffer_sizes_s8.c50 int32_t arm_transpose_conv_s8_get_buffer_size(const cmsis_nn_dims *input_dims, in arm_transpose_conv_s8_get_buffer_size() argument
55 … filter_dims->w * filter_dims->h * input_dims->w * input_dims->h * out_dims->c * sizeof(int32_t); in arm_transpose_conv_s8_get_buffer_size()
59 int32_t arm_transpose_conv_s8_get_buffer_size_dsp(const cmsis_nn_dims *input_dims, in arm_transpose_conv_s8_get_buffer_size_dsp() argument
63 return arm_transpose_conv_s8_get_buffer_size(input_dims, filter_dims, output_dims); in arm_transpose_conv_s8_get_buffer_size_dsp()
66 int32_t arm_transpose_conv_s8_get_buffer_size_mve(const cmsis_nn_dims *input_dims, in arm_transpose_conv_s8_get_buffer_size_mve() argument
70 return arm_transpose_conv_s8_get_buffer_size(input_dims, filter_dims, output_dims); in arm_transpose_conv_s8_get_buffer_size_mve()
Darm_convolve_get_buffer_sizes_s4.c44 __STATIC_INLINE int32_t arm_convolve_s4_get_buffer_size_mve(const cmsis_nn_dims *input_dims, in arm_convolve_s4_get_buffer_size_mve() argument
47 int32_t col_length = input_dims->c * filter_dims->w * filter_dims->h; in arm_convolve_s4_get_buffer_size_mve()
56 const cmsis_nn_dims *input_dims, in arm_convolve_1_x_n_s4_get_buffer_size_mve() argument
60 const int32_t input_x = input_dims->w; in arm_convolve_1_x_n_s4_get_buffer_size_mve()
74 return arm_convolve_s4_get_buffer_size_mve(input_dims, filter_dims); in arm_convolve_1_x_n_s4_get_buffer_size_mve()
80 int32_t arm_convolve_s4_get_buffer_size(const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filte… in arm_convolve_s4_get_buffer_size() argument
82 const int32_t rhs_cols = filter_dims->w * filter_dims->h * input_dims->c; in arm_convolve_s4_get_buffer_size()
87 const cmsis_nn_dims *input_dims, in arm_convolve_1_x_n_s4_get_buffer_size() argument
95 return arm_convolve_s4_get_buffer_size(input_dims, filter_dims); 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()
[all …]
Darm_convolve_wrapper_s8.c53 const cmsis_nn_dims *input_dims, in arm_convolve_wrapper_s8() argument
64 (input_dims->c == filter_dims->c)) in arm_convolve_wrapper_s8()
71 input_dims, in arm_convolve_wrapper_s8()
85 input_dims, 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()
101 input_dims, in arm_convolve_wrapper_s8()
115 input_dims, in arm_convolve_wrapper_s8()
Darm_convolve_wrapper_s4.c53 const cmsis_nn_dims *input_dims, in arm_convolve_wrapper_s4() argument
70 input_dims, in arm_convolve_wrapper_s4()
84 input_dims, 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()
100 input_dims, in arm_convolve_wrapper_s4()
114 input_dims, in arm_convolve_wrapper_s4()
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_max_pool_s8/
Dtest_arm_max_pool_s8.c41 cmsis_nn_dims input_dims; in maxpooling_arm_max_pool_s8() local
47 input_dims.n = MAXPOOLING_INPUT_BATCHES; in maxpooling_arm_max_pool_s8()
48 input_dims.w = MAXPOOLING_INPUT_W; in maxpooling_arm_max_pool_s8()
49 input_dims.h = MAXPOOLING_INPUT_H; in maxpooling_arm_max_pool_s8()
50 input_dims.c = MAXPOOLING_IN_CH; in maxpooling_arm_max_pool_s8()
68 … arm_max_pool_s8(&ctx, &pool_params, &input_dims, input_data, &filter_dims, &output_dims, output); in maxpooling_arm_max_pool_s8()
82 cmsis_nn_dims input_dims; in maxpooling_1_arm_max_pool_s8() local
88 input_dims.n = MAXPOOLING_1_INPUT_BATCHES; in maxpooling_1_arm_max_pool_s8()
89 input_dims.w = MAXPOOLING_1_INPUT_W; in maxpooling_1_arm_max_pool_s8()
90 input_dims.h = MAXPOOLING_1_INPUT_H; in maxpooling_1_arm_max_pool_s8()
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_max_pool_s16/
Dtest_arm_max_pool_s16.c36 cmsis_nn_dims input_dims; in maxpool_int16_arm_max_pool_s16() local
42 input_dims.n = MAXPOOL_INT16_INPUT_BATCHES; in maxpool_int16_arm_max_pool_s16()
43 input_dims.w = MAXPOOL_INT16_INPUT_W; in maxpool_int16_arm_max_pool_s16()
44 input_dims.h = MAXPOOL_INT16_INPUT_H; in maxpool_int16_arm_max_pool_s16()
45 input_dims.c = MAXPOOL_INT16_IN_CH; in maxpool_int16_arm_max_pool_s16()
63 … arm_max_pool_s16(&ctx, &pool_params, &input_dims, input_data, &filter_dims, &output_dims, output); in maxpool_int16_arm_max_pool_s16()
77 cmsis_nn_dims input_dims; in maxpool_int16_1_arm_max_pool_s16() local
83 input_dims.n = MAXPOOL_INT16_1_INPUT_BATCHES; in maxpool_int16_1_arm_max_pool_s16()
84 input_dims.w = MAXPOOL_INT16_1_INPUT_W; in maxpool_int16_1_arm_max_pool_s16()
85 input_dims.h = MAXPOOL_INT16_1_INPUT_H; in maxpool_int16_1_arm_max_pool_s16()
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_depthwise_conv_s16/
Dtest_arm_depthwise_conv_s16.c35 cmsis_nn_dims input_dims; in dw_int16xint8_arm_depthwise_conv_s16() local
46 input_dims.n = DW_INT16XINT8_INPUT_BATCHES; in dw_int16xint8_arm_depthwise_conv_s16()
47 input_dims.w = DW_INT16XINT8_INPUT_W; in dw_int16xint8_arm_depthwise_conv_s16()
48 input_dims.h = DW_INT16XINT8_INPUT_H; in dw_int16xint8_arm_depthwise_conv_s16()
49 input_dims.c = DW_INT16XINT8_IN_CH; in dw_int16xint8_arm_depthwise_conv_s16()
78 &input_dims, in dw_int16xint8_arm_depthwise_conv_s16()
97 …arm_depthwise_conv_wrapper_s16_get_buffer_size(&dw_conv_params, &input_dims, &filter_dims, &output… in dw_int16xint8_arm_depthwise_conv_s16()
106 &input_dims, in dw_int16xint8_arm_depthwise_conv_s16()
132 cmsis_nn_dims input_dims; in dw_int16xint8_dilation_arm_depthwise_conv_s16() local
143 input_dims.n = DW_INT16XINT8_DILATION_INPUT_BATCHES; in dw_int16xint8_dilation_arm_depthwise_conv_s16()
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_avgpool_s8/
Dtest_arm_avgpool_s8.c37 cmsis_nn_dims input_dims; in avgpooling_arm_avgpool_s8() local
43 input_dims.n = AVGPOOLING_INPUT_BATCHES; in avgpooling_arm_avgpool_s8()
44 input_dims.w = AVGPOOLING_INPUT_W; in avgpooling_arm_avgpool_s8()
45 input_dims.h = AVGPOOLING_INPUT_H; in avgpooling_arm_avgpool_s8()
46 input_dims.c = AVGPOOLING_IN_CH; in avgpooling_arm_avgpool_s8()
65 … arm_avgpool_s8(&ctx, &pool_params, &input_dims, input_data, &filter_dims, &output_dims, output); in avgpooling_arm_avgpool_s8()
84 cmsis_nn_dims input_dims; in avgpooling_1_arm_avgpool_s8() local
90 input_dims.n = AVGPOOLING_1_INPUT_BATCHES; in avgpooling_1_arm_avgpool_s8()
91 input_dims.w = AVGPOOLING_1_INPUT_W; in avgpooling_1_arm_avgpool_s8()
92 input_dims.h = AVGPOOLING_1_INPUT_H; in avgpooling_1_arm_avgpool_s8()
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_convolve_1x1_s8_fast/
Dtest_arm_convolve_1x1_s8_fast.c38 cmsis_nn_dims input_dims; in kernel1x1_arm_convolve_1x1_s8_fast() local
46 input_dims.n = KERNEL1X1_INPUT_BATCHES; in kernel1x1_arm_convolve_1x1_s8_fast()
47 input_dims.h = KERNEL1X1_INPUT_H; in kernel1x1_arm_convolve_1x1_s8_fast()
48 input_dims.w = KERNEL1X1_INPUT_W; in kernel1x1_arm_convolve_1x1_s8_fast()
49 input_dims.c = KERNEL1X1_IN_CH; in kernel1x1_arm_convolve_1x1_s8_fast()
76 const int32_t buf_size = arm_convolve_1x1_s8_fast_get_buffer_size(&input_dims); in kernel1x1_arm_convolve_1x1_s8_fast()
83 &input_dims, in kernel1x1_arm_convolve_1x1_s8_fast()
110 cmsis_nn_dims input_dims; in kernel1x1_stride_x_arm_convolve_1x1_s8() local
118 input_dims.n = KERNEL1X1_STRIDE_X_INPUT_BATCHES; in kernel1x1_stride_x_arm_convolve_1x1_s8()
119 input_dims.h = KERNEL1X1_STRIDE_X_INPUT_H; in kernel1x1_stride_x_arm_convolve_1x1_s8()
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_avgpool_s16/
Dtest_arm_avgpool_s16.c35 cmsis_nn_dims input_dims; in avgpooling_int16_arm_avgpool_s16() local
41 input_dims.n = AVGPOOLING_INT16_INPUT_BATCHES; in avgpooling_int16_arm_avgpool_s16()
42 input_dims.w = AVGPOOLING_INT16_INPUT_W; in avgpooling_int16_arm_avgpool_s16()
43 input_dims.h = AVGPOOLING_INT16_INPUT_H; in avgpooling_int16_arm_avgpool_s16()
44 input_dims.c = AVGPOOLING_INT16_IN_CH; in avgpooling_int16_arm_avgpool_s16()
63 … arm_avgpool_s16(&ctx, &pool_params, &input_dims, input_data, &filter_dims, &output_dims, output); in avgpooling_int16_arm_avgpool_s16()
82 cmsis_nn_dims input_dims; in avgpooling_int16_1_arm_avgpool_s16() local
88 input_dims.n = AVGPOOLING_INT16_1_INPUT_BATCHES; in avgpooling_int16_1_arm_avgpool_s16()
89 input_dims.w = AVGPOOLING_INT16_1_INPUT_W; in avgpooling_int16_1_arm_avgpool_s16()
90 input_dims.h = AVGPOOLING_INT16_1_INPUT_H; in avgpooling_int16_1_arm_avgpool_s16()
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_convolve_s16/
Dtest_arm_convolve_s16.c47 cmsis_nn_dims input_dims; in int16xint8_arm_convolve_s16() local
59 input_dims.n = INT16XINT8_INPUT_BATCHES; in int16xint8_arm_convolve_s16()
60 input_dims.w = INT16XINT8_INPUT_W; in int16xint8_arm_convolve_s16()
61 input_dims.h = INT16XINT8_INPUT_H; in int16xint8_arm_convolve_s16()
62 input_dims.c = INT16XINT8_IN_CH; in int16xint8_arm_convolve_s16()
83 int buf_size = arm_convolve_s16_get_buffer_size(&input_dims, &filter_dims); in int16xint8_arm_convolve_s16()
89 &input_dims, in int16xint8_arm_convolve_s16()
107 …buf_size = arm_convolve_wrapper_s16_get_buffer_size(&conv_params, &input_dims, &filter_dims, &outp… in int16xint8_arm_convolve_s16()
113 &input_dims, in int16xint8_arm_convolve_s16()
137 cmsis_nn_dims input_dims; in requantize_s64_arm_convolve_s16() local
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_convolve_1x1_s4_fast/
Dtest_arm_convolve_1x1_s4_fast.c40 cmsis_nn_dims input_dims; in kernel1x1_arm_convolve_1x1_s4_fast() local
48 input_dims.n = KERNEL1X1_INT4_INPUT_BATCHES; in kernel1x1_arm_convolve_1x1_s4_fast()
49 input_dims.h = KERNEL1X1_INT4_INPUT_H; in kernel1x1_arm_convolve_1x1_s4_fast()
50 input_dims.w = KERNEL1X1_INT4_INPUT_W; in kernel1x1_arm_convolve_1x1_s4_fast()
51 input_dims.c = KERNEL1X1_INT4_IN_CH; in kernel1x1_arm_convolve_1x1_s4_fast()
80 const int32_t buf_size = arm_convolve_1x1_s4_fast_get_buffer_size(&input_dims); in kernel1x1_arm_convolve_1x1_s4_fast()
87 &input_dims, in kernel1x1_arm_convolve_1x1_s4_fast()
114 cmsis_nn_dims input_dims; in kernel1x1_2_arm_convolve_1x1_s4_fast() local
122 input_dims.n = KERNEL1X1_INT4_2_INPUT_BATCHES; in kernel1x1_2_arm_convolve_1x1_s4_fast()
123 input_dims.h = KERNEL1X1_INT4_2_INPUT_H; in kernel1x1_2_arm_convolve_1x1_s4_fast()
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_convolve_s8/
Dtest_arm_convolve_s8.c47 cmsis_nn_dims input_dims; in basic_arm_convolve_s8() local
58 input_dims.n = BASIC_INPUT_BATCHES; in basic_arm_convolve_s8()
59 input_dims.w = BASIC_INPUT_W; in basic_arm_convolve_s8()
60 input_dims.h = BASIC_INPUT_H; in basic_arm_convolve_s8()
61 input_dims.c = BASIC_IN_CH; in basic_arm_convolve_s8()
83 int32_t buf_size = arm_convolve_s8_get_buffer_size(&input_dims, &filter_dims); in basic_arm_convolve_s8()
90 &input_dims, in basic_arm_convolve_s8()
109 …buf_size = arm_convolve_wrapper_s8_get_buffer_size(&conv_params, &input_dims, &filter_dims, &outpu… in basic_arm_convolve_s8()
116 &input_dims, in basic_arm_convolve_s8()
142 cmsis_nn_dims input_dims; in stride2pad1_arm_convolve_s8() local
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_depthwise_conv_fast_s16/
Dtest_arm_depthwise_conv_fast_s16.c42 cmsis_nn_dims input_dims; in dw_int16xint8_fast_arm_depthwise_conv_fast_s16() local
53 input_dims.n = DW_INT16XINT8_FAST_INPUT_BATCHES; in dw_int16xint8_fast_arm_depthwise_conv_fast_s16()
54 input_dims.w = DW_INT16XINT8_FAST_INPUT_W; in dw_int16xint8_fast_arm_depthwise_conv_fast_s16()
55 input_dims.h = DW_INT16XINT8_FAST_INPUT_H; in dw_int16xint8_fast_arm_depthwise_conv_fast_s16()
56 input_dims.c = DW_INT16XINT8_FAST_IN_CH; in dw_int16xint8_fast_arm_depthwise_conv_fast_s16()
79 int buf_size = arm_depthwise_conv_fast_s16_get_buffer_size(&input_dims, &filter_dims); in dw_int16xint8_fast_arm_depthwise_conv_fast_s16()
85 &input_dims, in dw_int16xint8_fast_arm_depthwise_conv_fast_s16()
106 …arm_depthwise_conv_wrapper_s16_get_buffer_size(&dw_conv_params, &input_dims, &filter_dims, &output… in dw_int16xint8_fast_arm_depthwise_conv_fast_s16()
113 &input_dims, in dw_int16xint8_fast_arm_depthwise_conv_fast_s16()
139 cmsis_nn_dims input_dims; in dw_int16xint8_fast_spill_arm_depthwise_conv_fast_s16() local
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_depthwise_conv_s8_opt/
Dtest_arm_depthwise_conv_s8_opt.c40 cmsis_nn_dims input_dims; in basic_arm_depthwise_conv_s8_opt() local
49 input_dims.n = BASIC_INPUT_BATCHES; in basic_arm_depthwise_conv_s8_opt()
50 input_dims.w = BASIC_INPUT_W; in basic_arm_depthwise_conv_s8_opt()
51 input_dims.h = BASIC_INPUT_H; in basic_arm_depthwise_conv_s8_opt()
52 input_dims.c = BASIC_IN_CH; in basic_arm_depthwise_conv_s8_opt()
75 ctx.size = arm_depthwise_conv_s8_opt_get_buffer_size(&input_dims, &filter_dims); in basic_arm_depthwise_conv_s8_opt()
88 &input_dims, in basic_arm_depthwise_conv_s8_opt()
107 …arm_depthwise_conv_wrapper_s8_get_buffer_size(&dw_conv_params, &input_dims, &filter_dims, &output_… in basic_arm_depthwise_conv_s8_opt()
116 &input_dims, in basic_arm_depthwise_conv_s8_opt()
142 cmsis_nn_dims input_dims; in depthwise_eq_in_out_ch_arm_depthwise_conv_s8_opt() local
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_convolve_s4/
Dtest_arm_convolve_s4.c53 cmsis_nn_dims input_dims; in basic_arm_convolve_s4() local
64 input_dims.n = BASIC_INT4_INPUT_BATCHES; in basic_arm_convolve_s4()
65 input_dims.w = BASIC_INT4_INPUT_W; in basic_arm_convolve_s4()
66 input_dims.h = BASIC_INT4_INPUT_H; in basic_arm_convolve_s4()
67 input_dims.c = BASIC_INT4_IN_CH; in basic_arm_convolve_s4()
88 int32_t buf_size = arm_convolve_s4_get_buffer_size(&input_dims, &filter_dims); in basic_arm_convolve_s4()
95 &input_dims, in basic_arm_convolve_s4()
114 …buf_size = arm_convolve_wrapper_s4_get_buffer_size(&conv_params, &input_dims, &filter_dims, &outpu… in basic_arm_convolve_s4()
121 &input_dims, in basic_arm_convolve_s4()
147 cmsis_nn_dims input_dims; in basic_2_arm_convolve_s4() local
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_depthwise_conv_s8/
Dtest_arm_depthwise_conv_s8.c37 cmsis_nn_dims input_dims; in depthwise_2_arm_depthwise_conv_s8() local
47 input_dims.n = DEPTHWISE_2_INPUT_BATCHES; in depthwise_2_arm_depthwise_conv_s8()
48 input_dims.w = DEPTHWISE_2_INPUT_W; in depthwise_2_arm_depthwise_conv_s8()
49 input_dims.h = DEPTHWISE_2_INPUT_H; in depthwise_2_arm_depthwise_conv_s8()
50 input_dims.c = DEPTHWISE_2_IN_CH; in depthwise_2_arm_depthwise_conv_s8()
79 &input_dims, in depthwise_2_arm_depthwise_conv_s8()
98 …arm_depthwise_conv_wrapper_s8_get_buffer_size(&dw_conv_params, &input_dims, &filter_dims, &output_… in depthwise_2_arm_depthwise_conv_s8()
107 &input_dims, in depthwise_2_arm_depthwise_conv_s8()
133 cmsis_nn_dims input_dims; in depthwise_mult_batches_arm_depthwise_conv_s8() local
143 input_dims.n = DEPTHWISE_MULT_BATCHES_INPUT_BATCHES; in depthwise_mult_batches_arm_depthwise_conv_s8()
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_depthwise_conv_3x3_s8/
Dtest_arm_depthwise_conv_3x3_s8.c37 cmsis_nn_dims input_dims; in depthwise_kernel_3x3_arm_depthwise_conv_3x3_s8() local
47 input_dims.n = DEPTHWISE_KERNEL_3X3_INPUT_BATCHES; in depthwise_kernel_3x3_arm_depthwise_conv_3x3_s8()
48 input_dims.w = DEPTHWISE_KERNEL_3X3_INPUT_W; in depthwise_kernel_3x3_arm_depthwise_conv_3x3_s8()
49 input_dims.h = DEPTHWISE_KERNEL_3X3_INPUT_H; in depthwise_kernel_3x3_arm_depthwise_conv_3x3_s8()
50 input_dims.c = DEPTHWISE_KERNEL_3X3_IN_CH; in depthwise_kernel_3x3_arm_depthwise_conv_3x3_s8()
79 &input_dims, in depthwise_kernel_3x3_arm_depthwise_conv_3x3_s8()
99 …arm_depthwise_conv_wrapper_s8_get_buffer_size(&dw_conv_params, &input_dims, &filter_dims, &output_… in depthwise_kernel_3x3_arm_depthwise_conv_3x3_s8()
113 &input_dims, in depthwise_kernel_3x3_arm_depthwise_conv_3x3_s8()
140 cmsis_nn_dims input_dims; in depthwise_kernel_3x3_arm_depthwise_conv_3x3_1_s8() local
149 input_dims.n = DEPTHWISE_KERNEL_3X3_INPUT_BATCHES; in depthwise_kernel_3x3_arm_depthwise_conv_3x3_1_s8()
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_fully_connected_s4/
Dtest_arm_fully_connected_s4.c40 cmsis_nn_dims input_dims; in fully_connected_int4_arm_fully_connected_s4() local
51 input_dims.n = FULLY_CONNECTED_INT4_INPUT_BATCHES; in fully_connected_int4_arm_fully_connected_s4()
52 input_dims.w = FULLY_CONNECTED_INT4_INPUT_W; in fully_connected_int4_arm_fully_connected_s4()
53 input_dims.h = FULLY_CONNECTED_INT4_INPUT_H; in fully_connected_int4_arm_fully_connected_s4()
54 input_dims.c = FULLY_CONNECTED_INT4_IN_CH; in fully_connected_int4_arm_fully_connected_s4()
76 &input_dims, in fully_connected_int4_arm_fully_connected_s4()
103 cmsis_nn_dims input_dims; in fully_connected_int4_arm_fully_connected_s4_2() local
114 input_dims.n = FULLY_CONNECTED_INT4_2_INPUT_BATCHES; in fully_connected_int4_arm_fully_connected_s4_2()
115 input_dims.w = FULLY_CONNECTED_INT4_2_INPUT_W; in fully_connected_int4_arm_fully_connected_s4_2()
116 input_dims.h = FULLY_CONNECTED_INT4_2_INPUT_H; in fully_connected_int4_arm_fully_connected_s4_2()
[all …]
/cmsis-nn-latest/Include/
Darm_nnfunctions.h95 const cmsis_nn_dims *input_dims,
119 const cmsis_nn_dims *input_dims,
133 const cmsis_nn_dims *input_dims,
146 const cmsis_nn_dims *input_dims,
180 const cmsis_nn_dims *input_dims,
204 const cmsis_nn_dims *input_dims,
217 const cmsis_nn_dims *input_dims,
230 const cmsis_nn_dims *input_dims,
265 const cmsis_nn_dims *input_dims,
289 const cmsis_nn_dims *input_dims,
[all …]

123