Home
last modified time | relevance | path

Searched refs:h (Results 1 – 25 of 78) sorted by relevance

1234

/cmsis-nn-latest/Source/ConvolutionFunctions/
Darm_transpose_conv_get_buffer_sizes_s8.c58 ((transpose_conv_params->stride.w <= 2) && (transpose_conv_params->stride.h <= 2)); in arm_transpose_conv_s8_get_buffer_size()
64 … input_dims->h * transpose_conv_params->stride.h, in arm_transpose_conv_s8_get_buffer_size()
72 MAX(filter_dims->w, transpose_conv_params->stride.h)) * in arm_transpose_conv_s8_get_buffer_size()
74 const int32_t buf_y = MAX(filter_dims->h, transpose_conv_params->stride.h); in arm_transpose_conv_s8_get_buffer_size()
84 ((transpose_conv_params->stride.w <= 2) && (transpose_conv_params->stride.h <= 2)); in arm_transpose_conv_s8_get_reverse_conv_buffer_size()
89 return input_dims->c * filter_dims->w * filter_dims->h * filter_dims->n; in arm_transpose_conv_s8_get_reverse_conv_buffer_size()
Darm_convolve_get_buffer_sizes_s8.c56 int32_t col_length = input_dims->c * filter_dims->w * filter_dims->h; in arm_convolve_s8_get_buffer_size_mve()
100 const int32_t rhs_cols = filter_dims->w * filter_dims->h * input_dims->c; in arm_convolve_s8_get_buffer_size()
150 if ((conv_params->padding.w == 0) && (conv_params->padding.h == 0) && (filter_dims->w == 1) && in arm_convolve_wrapper_s8_get_buffer_size()
151 (filter_dims->h == 1) && (conv_params->dilation.w == 1 && conv_params->dilation.h == 1)) in arm_convolve_wrapper_s8_get_buffer_size()
153 if ((conv_params->stride.w == 1) && (conv_params->stride.h == 1)) in arm_convolve_wrapper_s8_get_buffer_size()
162 else if ((input_dims->h == 1) && (conv_params->dilation.w == 1) && (filter_dims->h == 1) && in arm_convolve_wrapper_s8_get_buffer_size()
180 if ((conv_params->padding.w == 0) && (conv_params->padding.h == 0) && (filter_dims->w == 1) && in arm_convolve_wrapper_s8_get_buffer_size_mve()
181 (filter_dims->h == 1) && (conv_params->dilation.w == 1 && conv_params->dilation.h == 1)) in arm_convolve_wrapper_s8_get_buffer_size_mve()
183 if ((conv_params->stride.w == 1) && (conv_params->stride.h == 1)) in arm_convolve_wrapper_s8_get_buffer_size_mve()
192 else if ((input_dims->h == 1) && (conv_params->dilation.w == 1) && (filter_dims->h == 1) && in arm_convolve_wrapper_s8_get_buffer_size_mve()
[all …]
Darm_convolve_wrapper_s4.c62 if ((conv_params->padding.w == 0) && (conv_params->padding.h == 0) && (filter_dims->w == 1) && in arm_convolve_wrapper_s4()
63 (filter_dims->h == 1) && (conv_params->dilation.w == 1 && conv_params->dilation.h == 1)) in arm_convolve_wrapper_s4()
65 if ((conv_params->stride.w == 1) && (conv_params->stride.h == 1)) 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()
110 else if (((filter_dims->h * filter_dims->w * input_dims->c) & 0x1) == 0) in arm_convolve_wrapper_s4()
Darm_convolve_get_buffer_sizes_s4.c47 int32_t col_length = input_dims->c * filter_dims->w * filter_dims->h; in arm_convolve_s4_get_buffer_size_mve()
82 const int32_t rhs_cols = filter_dims->w * filter_dims->h * input_dims->c; in arm_convolve_s4_get_buffer_size()
123 if ((conv_params->padding.w == 0) && (conv_params->padding.h == 0) && (filter_dims->w == 1) && in arm_convolve_wrapper_s4_get_buffer_size()
124 (filter_dims->h == 1) && (conv_params->dilation.w == 1 && conv_params->dilation.h == 1)) in arm_convolve_wrapper_s4_get_buffer_size()
126 if ((conv_params->stride.w == 1) && (conv_params->stride.h == 1)) in arm_convolve_wrapper_s4_get_buffer_size()
149 if ((conv_params->padding.w == 0) && (conv_params->padding.h == 0) && (filter_dims->w == 1) && in arm_convolve_wrapper_s4_get_buffer_size_mve()
150 (filter_dims->h == 1) && (conv_params->dilation.w == 1 && conv_params->dilation.h == 1)) in arm_convolve_wrapper_s4_get_buffer_size_mve()
152 if ((conv_params->stride.w == 1) && (conv_params->stride.h == 1)) in arm_convolve_wrapper_s4_get_buffer_size_mve()
161 else if ((input_dims->h == 1) && (conv_params->dilation.w == 1) && (filter_dims->h == 1) && in arm_convolve_wrapper_s4_get_buffer_size_mve()
Darm_convolve_wrapper_s8.c62 if ((conv_params->padding.w == 0) && (conv_params->padding.h == 0) && (filter_dims->w == 1) && in arm_convolve_wrapper_s8()
63 (filter_dims->h == 1) && (conv_params->dilation.w == 1 && conv_params->dilation.h == 1) && in arm_convolve_wrapper_s8()
66 if ((conv_params->stride.w == 1) && (conv_params->stride.h == 1)) 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()
Darm_depthwise_conv_get_buffer_sizes_s8.c49 …const cmsis_nn_dims filter_conv_dims = {filter_dims->c, filter_dims->h, filter_dims->w, filter_dim… in arm_deptwise_conv_s8_one_in_ch_get_buffer_size_mve()
59 size += filter_dims->c * filter_dims->h * filter_dims->w * filter_dims->n; in arm_deptwise_conv_s8_one_in_ch_get_buffer_size_mve()
67 return (4 * CH_IN_BLOCK_MVE * filter_dims->w * filter_dims->h) * (int32_t)sizeof(int8_t); in arm_depthwise_conv_s8_opt_get_buffer_size_mve()
72 return (input_dims->c * filter_dims->w * filter_dims->h) * sizeof(int16_t); in arm_depthwise_conv_s8_opt_get_buffer_size_dsp()
103 dw_conv_params->dilation.h == 1) in arm_depthwise_conv_wrapper_s8_get_buffer_size()
106 if (filter_dims->w == 3 && filter_dims->h == 3 && dw_conv_params->padding.h <= 1 && in arm_depthwise_conv_wrapper_s8_get_buffer_size()
126 dw_conv_params->dilation.h == 1) in arm_depthwise_conv_wrapper_s8_get_buffer_size_dsp()
128 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()
147 dw_conv_params->dilation.h == 1) in arm_depthwise_conv_wrapper_s8_get_buffer_size_mve()
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_max_pool_s8/
Dtest_arm_max_pool_s8.c49 input_dims.h = MAXPOOLING_INPUT_H; in maxpooling_arm_max_pool_s8()
52 filter_dims.h = MAXPOOLING_FILTER_H; in maxpooling_arm_max_pool_s8()
54 output_dims.h = MAXPOOLING_OUTPUT_H; in maxpooling_arm_max_pool_s8()
58 pool_params.padding.h = MAXPOOLING_PADDING_H; in maxpooling_arm_max_pool_s8()
60 pool_params.stride.h = MAXPOOLING_STRIDE_H; in maxpooling_arm_max_pool_s8()
93 input_dims.h = MAXPOOLING_1_INPUT_H; in maxpooling_1_arm_max_pool_s8()
96 filter_dims.h = MAXPOOLING_1_FILTER_H; in maxpooling_1_arm_max_pool_s8()
98 output_dims.h = MAXPOOLING_1_OUTPUT_H; in maxpooling_1_arm_max_pool_s8()
102 pool_params.padding.h = MAXPOOLING_1_PADDING_H; in maxpooling_1_arm_max_pool_s8()
104 pool_params.stride.h = MAXPOOLING_1_STRIDE_H; in maxpooling_1_arm_max_pool_s8()
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_convolve_1x1_s4_fast/
Dtest_arm_convolve_1x1_s4_fast.c49 input_dims.h = KERNEL1X1_INT4_INPUT_H; in kernel1x1_arm_convolve_1x1_s4_fast()
53 filter_dims.h = KERNEL1X1_INT4_FILTER_Y; in kernel1x1_arm_convolve_1x1_s4_fast()
57 output_dims.h = KERNEL1X1_INT4_OUTPUT_H; in kernel1x1_arm_convolve_1x1_s4_fast()
61 conv_params.padding.h = KERNEL1X1_INT4_PAD_Y; in kernel1x1_arm_convolve_1x1_s4_fast()
63 conv_params.stride.h = KERNEL1X1_INT4_STRIDE_Y; in kernel1x1_arm_convolve_1x1_s4_fast()
66 conv_params.dilation.h = KERNEL1X1_INT4_DILATION_Y; in kernel1x1_arm_convolve_1x1_s4_fast()
69 bias_dims.h = 1; in kernel1x1_arm_convolve_1x1_s4_fast()
123 input_dims.h = KERNEL1X1_INT4_2_INPUT_H; in kernel1x1_2_arm_convolve_1x1_s4_fast()
127 filter_dims.h = KERNEL1X1_INT4_2_FILTER_Y; 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_max_pool_s16/
Dtest_arm_max_pool_s16.c45 input_dims.h = MAXPOOL_INT16_INPUT_H; in maxpool_int16_arm_max_pool_s16()
48 filter_dims.h = MAXPOOL_INT16_FILTER_H; in maxpool_int16_arm_max_pool_s16()
50 output_dims.h = MAXPOOL_INT16_OUTPUT_H; in maxpool_int16_arm_max_pool_s16()
54 pool_params.padding.h = MAXPOOL_INT16_PADDING_H; in maxpool_int16_arm_max_pool_s16()
56 pool_params.stride.h = MAXPOOL_INT16_STRIDE_H; in maxpool_int16_arm_max_pool_s16()
90 input_dims.h = MAXPOOL_INT16_1_INPUT_H; in maxpool_int16_1_arm_max_pool_s16()
93 filter_dims.h = MAXPOOL_INT16_1_FILTER_H; in maxpool_int16_1_arm_max_pool_s16()
95 output_dims.h = MAXPOOL_INT16_1_OUTPUT_H; in maxpool_int16_1_arm_max_pool_s16()
99 pool_params.padding.h = MAXPOOL_INT16_1_PADDING_H; in maxpool_int16_1_arm_max_pool_s16()
101 pool_params.stride.h = MAXPOOL_INT16_1_STRIDE_H; in maxpool_int16_1_arm_max_pool_s16()
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_ds_cnn_s_s8/
Dtest_arm_ds_cnn_s_s8.c48 conv_params.padding.h = CONV_2D_1_PAD_H; in ds_cnn_s_s8_get_buffer_size()
50 conv_params.stride.h = CONV_2D_1_STRIDE_H; in ds_cnn_s_s8_get_buffer_size()
52 conv_params.dilation.h = CONV_2D_1_DILATION_H; in ds_cnn_s_s8_get_buffer_size()
56 input_dims.h = CONV_2D_1_INPUT_H; in ds_cnn_s_s8_get_buffer_size()
60 filter_dims.h = CONV_2D_1_FILTER_H; in ds_cnn_s_s8_get_buffer_size()
65 output_dims.h = CONV_2D_1_OUTPUT_H; in ds_cnn_s_s8_get_buffer_size()
78 dw_conv_params.dilation.h = DEPTHWISE_CONV_2D_2_DILATION_H; in ds_cnn_s_s8_get_buffer_size()
82 dw_conv_params.padding.h = DEPTHWISE_CONV_2D_2_PAD_H; in ds_cnn_s_s8_get_buffer_size()
84 dw_conv_params.stride.h = DEPTHWISE_CONV_2D_2_STRIDE_H; in ds_cnn_s_s8_get_buffer_size()
87 filter_dims.h = DEPTHWISE_CONV_2D_2_FILTER_H; in ds_cnn_s_s8_get_buffer_size()
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_ds_cnn_l_s8/
Dtest_arm_ds_cnn_l_s8.c47 conv_params.padding.h = CONV_2D_1_PAD_H; in ds_cnn_l_s8_get_buffer_size()
49 conv_params.stride.h = CONV_2D_1_STRIDE_H; in ds_cnn_l_s8_get_buffer_size()
51 conv_params.dilation.h = CONV_2D_1_DILATION_H; in ds_cnn_l_s8_get_buffer_size()
55 input_dims.h = CONV_2D_1_INPUT_H; in ds_cnn_l_s8_get_buffer_size()
59 filter_dims.h = CONV_2D_1_FILTER_H; in ds_cnn_l_s8_get_buffer_size()
64 output_dims.h = CONV_2D_1_OUTPUT_H; in ds_cnn_l_s8_get_buffer_size()
77 dw_conv_params.dilation.h = DEPTHWISE_CONV_2D_2_DILATION_H; in ds_cnn_l_s8_get_buffer_size()
81 dw_conv_params.padding.h = DEPTHWISE_CONV_2D_2_PAD_H; in ds_cnn_l_s8_get_buffer_size()
83 dw_conv_params.stride.h = DEPTHWISE_CONV_2D_2_STRIDE_H; in ds_cnn_l_s8_get_buffer_size()
86 filter_dims.h = DEPTHWISE_CONV_2D_2_FILTER_H; in ds_cnn_l_s8_get_buffer_size()
[all …]
/cmsis-nn-latest/Source/PadFunctions/
Darm_pad_s8.c60 pre_pad->h + input_size->h + post_pad->h, in arm_pad_s8()
64 const int32_t batch_block_size = output_size.h * output_size.w * output_size.c; in arm_pad_s8()
73 arm_memset_s8(output, pad_value, row_block_size * pre_pad->h); in arm_pad_s8()
74 output += row_block_size * pre_pad->h; in arm_pad_s8()
75 for (int32_t y = 0; y < input_size->h; y++) in arm_pad_s8()
107 arm_memset_s8(output, pad_value, row_block_size * post_pad->h); in arm_pad_s8()
108 output += row_block_size * post_pad->h; in arm_pad_s8()
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_convolve_1x1_s8_fast/
Dtest_arm_convolve_1x1_s8_fast.c47 input_dims.h = KERNEL1X1_INPUT_H; in kernel1x1_arm_convolve_1x1_s8_fast()
51 filter_dims.h = KERNEL1X1_FILTER_Y; in kernel1x1_arm_convolve_1x1_s8_fast()
55 output_dims.h = KERNEL1X1_OUTPUT_H; in kernel1x1_arm_convolve_1x1_s8_fast()
59 conv_params.padding.h = KERNEL1X1_PAD_Y; in kernel1x1_arm_convolve_1x1_s8_fast()
61 conv_params.stride.h = KERNEL1X1_STRIDE_Y; in kernel1x1_arm_convolve_1x1_s8_fast()
65 bias_dims.h = 1; in kernel1x1_arm_convolve_1x1_s8_fast()
119 input_dims.h = KERNEL1X1_STRIDE_X_INPUT_H; in kernel1x1_stride_x_arm_convolve_1x1_s8()
124 filter_dims.h = KERNEL1X1_STRIDE_X_FILTER_Y; 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()
134 bias_dims.h = 1; in kernel1x1_stride_x_arm_convolve_1x1_s8()
[all …]
/cmsis-nn-latest/Source/FullyConnectedFunctions/
Darm_batch_matmul_s16.c58 const int32_t output_height = output_dims->h; in arm_batch_matmul_s16()
63 const int32_t inner_lhs_diff = input_lhs_dims->h >= input_rhs_dims->h ? 0 : lhs_rows * rhs_cols; in arm_batch_matmul_s16()
64 const int32_t inner_rhs_diff = input_rhs_dims->h >= input_lhs_dims->h ? rhs_rows * rhs_cols : 0; in arm_batch_matmul_s16()
67 : -((lhs_rows * rhs_cols) - inner_lhs_diff) * input_lhs_dims->h; in arm_batch_matmul_s16()
69 … : -inner_rhs_diff * input_rhs_dims->h; in arm_batch_matmul_s16()
Darm_batch_matmul_s8.c65 const int32_t output_height = output_dims->h; in arm_batch_matmul_s8()
70 const int32_t inner_lhs_diff = input_lhs_dims->h >= input_rhs_dims->h ? 0 : lhs_rows * rhs_cols; in arm_batch_matmul_s8()
71 const int32_t inner_rhs_diff = input_rhs_dims->h >= input_lhs_dims->h ? rhs_rows * rhs_cols : 0; in arm_batch_matmul_s8()
74 : -((lhs_rows * rhs_cols) - inner_lhs_diff) * input_lhs_dims->h; in arm_batch_matmul_s8()
76 … : -inner_rhs_diff * input_rhs_dims->h; in arm_batch_matmul_s8()
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_avgpool_s16/
Dtest_arm_avgpool_s16.c44 input_dims.h = AVGPOOLING_INT16_INPUT_H; in avgpooling_int16_arm_avgpool_s16()
47 filter_dims.h = AVGPOOLING_INT16_FILTER_H; in avgpooling_int16_arm_avgpool_s16()
49 output_dims.h = AVGPOOLING_INT16_OUTPUT_H; in avgpooling_int16_arm_avgpool_s16()
53 pool_params.padding.h = AVGPOOLING_INT16_PADDING_H; in avgpooling_int16_arm_avgpool_s16()
55 pool_params.stride.h = AVGPOOLING_INT16_STRIDE_H; in avgpooling_int16_arm_avgpool_s16()
95 input_dims.h = AVGPOOLING_INT16_1_INPUT_H; in avgpooling_int16_1_arm_avgpool_s16()
98 filter_dims.h = AVGPOOLING_INT16_1_FILTER_H; in avgpooling_int16_1_arm_avgpool_s16()
100 output_dims.h = AVGPOOLING_INT16_1_OUTPUT_H; in avgpooling_int16_1_arm_avgpool_s16()
104 pool_params.padding.h = AVGPOOLING_INT16_1_PADDING_H; in avgpooling_int16_1_arm_avgpool_s16()
106 pool_params.stride.h = AVGPOOLING_INT16_1_STRIDE_H; in avgpooling_int16_1_arm_avgpool_s16()
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_avgpool_s8/
Dtest_arm_avgpool_s8.c45 input_dims.h = AVGPOOLING_INPUT_H; in avgpooling_arm_avgpool_s8()
48 filter_dims.h = AVGPOOLING_FILTER_H; in avgpooling_arm_avgpool_s8()
50 output_dims.h = AVGPOOLING_OUTPUT_H; in avgpooling_arm_avgpool_s8()
54 pool_params.padding.h = AVGPOOLING_PADDING_H; in avgpooling_arm_avgpool_s8()
56 pool_params.stride.h = AVGPOOLING_STRIDE_H; in avgpooling_arm_avgpool_s8()
95 input_dims.h = AVGPOOLING_1_INPUT_H; in avgpooling_1_arm_avgpool_s8()
98 filter_dims.h = AVGPOOLING_1_FILTER_H; in avgpooling_1_arm_avgpool_s8()
100 output_dims.h = AVGPOOLING_1_OUTPUT_H; in avgpooling_1_arm_avgpool_s8()
104 pool_params.padding.h = AVGPOOLING_1_PADDING_H; in avgpooling_1_arm_avgpool_s8()
106 pool_params.stride.h = AVGPOOLING_1_STRIDE_H; in avgpooling_1_arm_avgpool_s8()
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_depthwise_conv_s16/
Dtest_arm_depthwise_conv_s16.c48 input_dims.h = DW_INT16XINT8_INPUT_H; in dw_int16xint8_arm_depthwise_conv_s16()
51 filter_dims.h = DW_INT16XINT8_FILTER_Y; in dw_int16xint8_arm_depthwise_conv_s16()
53 output_dims.h = DW_INT16XINT8_OUTPUT_H; in dw_int16xint8_arm_depthwise_conv_s16()
57 dw_conv_params.padding.h = DW_INT16XINT8_PAD_Y; in dw_int16xint8_arm_depthwise_conv_s16()
59 dw_conv_params.stride.h = DW_INT16XINT8_STRIDE_Y; in dw_int16xint8_arm_depthwise_conv_s16()
61 dw_conv_params.dilation.h = DW_INT16XINT8_DILATION_Y; in dw_int16xint8_arm_depthwise_conv_s16()
145 input_dims.h = DW_INT16XINT8_DILATION_INPUT_H; in dw_int16xint8_dilation_arm_depthwise_conv_s16()
148 filter_dims.h = DW_INT16XINT8_DILATION_FILTER_Y; 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()
154 dw_conv_params.padding.h = DW_INT16XINT8_DILATION_PAD_Y; in dw_int16xint8_dilation_arm_depthwise_conv_s16()
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_depthwise_conv_s8/
Dtest_arm_depthwise_conv_s8.c50 input_dims.h = DEPTHWISE_2_INPUT_H; in depthwise_2_arm_depthwise_conv_s8()
53 filter_dims.h = DEPTHWISE_2_FILTER_Y; in depthwise_2_arm_depthwise_conv_s8()
55 output_dims.h = DEPTHWISE_2_OUTPUT_H; in depthwise_2_arm_depthwise_conv_s8()
59 dw_conv_params.padding.h = DEPTHWISE_2_PAD_Y; in depthwise_2_arm_depthwise_conv_s8()
61 dw_conv_params.stride.h = DEPTHWISE_2_STRIDE_Y; in depthwise_2_arm_depthwise_conv_s8()
63 dw_conv_params.dilation.h = DEPTHWISE_2_DILATION_Y; in depthwise_2_arm_depthwise_conv_s8()
146 input_dims.h = DEPTHWISE_MULT_BATCHES_INPUT_H; in depthwise_mult_batches_arm_depthwise_conv_s8()
149 filter_dims.h = DEPTHWISE_MULT_BATCHES_FILTER_Y; in depthwise_mult_batches_arm_depthwise_conv_s8()
151 output_dims.h = DEPTHWISE_MULT_BATCHES_OUTPUT_H; in depthwise_mult_batches_arm_depthwise_conv_s8()
155 dw_conv_params.padding.h = DEPTHWISE_MULT_BATCHES_PAD_Y; in depthwise_mult_batches_arm_depthwise_conv_s8()
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_grouped_convolve_s8/
Dtest_arm_grouped_convolve_s8.c51 input_dims.h = GROUPED_CONV_1_INPUT_H; in grouped_conv_arm_grouped_convolve_1_s8()
54 filter_dims.h = GROUPED_CONV_1_FILTER_Y; in grouped_conv_arm_grouped_convolve_1_s8()
57 output_dims.h = GROUPED_CONV_1_OUTPUT_H; in grouped_conv_arm_grouped_convolve_1_s8()
61 conv_params.padding.h = GROUPED_CONV_1_PAD_Y; in grouped_conv_arm_grouped_convolve_1_s8()
63 conv_params.stride.h = GROUPED_CONV_1_STRIDE_Y; in grouped_conv_arm_grouped_convolve_1_s8()
65 conv_params.dilation.h = GROUPED_CONV_1_DILATION_Y; in grouped_conv_arm_grouped_convolve_1_s8()
123 input_dims.h = GROUPED_CONV_2_INPUT_H; in grouped_conv_arm_grouped_convolve_2_s8()
126 filter_dims.h = GROUPED_CONV_2_FILTER_Y; in grouped_conv_arm_grouped_convolve_2_s8()
129 output_dims.h = GROUPED_CONV_2_OUTPUT_H; in grouped_conv_arm_grouped_convolve_2_s8()
133 conv_params.padding.h = GROUPED_CONV_2_PAD_Y; in grouped_conv_arm_grouped_convolve_2_s8()
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_depthwise_conv_3x3_s8/
Dtest_arm_depthwise_conv_3x3_s8.c49 input_dims.h = DEPTHWISE_KERNEL_3X3_INPUT_H; in depthwise_kernel_3x3_arm_depthwise_conv_3x3_s8()
52 filter_dims.h = DEPTHWISE_KERNEL_3X3_FILTER_Y; 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()
58 dw_conv_params.padding.h = DEPTHWISE_KERNEL_3X3_PAD_Y; 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()
62 dw_conv_params.dilation.h = DEPTHWISE_KERNEL_3X3_DILATION_Y; in depthwise_kernel_3x3_arm_depthwise_conv_3x3_s8()
151 input_dims.h = DEPTHWISE_KERNEL_3X3_INPUT_H; in depthwise_kernel_3x3_arm_depthwise_conv_3x3_1_s8()
154 filter_dims.h = DEPTHWISE_KERNEL_3X3_FILTER_Y; 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()
160 dw_conv_params.padding.h = DEPTHWISE_KERNEL_3X3_PAD_Y; in depthwise_kernel_3x3_arm_depthwise_conv_3x3_1_s8()
[all …]
/cmsis-nn-latest/Source/TransposeFunctions/
Darm_transpose_s8.c50 const int32_t h = input_dims->h; in arm_transpose_s8_nhcw() local
73 for (int32_t y = 0; y < h; y++) in arm_transpose_s8_nhcw()
133 const int32_t h = input_dims->h; in arm_transpose_s8_default() local
139 for (int32_t y = 0; y < h; y++) in arm_transpose_s8_default()
178 const int32_t h = input_dims->h; in arm_transpose_s8() local
182 in_strides[0] = h * w * c; in arm_transpose_s8()
195 const cmsis_nn_dims smaller_input_dims = {1, 1, n, h}; in arm_transpose_s8()
201 const cmsis_nn_dims smaller_input_dims = {1, n, h, w}; in arm_transpose_s8()
204 in_strides[1] = h * w; in arm_transpose_s8()
214 out_strides[perm[0] + 1] = output_dims->h * output_dims->w; in arm_transpose_s8()
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_reverse_transpose_conv_s8/
Dtest_arm_reverse_transpose_conv_s8.c51 input_dims.h = REVERSE_TRANSPOSE_CONV_1_INPUT_H; in reverse_transpose_conv_1_arm_transpose_conv_s8()
54 filter_dims.h = REVERSE_TRANSPOSE_CONV_1_FILTER_Y; in reverse_transpose_conv_1_arm_transpose_conv_s8()
59 output_dims.h = REVERSE_TRANSPOSE_CONV_1_OUTPUT_H; in reverse_transpose_conv_1_arm_transpose_conv_s8()
63 transpose_conv_params.padding.h = REVERSE_TRANSPOSE_CONV_1_PAD_Y; in reverse_transpose_conv_1_arm_transpose_conv_s8()
65 transpose_conv_params.padding_offsets.h = REVERSE_TRANSPOSE_CONV_1_PAD_Y_WITH_OFFSET; in reverse_transpose_conv_1_arm_transpose_conv_s8()
68 transpose_conv_params.stride.h = REVERSE_TRANSPOSE_CONV_1_STRIDE_Y; in reverse_transpose_conv_1_arm_transpose_conv_s8()
70 transpose_conv_params.dilation.h = REVERSE_TRANSPOSE_CONV_1_DILATION_Y; in reverse_transpose_conv_1_arm_transpose_conv_s8()
142 input_dims.h = REVERSE_TRANSPOSE_CONV_2_INPUT_H; in reverse_transpose_conv_2_arm_transpose_conv_s8()
145 filter_dims.h = REVERSE_TRANSPOSE_CONV_2_FILTER_Y; in reverse_transpose_conv_2_arm_transpose_conv_s8()
150 output_dims.h = REVERSE_TRANSPOSE_CONV_2_OUTPUT_H; in reverse_transpose_conv_2_arm_transpose_conv_s8()
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_transpose_conv_s8/
Dtest_arm_transpose_conv_s8.c51 input_dims.h = TRANSPOSE_CONV_1_INPUT_H; in transpose_conv_1_arm_transpose_conv_s8()
54 filter_dims.h = TRANSPOSE_CONV_1_FILTER_Y; in transpose_conv_1_arm_transpose_conv_s8()
59 output_dims.h = TRANSPOSE_CONV_1_OUTPUT_H; in transpose_conv_1_arm_transpose_conv_s8()
63 transpose_conv_params.padding.h = TRANSPOSE_CONV_1_PAD_Y; in transpose_conv_1_arm_transpose_conv_s8()
65 transpose_conv_params.padding_offsets.h = TRANSPOSE_CONV_1_PAD_Y_WITH_OFFSET; in transpose_conv_1_arm_transpose_conv_s8()
68 transpose_conv_params.stride.h = TRANSPOSE_CONV_1_STRIDE_Y; in transpose_conv_1_arm_transpose_conv_s8()
70 transpose_conv_params.dilation.h = TRANSPOSE_CONV_1_DILATION_Y; in transpose_conv_1_arm_transpose_conv_s8()
142 input_dims.h = TRANSPOSE_CONV_2_INPUT_H; in transpose_conv_2_arm_transpose_conv_s8()
145 filter_dims.h = TRANSPOSE_CONV_2_FILTER_Y; in transpose_conv_2_arm_transpose_conv_s8()
150 output_dims.h = TRANSPOSE_CONV_2_OUTPUT_H; in transpose_conv_2_arm_transpose_conv_s8()
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_depthwise_conv_s8_opt/
Dtest_arm_depthwise_conv_s8_opt.c51 input_dims.h = BASIC_INPUT_H; in basic_arm_depthwise_conv_s8_opt()
54 filter_dims.h = BASIC_FILTER_Y; in basic_arm_depthwise_conv_s8_opt()
56 output_dims.h = BASIC_OUTPUT_H; in basic_arm_depthwise_conv_s8_opt()
60 dw_conv_params.padding.h = BASIC_PAD_Y; in basic_arm_depthwise_conv_s8_opt()
62 dw_conv_params.stride.h = BASIC_STRIDE_Y; in basic_arm_depthwise_conv_s8_opt()
64 dw_conv_params.dilation.h = BASIC_DILATION_Y; in basic_arm_depthwise_conv_s8_opt()
153 input_dims.h = DEPTHWISE_EQ_IN_OUT_CH_INPUT_H; in depthwise_eq_in_out_ch_arm_depthwise_conv_s8_opt()
156 filter_dims.h = DEPTHWISE_EQ_IN_OUT_CH_FILTER_Y; 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()
162 dw_conv_params.padding.h = DEPTHWISE_EQ_IN_OUT_CH_PAD_Y; in depthwise_eq_in_out_ch_arm_depthwise_conv_s8_opt()
[all …]

1234