| /cmsis-nn-3.6.0-3.5.0/Source/ConvolutionFunctions/ |
| D | arm_convolve_get_buffer_sizes_s16.c | 77 const cmsis_nn_dims *output_dims) in arm_convolve_wrapper_s16_get_buffer_size() argument
81 …rn arm_convolve_wrapper_s16_get_buffer_size_dsp(conv_params, input_dims, filter_dims, output_dims); in arm_convolve_wrapper_s16_get_buffer_size()
84 (void)output_dims; in arm_convolve_wrapper_s16_get_buffer_size()
94 const cmsis_nn_dims *output_dims) in arm_convolve_wrapper_s16_get_buffer_size_dsp() argument
96 (void)output_dims; in arm_convolve_wrapper_s16_get_buffer_size_dsp()
113 const cmsis_nn_dims *output_dims) in arm_convolve_wrapper_s16_get_buffer_size_mve() argument
115 …return arm_convolve_wrapper_s16_get_buffer_size(conv_params, input_dims, filter_dims, output_dims); in arm_convolve_wrapper_s16_get_buffer_size_mve()
|
| D | arm_depthwise_conv_get_buffer_sizes_s16.c | 73 const cmsis_nn_dims *output_dims) in arm_depthwise_conv_wrapper_s16_get_buffer_size() argument
75 (void)output_dims; in arm_depthwise_conv_wrapper_s16_get_buffer_size()
90 const cmsis_nn_dims *output_dims) in arm_depthwise_conv_wrapper_s16_get_buffer_size_mve() argument
92 (void)output_dims; in arm_depthwise_conv_wrapper_s16_get_buffer_size_mve()
107 const cmsis_nn_dims *output_dims) in arm_depthwise_conv_wrapper_s16_get_buffer_size_dsp() argument
109 (void)output_dims; in arm_depthwise_conv_wrapper_s16_get_buffer_size_dsp()
|
| D | arm_convolve_wrapper_s16.c | 59 const cmsis_nn_dims *output_dims, in arm_convolve_wrapper_s16() argument
75 output_dims, in arm_convolve_wrapper_s16()
89 output_dims, in arm_convolve_wrapper_s16()
102 output_dims, in arm_convolve_wrapper_s16()
|
| D | arm_convolve_wrapper_s8.c | 59 const cmsis_nn_dims *output_dims, in arm_convolve_wrapper_s8() argument
76 output_dims, in arm_convolve_wrapper_s8()
90 output_dims, in arm_convolve_wrapper_s8()
106 output_dims, in arm_convolve_wrapper_s8()
120 output_dims, in arm_convolve_wrapper_s8()
|
| D | arm_depthwise_conv_wrapper_s16.c | 59 const cmsis_nn_dims *output_dims, in arm_depthwise_conv_wrapper_s16() argument
75 output_dims, in arm_depthwise_conv_wrapper_s16()
89 output_dims, in arm_depthwise_conv_wrapper_s16()
|
| D | arm_depthwise_conv_wrapper_s8.c | 58 const cmsis_nn_dims *output_dims, in arm_depthwise_conv_wrapper_s8() argument
78 output_dims, in arm_depthwise_conv_wrapper_s8()
93 output_dims, in arm_depthwise_conv_wrapper_s8()
108 output_dims, in arm_depthwise_conv_wrapper_s8()
|
| D | arm_convolve_1x1_s8.c | 59 const cmsis_nn_dims *output_dims, in arm_convolve_1x1_s8() argument
70 const int32_t lhs_rows = output_dims->w; in arm_convolve_1x1_s8()
71 const int32_t rhs_rows = output_dims->c; in arm_convolve_1x1_s8()
75 const int32_t output_inc = output_dims->w * rhs_rows; in arm_convolve_1x1_s8()
76 const int32_t output_h = output_dims->h; in arm_convolve_1x1_s8()
|
| D | arm_depthwise_conv_get_buffer_sizes_s8.c | 72 const cmsis_nn_dims *output_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()
95 const cmsis_nn_dims *output_dims) 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()
116 const cmsis_nn_dims *output_dims) in arm_depthwise_conv_wrapper_s8_get_buffer_size_mve() argument
120 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()
|
| D | arm_convolve_get_buffer_sizes_s8.c | 96 const cmsis_nn_dims *output_dims) in arm_convolve_wrapper_s8_get_buffer_size() argument
99 …urn arm_convolve_wrapper_s8_get_buffer_size_mve(conv_params, input_dims, filter_dims, output_dims); in arm_convolve_wrapper_s8_get_buffer_size()
101 (void)output_dims; in arm_convolve_wrapper_s8_get_buffer_size()
129 const cmsis_nn_dims *output_dims) in arm_convolve_wrapper_s8_get_buffer_size_mve() argument
131 (void)output_dims; in arm_convolve_wrapper_s8_get_buffer_size_mve()
158 const cmsis_nn_dims *output_dims) in arm_convolve_wrapper_s8_get_buffer_size_dsp() argument
160 … return arm_convolve_wrapper_s8_get_buffer_size(conv_params, input_dims, filter_dims, output_dims); in arm_convolve_wrapper_s8_get_buffer_size_dsp()
|
| /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 | 41 cmsis_nn_dims output_dims; in kernel1x1_arm_convolve_1x1_s8_fast() local
54 output_dims.n = KERNEL1X1_INPUT_BATCHES; in kernel1x1_arm_convolve_1x1_s8_fast()
55 output_dims.h = KERNEL1X1_OUTPUT_H; in kernel1x1_arm_convolve_1x1_s8_fast()
56 output_dims.w = KERNEL1X1_OUTPUT_W; 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()
89 &output_dims, in kernel1x1_arm_convolve_1x1_s8_fast()
113 cmsis_nn_dims output_dims; in kernel1x1_stride_x_arm_convolve_1x1_s8() local
128 output_dims.n = KERNEL1X1_STRIDE_X_INPUT_BATCHES; in kernel1x1_stride_x_arm_convolve_1x1_s8()
129 output_dims.h = KERNEL1X1_STRIDE_X_OUTPUT_H; in kernel1x1_stride_x_arm_convolve_1x1_s8()
[all …]
|
| /cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_max_pool_s8/ |
| D | test_arm_max_pool_s8.c | 43 cmsis_nn_dims output_dims; in maxpooling_arm_max_pool_s8() local
53 output_dims.w = MAXPOOLING_OUTPUT_W; in maxpooling_arm_max_pool_s8()
54 output_dims.h = MAXPOOLING_OUTPUT_H; in maxpooling_arm_max_pool_s8()
55 output_dims.c = MAXPOOLING_OUT_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()
84 cmsis_nn_dims output_dims; in maxpooling_1_arm_max_pool_s8() local
94 output_dims.w = MAXPOOLING_1_OUTPUT_W; in maxpooling_1_arm_max_pool_s8()
95 output_dims.h = MAXPOOLING_1_OUTPUT_H; in maxpooling_1_arm_max_pool_s8()
96 output_dims.c = MAXPOOLING_1_OUT_CH; in maxpooling_1_arm_max_pool_s8()
109 … arm_max_pool_s8(&ctx, &pool_params, &input_dims, input_data, &filter_dims, &output_dims, output); 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 | 39 cmsis_nn_dims output_dims; in int16xint8_arm_convolve_fast_s16() local
53 output_dims.w = INT16XINT8_OUTPUT_W; in int16xint8_arm_convolve_fast_s16()
54 output_dims.h = INT16XINT8_OUTPUT_H; in int16xint8_arm_convolve_fast_s16()
55 output_dims.c = INT16XINT8_OUT_CH; in int16xint8_arm_convolve_fast_s16()
71 … = arm_convolve_wrapper_s16_get_buffer_size(&conv_params, &input_dims, &filter_dims, &output_dims); in int16xint8_arm_convolve_fast_s16()
83 &output_dims, in int16xint8_arm_convolve_fast_s16()
108 &output_dims, in int16xint8_arm_convolve_fast_s16()
133 cmsis_nn_dims output_dims; in requantize_s64_arm_convolve_fast_s16() local
147 output_dims.w = REQUANTIZE_S64_OUTPUT_W; in requantize_s64_arm_convolve_fast_s16()
148 output_dims.h = REQUANTIZE_S64_OUTPUT_H; in requantize_s64_arm_convolve_fast_s16()
[all …]
|
| /cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_max_pool_s16/ |
| D | test_arm_max_pool_s16.c | 38 cmsis_nn_dims output_dims; in maxpool_int16_arm_max_pool_s16() local
48 output_dims.w = MAXPOOL_INT16_OUTPUT_W; in maxpool_int16_arm_max_pool_s16()
49 output_dims.h = MAXPOOL_INT16_OUTPUT_H; in maxpool_int16_arm_max_pool_s16()
50 output_dims.c = MAXPOOL_INT16_OUT_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()
79 cmsis_nn_dims output_dims; in maxpool_int16_1_arm_max_pool_s16() local
89 output_dims.w = MAXPOOL_INT16_1_OUTPUT_W; in maxpool_int16_1_arm_max_pool_s16()
90 output_dims.h = MAXPOOL_INT16_1_OUTPUT_H; in maxpool_int16_1_arm_max_pool_s16()
91 output_dims.c = MAXPOOL_INT16_1_OUT_CH; in maxpool_int16_1_arm_max_pool_s16()
104 … arm_max_pool_s16(&ctx, &pool_params, &input_dims, input_data, &filter_dims, &output_dims, output); 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 | 38 cmsis_nn_dims output_dims; in dw_int16xint8_arm_depthwise_conv_s16() local
52 output_dims.w = DW_INT16XINT8_OUTPUT_W; in dw_int16xint8_arm_depthwise_conv_s16()
53 output_dims.h = DW_INT16XINT8_OUTPUT_H; in dw_int16xint8_arm_depthwise_conv_s16()
54 output_dims.c = DW_INT16XINT8_OUT_CH; in dw_int16xint8_arm_depthwise_conv_s16()
84 &output_dims, in dw_int16xint8_arm_depthwise_conv_s16()
97 …pthwise_conv_wrapper_s16_get_buffer_size(&dw_conv_params, &input_dims, &filter_dims, &output_dims); in dw_int16xint8_arm_depthwise_conv_s16()
112 &output_dims, in dw_int16xint8_arm_depthwise_conv_s16()
135 cmsis_nn_dims output_dims; in dw_int16xint8_dilation_arm_depthwise_conv_s16() local
149 output_dims.w = DW_INT16XINT8_DILATION_OUTPUT_W; in dw_int16xint8_dilation_arm_depthwise_conv_s16()
150 output_dims.h = DW_INT16XINT8_DILATION_OUTPUT_H; in dw_int16xint8_dilation_arm_depthwise_conv_s16()
[all …]
|
| /cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_avgpool_s8/ |
| D | test_arm_avgpool_s8.c | 39 cmsis_nn_dims output_dims; in avgpooling_arm_avgpool_s8() local
49 output_dims.w = AVGPOOLING_OUTPUT_W; in avgpooling_arm_avgpool_s8()
50 output_dims.h = AVGPOOLING_OUTPUT_H; in avgpooling_arm_avgpool_s8()
51 output_dims.c = AVGPOOLING_OUT_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()
86 cmsis_nn_dims output_dims; in avgpooling_1_arm_avgpool_s8() local
96 output_dims.w = AVGPOOLING_1_OUTPUT_W; in avgpooling_1_arm_avgpool_s8()
97 output_dims.h = AVGPOOLING_1_OUTPUT_H; in avgpooling_1_arm_avgpool_s8()
98 output_dims.c = AVGPOOLING_1_OUT_CH; in avgpooling_1_arm_avgpool_s8()
112 … arm_avgpool_s8(&ctx, &pool_params, &input_dims, input_data, &filter_dims, &output_dims, output); in avgpooling_1_arm_avgpool_s8()
[all …]
|
| /cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_convolve_s16/ |
| D | test_arm_convolve_s16.c | 42 cmsis_nn_dims output_dims; in int16xint8_arm_convolve_s16() local
56 output_dims.w = INT16XINT8_OUTPUT_W; in int16xint8_arm_convolve_s16()
57 output_dims.h = INT16XINT8_OUTPUT_H; in int16xint8_arm_convolve_s16()
58 output_dims.c = INT16XINT8_OUT_CH; in int16xint8_arm_convolve_s16()
86 &output_dims, in int16xint8_arm_convolve_s16()
98 … = arm_convolve_wrapper_s16_get_buffer_size(&conv_params, &input_dims, &filter_dims, &output_dims); in int16xint8_arm_convolve_s16()
110 &output_dims, in int16xint8_arm_convolve_s16()
132 cmsis_nn_dims output_dims; in requantize_s64_arm_convolve_s16() local
146 output_dims.w = REQUANTIZE_S64_OUTPUT_W; in requantize_s64_arm_convolve_s16()
147 output_dims.h = REQUANTIZE_S64_OUTPUT_H; in requantize_s64_arm_convolve_s16()
[all …]
|
| /cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_avgpool_s16/ |
| D | test_arm_avgpool_s16.c | 37 cmsis_nn_dims output_dims; in avgpooling_int16_arm_avgpool_s16() local
47 output_dims.w = AVGPOOLING_INT16_OUTPUT_W; in avgpooling_int16_arm_avgpool_s16()
48 output_dims.h = AVGPOOLING_INT16_OUTPUT_H; in avgpooling_int16_arm_avgpool_s16()
49 output_dims.c = AVGPOOLING_INT16_OUT_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()
84 cmsis_nn_dims output_dims; in avgpooling_int16_1_arm_avgpool_s16() local
94 output_dims.w = AVGPOOLING_INT16_1_OUTPUT_W; in avgpooling_int16_1_arm_avgpool_s16()
95 output_dims.h = AVGPOOLING_INT16_1_OUTPUT_H; in avgpooling_int16_1_arm_avgpool_s16()
96 output_dims.c = AVGPOOLING_INT16_1_OUT_CH; in avgpooling_int16_1_arm_avgpool_s16()
110 … arm_avgpool_s16(&ctx, &pool_params, &input_dims, input_data, &filter_dims, &output_dims, output); 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 | 40 cmsis_nn_dims output_dims; in depthwise_kernel_3x3_arm_depthwise_conv_3x3_s8() local
53 output_dims.w = DEPTHWISE_KERNEL_3X3_OUTPUT_W; in depthwise_kernel_3x3_arm_depthwise_conv_3x3_s8()
54 output_dims.h = DEPTHWISE_KERNEL_3X3_OUTPUT_H; in depthwise_kernel_3x3_arm_depthwise_conv_3x3_s8()
55 output_dims.c = DEPTHWISE_KERNEL_3X3_OUT_CH; in depthwise_kernel_3x3_arm_depthwise_conv_3x3_s8()
85 &output_dims, in depthwise_kernel_3x3_arm_depthwise_conv_3x3_s8()
99 …epthwise_conv_wrapper_s8_get_buffer_size(&dw_conv_params, &input_dims, &filter_dims, &output_dims); in depthwise_kernel_3x3_arm_depthwise_conv_3x3_s8()
119 &output_dims, in depthwise_kernel_3x3_arm_depthwise_conv_3x3_s8()
143 cmsis_nn_dims output_dims; in depthwise_kernel_3x3_arm_depthwise_conv_3x3_1_s8() local
155 output_dims.w = DEPTHWISE_KERNEL_3X3_OUTPUT_W; in depthwise_kernel_3x3_arm_depthwise_conv_3x3_1_s8()
156 output_dims.h = DEPTHWISE_KERNEL_3X3_OUTPUT_H; 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 | 40 cmsis_nn_dims output_dims; in depthwise_2_arm_depthwise_conv_s8() local
53 output_dims.w = DEPTHWISE_2_OUTPUT_W; in depthwise_2_arm_depthwise_conv_s8()
54 output_dims.h = DEPTHWISE_2_OUTPUT_H; in depthwise_2_arm_depthwise_conv_s8()
55 output_dims.c = DEPTHWISE_2_OUT_CH; in depthwise_2_arm_depthwise_conv_s8()
85 &output_dims, in depthwise_2_arm_depthwise_conv_s8()
98 …epthwise_conv_wrapper_s8_get_buffer_size(&dw_conv_params, &input_dims, &filter_dims, &output_dims); in depthwise_2_arm_depthwise_conv_s8()
113 &output_dims, in depthwise_2_arm_depthwise_conv_s8()
136 cmsis_nn_dims output_dims; in depthwise_mult_batches_arm_depthwise_conv_s8() local
149 output_dims.w = DEPTHWISE_MULT_BATCHES_OUTPUT_W; in depthwise_mult_batches_arm_depthwise_conv_s8()
150 output_dims.h = DEPTHWISE_MULT_BATCHES_OUTPUT_H; in depthwise_mult_batches_arm_depthwise_conv_s8()
[all …]
|
| /cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_convolve_s8/ |
| D | test_arm_convolve_s8.c | 50 cmsis_nn_dims output_dims; in basic_arm_convolve_s8() local
64 output_dims.w = BASIC_OUTPUT_W; in basic_arm_convolve_s8()
65 output_dims.h = BASIC_OUTPUT_H; in basic_arm_convolve_s8()
66 output_dims.c = BASIC_OUT_CH; in basic_arm_convolve_s8()
95 &output_dims, in basic_arm_convolve_s8()
108 …e = arm_convolve_wrapper_s8_get_buffer_size(&conv_params, &input_dims, &filter_dims, &output_dims); in basic_arm_convolve_s8()
121 &output_dims, in basic_arm_convolve_s8()
144 cmsis_nn_dims output_dims; in stride2pad1_arm_convolve_s8() local
158 output_dims.w = STRIDE2PAD1_OUTPUT_W; in stride2pad1_arm_convolve_s8()
159 output_dims.h = STRIDE2PAD1_OUTPUT_H; in stride2pad1_arm_convolve_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 | 45 cmsis_nn_dims output_dims; in dw_int16xint8_fast_arm_depthwise_conv_fast_s16() local
59 output_dims.w = DW_INT16XINT8_FAST_OUTPUT_W; in dw_int16xint8_fast_arm_depthwise_conv_fast_s16()
60 output_dims.h = DW_INT16XINT8_FAST_OUTPUT_H; in dw_int16xint8_fast_arm_depthwise_conv_fast_s16()
61 output_dims.c = DW_INT16XINT8_FAST_OUT_CH; in dw_int16xint8_fast_arm_depthwise_conv_fast_s16()
91 &output_dims, in dw_int16xint8_fast_arm_depthwise_conv_fast_s16()
106 …thwise_conv_wrapper_s16_get_buffer_size(&dw_conv_params, &input_dims, &filter_dims, &output_dims)); in dw_int16xint8_fast_arm_depthwise_conv_fast_s16()
119 &output_dims, in dw_int16xint8_fast_arm_depthwise_conv_fast_s16()
142 cmsis_nn_dims output_dims; in dw_int16xint8_fast_spill_arm_depthwise_conv_fast_s16() local
156 output_dims.w = DW_INT16XINT8_FAST_SPILL_OUTPUT_W; in dw_int16xint8_fast_spill_arm_depthwise_conv_fast_s16()
157 output_dims.h = DW_INT16XINT8_FAST_SPILL_OUTPUT_H; in dw_int16xint8_fast_spill_arm_depthwise_conv_fast_s16()
[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 | 43 cmsis_nn_dims output_dims; in basic_arm_depthwise_conv_s8_opt() local
55 output_dims.w = BASIC_OUTPUT_W; in basic_arm_depthwise_conv_s8_opt()
56 output_dims.h = BASIC_OUTPUT_H; in basic_arm_depthwise_conv_s8_opt()
57 output_dims.c = BASIC_OUT_CH; in basic_arm_depthwise_conv_s8_opt()
94 &output_dims, in basic_arm_depthwise_conv_s8_opt()
107 …epthwise_conv_wrapper_s8_get_buffer_size(&dw_conv_params, &input_dims, &filter_dims, &output_dims); in basic_arm_depthwise_conv_s8_opt()
122 &output_dims, in basic_arm_depthwise_conv_s8_opt()
145 cmsis_nn_dims output_dims; in depthwise_eq_in_out_ch_arm_depthwise_conv_s8_opt() local
157 output_dims.w = DEPTHWISE_EQ_IN_OUT_CH_OUTPUT_W; in depthwise_eq_in_out_ch_arm_depthwise_conv_s8_opt()
158 output_dims.h = DEPTHWISE_EQ_IN_OUT_CH_OUTPUT_H; in depthwise_eq_in_out_ch_arm_depthwise_conv_s8_opt()
[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 | 42 cmsis_nn_dims output_dims; in conv_1_x_n_1_arm_convolve_s8() local
56 output_dims.w = CONV_1_X_N_1_OUTPUT_W; in conv_1_x_n_1_arm_convolve_s8()
57 output_dims.h = CONV_1_X_N_1_OUTPUT_H; in conv_1_x_n_1_arm_convolve_s8()
58 output_dims.c = CONV_1_X_N_1_OUT_CH; in conv_1_x_n_1_arm_convolve_s8()
87 &output_dims, in conv_1_x_n_1_arm_convolve_s8()
111 &output_dims, in conv_1_x_n_1_arm_convolve_s8()
133 cmsis_nn_dims output_dims; in conv_1_x_n_2_arm_convolve_s8() local
147 output_dims.w = CONV_1_X_N_2_OUTPUT_W; in conv_1_x_n_2_arm_convolve_s8()
148 output_dims.h = CONV_1_X_N_2_OUTPUT_H; in conv_1_x_n_2_arm_convolve_s8()
149 output_dims.c = CONV_1_X_N_2_OUT_CH; in conv_1_x_n_2_arm_convolve_s8()
[all …]
|
| /cmsis-nn-3.6.0-3.5.0/Include/ |
| D | arm_nnfunctions.h | 176 const cmsis_nn_dims *output_dims,
196 const cmsis_nn_dims *output_dims);
209 const cmsis_nn_dims *output_dims);
222 const cmsis_nn_dims *output_dims);
260 const cmsis_nn_dims *output_dims,
280 const cmsis_nn_dims *output_dims);
293 const cmsis_nn_dims *output_dims);
306 const cmsis_nn_dims *output_dims);
344 const cmsis_nn_dims *output_dims,
394 const cmsis_nn_dims *output_dims,
[all …]
|
| /cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_fully_connected_s8/ |
| D | test_arm_fully_connected_s8.c | 41 cmsis_nn_dims output_dims; in fully_connected_arm_fully_connected_s8() local
55 output_dims.n = FULLY_CONNECTED_INPUT_BATCHES; in fully_connected_arm_fully_connected_s8()
56 output_dims.c = FULLY_CONNECTED_OUT_CH; in fully_connected_arm_fully_connected_s8()
80 &output_dims, in fully_connected_arm_fully_connected_s8()
103 cmsis_nn_dims output_dims; in fully_connected_mve_0_arm_fully_connected_s8() local
115 output_dims.n = FULLY_CONNECTED_MVE_0_INPUT_BATCHES; in fully_connected_mve_0_arm_fully_connected_s8()
116 output_dims.c = FULLY_CONNECTED_MVE_0_OUT_CH; in fully_connected_mve_0_arm_fully_connected_s8()
137 &output_dims, in fully_connected_mve_0_arm_fully_connected_s8()
159 cmsis_nn_dims output_dims; in fully_connected_mve_1_arm_fully_connected_s8() local
171 output_dims.n = FULLY_CONNECTED_MVE_1_INPUT_BATCHES; in fully_connected_mve_1_arm_fully_connected_s8()
[all …]
|
