Home
last modified time | relevance | path

Searched refs:output_dims (Results 1 – 25 of 66) sorted by relevance

123

/cmsis-nn-latest/Source/ConvolutionFunctions/
Darm_convolve_get_buffer_sizes_s16.c73 const cmsis_nn_dims *output_dims) in arm_convolve_wrapper_s16_get_buffer_size() argument
76 (void)output_dims; in arm_convolve_wrapper_s16_get_buffer_size()
84 const cmsis_nn_dims *output_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()
92 const cmsis_nn_dims *output_dims) in arm_convolve_wrapper_s16_get_buffer_size_mve() argument
95 (void)output_dims; in arm_convolve_wrapper_s16_get_buffer_size_mve()
Darm_transpose_conv_get_buffer_sizes_s8.c61 const cmsis_nn_dims *output_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()
68 const cmsis_nn_dims *output_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_depthwise_conv_get_buffer_sizes_s4.c55 const cmsis_nn_dims *output_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()
71 const cmsis_nn_dims *output_dims) in arm_depthwise_conv_wrapper_s4_get_buffer_size_dsp() argument
73 …rm_depthwise_conv_wrapper_s4_get_buffer_size(dw_conv_params, input_dims, filter_dims, output_dims); in arm_depthwise_conv_wrapper_s4_get_buffer_size_dsp()
79 const cmsis_nn_dims *output_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()
Darm_convolve_get_buffer_sizes_s4.c58 const cmsis_nn_dims *output_dims) in arm_convolve_1_x_n_s4_get_buffer_size_mve() argument
63 const int32_t output_x = output_dims->w; in arm_convolve_1_x_n_s4_get_buffer_size_mve()
89 const cmsis_nn_dims *output_dims) in arm_convolve_1_x_n_s4_get_buffer_size() argument
93 (void)output_dims; in arm_convolve_1_x_n_s4_get_buffer_size()
97 …eturn 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()
117 const cmsis_nn_dims *output_dims) in arm_convolve_wrapper_s4_get_buffer_size() argument
120 …urn arm_convolve_wrapper_s8_get_buffer_size_mve(conv_params, input_dims, filter_dims, output_dims); in arm_convolve_wrapper_s4_get_buffer_size()
122 (void)output_dims; in arm_convolve_wrapper_s4_get_buffer_size()
145 const cmsis_nn_dims *output_dims) in arm_convolve_wrapper_s4_get_buffer_size_mve() argument
148 (void)output_dims; in arm_convolve_wrapper_s4_get_buffer_size_mve()
[all …]
Darm_convolve_get_buffer_sizes_s8.c58 const cmsis_nn_dims *output_dims) in arm_convolve_1_x_n_s8_get_buffer_size_mve() argument
63 const int32_t output_x = output_dims->w; in arm_convolve_1_x_n_s8_get_buffer_size_mve()
101 const cmsis_nn_dims *output_dims) in arm_convolve_1_x_n_s8_get_buffer_size() argument
105 (void)output_dims; in arm_convolve_1_x_n_s8_get_buffer_size()
109 …eturn 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()
129 const cmsis_nn_dims *output_dims) in arm_convolve_wrapper_s8_get_buffer_size() argument
132 …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()
134 (void)output_dims; in arm_convolve_wrapper_s8_get_buffer_size()
150 … return arm_convolve_1_x_n_s8_get_buffer_size(conv_params, input_dims, filter_dims, output_dims); in arm_convolve_wrapper_s8_get_buffer_size()
162 const cmsis_nn_dims *output_dims) in arm_convolve_wrapper_s8_get_buffer_size_mve() argument
[all …]
Darm_depthwise_conv_get_buffer_sizes_s16.c73 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()
Darm_convolve_wrapper_s4.c59 const cmsis_nn_dims *output_dims, in arm_convolve_wrapper_s4() argument
76 output_dims, in arm_convolve_wrapper_s4()
90 output_dims, in arm_convolve_wrapper_s4()
106 output_dims, in arm_convolve_wrapper_s4()
120 output_dims, in arm_convolve_wrapper_s4()
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_max_pool_s8/
Dtest_arm_max_pool_s8.c43 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-latest/Tests/UnitTest/TestCases/test_arm_convolve_1x1_s8_fast/
Dtest_arm_convolve_1x1_s8_fast.c41 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-latest/Tests/UnitTest/TestCases/test_arm_max_pool_s16/
Dtest_arm_max_pool_s16.c38 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-latest/Tests/UnitTest/TestCases/test_arm_transpose_conv_s8/
Dtest_arm_transpose_conv_s8.c41 cmsis_nn_dims output_dims; in transpose_conv_1_arm_transpose_conv_s8() local
55 output_dims.n = TRANSPOSE_CONV_1_INPUT_BATCHES; in transpose_conv_1_arm_transpose_conv_s8()
56 output_dims.w = TRANSPOSE_CONV_1_OUTPUT_W; in transpose_conv_1_arm_transpose_conv_s8()
57 output_dims.h = TRANSPOSE_CONV_1_OUTPUT_H; in transpose_conv_1_arm_transpose_conv_s8()
58 output_dims.c = TRANSPOSE_CONV_1_OUT_CH; in transpose_conv_1_arm_transpose_conv_s8()
60 output_ctx.size = output_dims.w * output_dims.h * output_dims.c * sizeof(int32_t); in transpose_conv_1_arm_transpose_conv_s8()
80 … int32_t buf_size = arm_transpose_conv_s8_get_buffer_size(&input_dims, &filter_dims, &output_dims); in transpose_conv_1_arm_transpose_conv_s8()
94 &output_dims, in transpose_conv_1_arm_transpose_conv_s8()
127 cmsis_nn_dims output_dims; in transpose_conv_2_arm_transpose_conv_s8() local
141 output_dims.n = TRANSPOSE_CONV_2_INPUT_BATCHES; in transpose_conv_2_arm_transpose_conv_s8()
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_convolve_1x1_s4_fast/
Dtest_arm_convolve_1x1_s4_fast.c43 cmsis_nn_dims output_dims; in kernel1x1_arm_convolve_1x1_s4_fast() local
56 output_dims.n = KERNEL1X1_INT4_INPUT_BATCHES; in kernel1x1_arm_convolve_1x1_s4_fast()
57 output_dims.h = KERNEL1X1_INT4_OUTPUT_H; in kernel1x1_arm_convolve_1x1_s4_fast()
58 output_dims.w = KERNEL1X1_INT4_OUTPUT_W; in kernel1x1_arm_convolve_1x1_s4_fast()
59 output_dims.c = KERNEL1X1_INT4_OUT_CH; in kernel1x1_arm_convolve_1x1_s4_fast()
71 bias_dims.c = output_dims.c; in kernel1x1_arm_convolve_1x1_s4_fast()
93 &output_dims, in kernel1x1_arm_convolve_1x1_s4_fast()
117 cmsis_nn_dims output_dims; in kernel1x1_2_arm_convolve_1x1_s4_fast() local
130 output_dims.n = KERNEL1X1_INT4_2_INPUT_BATCHES; in kernel1x1_2_arm_convolve_1x1_s4_fast()
131 output_dims.h = KERNEL1X1_INT4_2_OUTPUT_H; in kernel1x1_2_arm_convolve_1x1_s4_fast()
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_depthwise_conv_s16/
Dtest_arm_depthwise_conv_s16.c38 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-latest/Tests/UnitTest/TestCases/test_arm_avgpool_s8/
Dtest_arm_avgpool_s8.c39 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-latest/Tests/UnitTest/TestCases/test_arm_avgpool_s16/
Dtest_arm_avgpool_s16.c37 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-latest/Tests/UnitTest/TestCases/test_arm_depthwise_conv_s8/
Dtest_arm_depthwise_conv_s8.c40 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-latest/Tests/UnitTest/TestCases/test_arm_depthwise_conv_3x3_s8/
Dtest_arm_depthwise_conv_3x3_s8.c40 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-latest/Tests/UnitTest/TestCases/test_arm_convolve_s16/
Dtest_arm_convolve_s16.c50 cmsis_nn_dims output_dims; in int16xint8_arm_convolve_s16() local
65 output_dims.w = INT16XINT8_OUTPUT_W; in int16xint8_arm_convolve_s16()
66 output_dims.h = INT16XINT8_OUTPUT_H; in int16xint8_arm_convolve_s16()
67 output_dims.c = INT16XINT8_OUT_CH; in int16xint8_arm_convolve_s16()
95 &output_dims, in int16xint8_arm_convolve_s16()
107 … = arm_convolve_wrapper_s16_get_buffer_size(&conv_params, &input_dims, &filter_dims, &output_dims); in int16xint8_arm_convolve_s16()
119 &output_dims, in int16xint8_arm_convolve_s16()
140 cmsis_nn_dims output_dims; in requantize_s64_arm_convolve_s16() local
155 output_dims.w = REQUANTIZE_S64_OUTPUT_W; in requantize_s64_arm_convolve_s16()
156 output_dims.h = REQUANTIZE_S64_OUTPUT_H; in requantize_s64_arm_convolve_s16()
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_convolve_s8/
Dtest_arm_convolve_s8.c50 cmsis_nn_dims output_dims; in basic_arm_convolve_s8() local
65 output_dims.w = BASIC_OUTPUT_W; in basic_arm_convolve_s8()
66 output_dims.h = BASIC_OUTPUT_H; in basic_arm_convolve_s8()
67 output_dims.c = BASIC_OUT_CH; in basic_arm_convolve_s8()
96 &output_dims, in basic_arm_convolve_s8()
109 …e = arm_convolve_wrapper_s8_get_buffer_size(&conv_params, &input_dims, &filter_dims, &output_dims); in basic_arm_convolve_s8()
122 &output_dims, in basic_arm_convolve_s8()
145 cmsis_nn_dims output_dims; in stride2pad1_arm_convolve_s8() local
160 output_dims.w = STRIDE2PAD1_OUTPUT_W; in stride2pad1_arm_convolve_s8()
161 output_dims.h = STRIDE2PAD1_OUTPUT_H; in stride2pad1_arm_convolve_s8()
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_depthwise_conv_fast_s16/
Dtest_arm_depthwise_conv_fast_s16.c45 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-latest/Tests/UnitTest/TestCases/test_arm_depthwise_conv_s8_opt/
Dtest_arm_depthwise_conv_s8_opt.c43 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-latest/Tests/UnitTest/TestCases/test_arm_convolve_1_x_n_s8/
Dtest_arm_convolve_1_x_n_s8.c46 cmsis_nn_dims output_dims; in conv_1_x_n_1_arm_convolve_s8() local
61 output_dims.w = CONV_1_X_N_1_OUTPUT_W; in conv_1_x_n_1_arm_convolve_s8()
62 output_dims.h = CONV_1_X_N_1_OUTPUT_H; in conv_1_x_n_1_arm_convolve_s8()
63 output_dims.c = CONV_1_X_N_1_OUT_CH; in conv_1_x_n_1_arm_convolve_s8()
79 …ize = arm_convolve_1_x_n_s8_get_buffer_size(&conv_params, &input_dims, &filter_dims, &output_dims); in conv_1_x_n_1_arm_convolve_s8()
92 &output_dims, in conv_1_x_n_1_arm_convolve_s8()
116 &output_dims, in conv_1_x_n_1_arm_convolve_s8()
138 cmsis_nn_dims output_dims; in conv_1_x_n_2_arm_convolve_s8() local
153 output_dims.w = CONV_1_X_N_2_OUTPUT_W; in conv_1_x_n_2_arm_convolve_s8()
154 output_dims.h = CONV_1_X_N_2_OUTPUT_H; in conv_1_x_n_2_arm_convolve_s8()
[all …]
/cmsis-nn-latest/Include/
Darm_nnfunctions.h101 const cmsis_nn_dims *output_dims,
121 const cmsis_nn_dims *output_dims);
135 const cmsis_nn_dims *output_dims);
148 const cmsis_nn_dims *output_dims);
186 const cmsis_nn_dims *output_dims,
206 const cmsis_nn_dims *output_dims);
219 const cmsis_nn_dims *output_dims);
232 const cmsis_nn_dims *output_dims);
271 const cmsis_nn_dims *output_dims,
291 const cmsis_nn_dims *output_dims);
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_depthwise_conv_s4/
Dtest_arm_depthwise_conv_s4.c43 cmsis_nn_dims output_dims; in depthwise_int4_generic_arm_depthwise_conv_s4() local
55 output_dims.w = DEPTHWISE_INT4_GENERIC_OUTPUT_W; in depthwise_int4_generic_arm_depthwise_conv_s4()
56 output_dims.h = DEPTHWISE_INT4_GENERIC_OUTPUT_H; in depthwise_int4_generic_arm_depthwise_conv_s4()
57 output_dims.c = DEPTHWISE_INT4_GENERIC_OUT_CH; in depthwise_int4_generic_arm_depthwise_conv_s4()
75 …epthwise_conv_wrapper_s4_get_buffer_size(&dw_conv_params, &input_dims, &filter_dims, &output_dims); in depthwise_int4_generic_arm_depthwise_conv_s4()
89 &output_dims, in depthwise_int4_generic_arm_depthwise_conv_s4()
112 &output_dims, in depthwise_int4_generic_arm_depthwise_conv_s4()
136 cmsis_nn_dims output_dims; in depthwise_int4_generic_2_arm_depthwise_conv_s4() local
148 output_dims.w = DEPTHWISE_INT4_GENERIC_2_OUTPUT_W; in depthwise_int4_generic_2_arm_depthwise_conv_s4()
149 output_dims.h = DEPTHWISE_INT4_GENERIC_2_OUTPUT_H; in depthwise_int4_generic_2_arm_depthwise_conv_s4()
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_convolve_s4/
Dtest_arm_convolve_s4.c56 cmsis_nn_dims output_dims; in basic_arm_convolve_s4() local
70 output_dims.w = BASIC_INT4_OUTPUT_W; in basic_arm_convolve_s4()
71 output_dims.h = BASIC_INT4_OUTPUT_H; in basic_arm_convolve_s4()
72 output_dims.c = BASIC_INT4_OUT_CH; in basic_arm_convolve_s4()
101 &output_dims, in basic_arm_convolve_s4()
114 …e = arm_convolve_wrapper_s4_get_buffer_size(&conv_params, &input_dims, &filter_dims, &output_dims); in basic_arm_convolve_s4()
127 &output_dims, in basic_arm_convolve_s4()
150 cmsis_nn_dims output_dims; in basic_2_arm_convolve_s4() local
164 output_dims.w = BASIC_2_INT4_OUTPUT_W; in basic_2_arm_convolve_s4()
165 output_dims.h = BASIC_2_INT4_OUTPUT_H; in basic_2_arm_convolve_s4()
[all …]

123