| /cmsis-nn-3.6.0-3.5.0/Source/ConvolutionFunctions/ |
| D | arm_convolve_get_buffer_sizes_s16.c | 43 __STATIC_INLINE int32_t arm_convolve_fast_s16_get_buffer_size_dsp(const cmsis_nn_dims *input_dims, in arm_convolve_fast_s16_get_buffer_size_dsp() argument
46 return (2 * input_dims->c * filter_dims->w * filter_dims->h) * (int32_t)sizeof(int16_t); in arm_convolve_fast_s16_get_buffer_size_dsp()
49 int32_t arm_convolve_fast_s16_get_buffer_size(const cmsis_nn_dims *input_dims, const cmsis_nn_dims … in arm_convolve_fast_s16_get_buffer_size() argument
52 return arm_convolve_fast_s16_get_buffer_size_dsp(input_dims, filter_dims); in arm_convolve_fast_s16_get_buffer_size()
54 (void)input_dims; in arm_convolve_fast_s16_get_buffer_size()
60 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
62 (void)input_dims; in arm_convolve_s16_get_buffer_size()
75 const cmsis_nn_dims *input_dims, in arm_convolve_wrapper_s16_get_buffer_size() argument
81 …return arm_convolve_wrapper_s16_get_buffer_size_dsp(conv_params, input_dims, filter_dims, output_d… in arm_convolve_wrapper_s16_get_buffer_size()
87 return arm_convolve_s16_get_buffer_size(input_dims, filter_dims); in arm_convolve_wrapper_s16_get_buffer_size()
[all …]
|
| D | arm_depthwise_conv_get_buffer_sizes_s16.c | 42 …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 …]
|
| D | arm_convolve_get_buffer_sizes_s8.c | 43 __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
46 int32_t col_length = input_dims->c * filter_dims->w * filter_dims->h; in arm_convolve_s8_get_buffer_size_mve()
54 __STATIC_INLINE int32_t arm_convolve_1_x_n_s8_get_buffer_size_mve(const cmsis_nn_dims *input_dims, in arm_convolve_1_x_n_s8_get_buffer_size_mve() argument
57 (void)input_dims; in arm_convolve_1_x_n_s8_get_buffer_size_mve()
62 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
65 return arm_convolve_s8_get_buffer_size_mve(input_dims, filter_dims); in arm_convolve_s8_get_buffer_size()
67 return (2 * input_dims->c * filter_dims->w * filter_dims->h) * (int32_t)sizeof(int16_t); in arm_convolve_s8_get_buffer_size()
71 int32_t arm_convolve_1_x_n_s8_get_buffer_size(const cmsis_nn_dims *input_dims, const cmsis_nn_dims … in arm_convolve_1_x_n_s8_get_buffer_size() argument
74 return arm_convolve_s8_get_buffer_size(input_dims, filter_dims); in arm_convolve_1_x_n_s8_get_buffer_size()
76 return arm_convolve_1_x_n_s8_get_buffer_size_mve(input_dims, filter_dims); in arm_convolve_1_x_n_s8_get_buffer_size()
[all …]
|
| D | arm_depthwise_conv_get_buffer_sizes_s8.c | 43 …TATIC_INLINE int32_t arm_depthwise_conv_s8_opt_get_buffer_size_mve(const cmsis_nn_dims *input_dims, in arm_depthwise_conv_s8_opt_get_buffer_size_mve() argument
46 (void)input_dims; in arm_depthwise_conv_s8_opt_get_buffer_size_mve()
50 …TATIC_INLINE int32_t arm_depthwise_conv_s8_opt_get_buffer_size_dsp(const cmsis_nn_dims *input_dims, in arm_depthwise_conv_s8_opt_get_buffer_size_dsp() argument
53 return (input_dims->c * filter_dims->w * filter_dims->h) * sizeof(int16_t); in arm_depthwise_conv_s8_opt_get_buffer_size_dsp()
56 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
59 return arm_depthwise_conv_s8_opt_get_buffer_size_mve(input_dims, filter_dims); in arm_depthwise_conv_s8_opt_get_buffer_size()
61 return arm_depthwise_conv_s8_opt_get_buffer_size_dsp(input_dims, filter_dims); in arm_depthwise_conv_s8_opt_get_buffer_size()
63 (void)input_dims; in arm_depthwise_conv_s8_opt_get_buffer_size()
70 const cmsis_nn_dims *input_dims, 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()
[all …]
|
| D | arm_convolve_wrapper_s8.c | 53 const cmsis_nn_dims *input_dims, in arm_convolve_wrapper_s8() argument
70 input_dims, in arm_convolve_wrapper_s8()
84 input_dims, in arm_convolve_wrapper_s8()
94 else if ((input_dims->h == 1) && conv_params->dilation.w == 1 && (filter_dims->h == 1) && in arm_convolve_wrapper_s8()
95 ((conv_params->stride.w * input_dims->c) % 4 == 0)) in arm_convolve_wrapper_s8()
100 input_dims, in arm_convolve_wrapper_s8()
114 input_dims, in arm_convolve_wrapper_s8()
|
| D | arm_convolve_wrapper_s16.c | 53 const cmsis_nn_dims *input_dims, in arm_convolve_wrapper_s16() argument
63 if (filter_dims->w * filter_dims->h * input_dims->c < 512 && in arm_convolve_wrapper_s16()
69 input_dims, in arm_convolve_wrapper_s16()
83 input_dims, in arm_convolve_wrapper_s16()
96 input_dims, in arm_convolve_wrapper_s16()
|
| D | arm_depthwise_conv_wrapper_s16.c | 53 const cmsis_nn_dims *input_dims, 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()
69 input_dims, in arm_depthwise_conv_wrapper_s16()
83 input_dims, in arm_depthwise_conv_wrapper_s16()
|
| D | arm_convolve_1x1_s8_fast.c | 53 const cmsis_nn_dims *input_dims, in arm_convolve_1x1_s8_fast() argument
72 const int32_t lhs_rows = input_dims->w * input_dims->h * input_dims->n; in arm_convolve_1x1_s8_fast()
74 const int32_t rhs_cols = input_dims->c; in arm_convolve_1x1_s8_fast()
|
| D | arm_depthwise_conv_wrapper_s8.c | 52 const cmsis_nn_dims *input_dims, 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()
72 input_dims, in arm_depthwise_conv_wrapper_s8()
87 input_dims, in arm_depthwise_conv_wrapper_s8()
102 input_dims, in arm_depthwise_conv_wrapper_s8()
|
| /cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_max_pool_s8/ |
| D | test_arm_max_pool_s8.c | 41 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-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_convolve_fast_s16/ |
| D | test_arm_convolve_fast_s16.c | 36 cmsis_nn_dims input_dims; in int16xint8_arm_convolve_fast_s16() local
47 input_dims.n = INT16XINT8_INPUT_BATCHES; in int16xint8_arm_convolve_fast_s16()
48 input_dims.w = INT16XINT8_INPUT_W; in int16xint8_arm_convolve_fast_s16()
49 input_dims.h = INT16XINT8_INPUT_H; in int16xint8_arm_convolve_fast_s16()
50 input_dims.c = INT16XINT8_IN_CH; in int16xint8_arm_convolve_fast_s16()
71 …int buf_size = arm_convolve_wrapper_s16_get_buffer_size(&conv_params, &input_dims, &filter_dims, &… in int16xint8_arm_convolve_fast_s16()
77 &input_dims, in int16xint8_arm_convolve_fast_s16()
96 buf_size = arm_convolve_fast_s16_get_buffer_size(&input_dims, &filter_dims); in int16xint8_arm_convolve_fast_s16()
102 &input_dims, in int16xint8_arm_convolve_fast_s16()
130 cmsis_nn_dims input_dims; in requantize_s64_arm_convolve_fast_s16() local
[all …]
|
| /cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_max_pool_s16/ |
| D | test_arm_max_pool_s16.c | 36 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-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_depthwise_conv_s16/ |
| D | test_arm_depthwise_conv_s16.c | 35 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-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_convolve_1x1_s8_fast/ |
| D | test_arm_convolve_1x1_s8_fast.c | 38 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-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_convolve_s16/ |
| D | test_arm_convolve_s16.c | 39 cmsis_nn_dims input_dims; in int16xint8_arm_convolve_s16() local
50 input_dims.n = INT16XINT8_INPUT_BATCHES; in int16xint8_arm_convolve_s16()
51 input_dims.w = INT16XINT8_INPUT_W; in int16xint8_arm_convolve_s16()
52 input_dims.h = INT16XINT8_INPUT_H; in int16xint8_arm_convolve_s16()
53 input_dims.c = INT16XINT8_IN_CH; in int16xint8_arm_convolve_s16()
74 int buf_size = arm_convolve_s16_get_buffer_size(&input_dims, &filter_dims); in int16xint8_arm_convolve_s16()
80 &input_dims, in int16xint8_arm_convolve_s16()
98 …buf_size = arm_convolve_wrapper_s16_get_buffer_size(&conv_params, &input_dims, &filter_dims, &outp… in int16xint8_arm_convolve_s16()
104 &input_dims, in int16xint8_arm_convolve_s16()
129 cmsis_nn_dims input_dims; in requantize_s64_arm_convolve_s16() local
[all …]
|
| /cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_convolve_s8/ |
| D | test_arm_convolve_s8.c | 47 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()
82 int32_t buf_size = arm_convolve_s8_get_buffer_size(&input_dims, &filter_dims); in basic_arm_convolve_s8()
89 &input_dims, in basic_arm_convolve_s8()
108 …buf_size = arm_convolve_wrapper_s8_get_buffer_size(&conv_params, &input_dims, &filter_dims, &outpu… in basic_arm_convolve_s8()
115 &input_dims, in basic_arm_convolve_s8()
141 cmsis_nn_dims input_dims; in stride2pad1_arm_convolve_s8() local
[all …]
|
| /cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_convolve_1_x_n_s8/ |
| D | test_arm_convolve_1_x_n_s8.c | 39 cmsis_nn_dims input_dims; in conv_1_x_n_1_arm_convolve_s8() local
50 input_dims.n = CONV_1_X_N_1_INPUT_BATCHES; in conv_1_x_n_1_arm_convolve_s8()
51 input_dims.w = CONV_1_X_N_1_INPUT_W; in conv_1_x_n_1_arm_convolve_s8()
52 input_dims.h = CONV_1_X_N_1_INPUT_H; in conv_1_x_n_1_arm_convolve_s8()
53 input_dims.c = CONV_1_X_N_1_IN_CH; in conv_1_x_n_1_arm_convolve_s8()
74 int32_t buf_size = arm_convolve_1_x_n_s8_get_buffer_size(&input_dims, &filter_dims); in conv_1_x_n_1_arm_convolve_s8()
81 &input_dims, in conv_1_x_n_1_arm_convolve_s8()
99 buf_size = arm_convolve_s8_get_buffer_size(&input_dims, &filter_dims); in conv_1_x_n_1_arm_convolve_s8()
105 &input_dims, in conv_1_x_n_1_arm_convolve_s8()
130 cmsis_nn_dims input_dims; in conv_1_x_n_2_arm_convolve_s8() local
[all …]
|
| /cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_avgpool_s8/ |
| D | test_arm_avgpool_s8.c | 37 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-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_depthwise_conv_fast_s16/ |
| D | test_arm_depthwise_conv_fast_s16.c | 42 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-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_depthwise_conv_s8_opt/ |
| D | test_arm_depthwise_conv_s8_opt.c | 40 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-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_avgpool_s16/ |
| D | test_arm_avgpool_s16.c | 35 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-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_depthwise_conv_3x3_s8/ |
| D | test_arm_depthwise_conv_3x3_s8.c | 37 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-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_depthwise_conv_s8/ |
| D | test_arm_depthwise_conv_s8.c | 37 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-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_fully_connected_s8/ |
| D | test_arm_fully_connected_s8.c | 38 cmsis_nn_dims input_dims; in fully_connected_arm_fully_connected_s8() local
49 input_dims.n = FULLY_CONNECTED_INPUT_BATCHES; in fully_connected_arm_fully_connected_s8()
50 input_dims.w = FULLY_CONNECTED_INPUT_W; in fully_connected_arm_fully_connected_s8()
51 input_dims.h = FULLY_CONNECTED_INPUT_H; in fully_connected_arm_fully_connected_s8()
52 input_dims.c = FULLY_CONNECTED_IN_CH; in fully_connected_arm_fully_connected_s8()
74 &input_dims, in fully_connected_arm_fully_connected_s8()
100 cmsis_nn_dims input_dims; in fully_connected_mve_0_arm_fully_connected_s8() local
109 input_dims.n = FULLY_CONNECTED_MVE_0_INPUT_BATCHES; in fully_connected_mve_0_arm_fully_connected_s8()
110 input_dims.w = FULLY_CONNECTED_MVE_0_INPUT_W; in fully_connected_mve_0_arm_fully_connected_s8()
111 input_dims.h = FULLY_CONNECTED_MVE_0_INPUT_H; in fully_connected_mve_0_arm_fully_connected_s8()
[all …]
|
| /cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_fully_connected_s16/ |
| D | test_arm_fully_connected_s16.c | 36 cmsis_nn_dims input_dims; in fully_connected_int16_arm_fully_connected_s16() local
47 input_dims.n = FULLY_CONNECTED_INT16_INPUT_BATCHES; in fully_connected_int16_arm_fully_connected_s16()
48 input_dims.w = FULLY_CONNECTED_INT16_INPUT_W; in fully_connected_int16_arm_fully_connected_s16()
49 input_dims.h = FULLY_CONNECTED_INT16_INPUT_H; in fully_connected_int16_arm_fully_connected_s16()
50 input_dims.c = FULLY_CONNECTED_INT16_IN_CH; in fully_connected_int16_arm_fully_connected_s16()
74 &input_dims, in fully_connected_int16_arm_fully_connected_s16()
101 cmsis_nn_dims input_dims; in fully_connected_int16_big_arm_fully_connected_s16() local
112 input_dims.n = FULLY_CONNECTED_INT16_BIG_INPUT_BATCHES; in fully_connected_int16_big_arm_fully_connected_s16()
113 input_dims.w = FULLY_CONNECTED_INT16_BIG_INPUT_W; in fully_connected_int16_big_arm_fully_connected_s16()
114 input_dims.h = FULLY_CONNECTED_INT16_BIG_INPUT_H; in fully_connected_int16_big_arm_fully_connected_s16()
[all …]
|
