Home
last modified time | relevance | path

Searched refs:kernel_x (Results 1 – 8 of 8) sorted by relevance

/hal_nxp-latest/mcux/mcux-sdk/CMSIS/NN/Source/ConvolutionFunctions/
Darm_convolve_1_x_n_s8.c74 const uint16_t kernel_x = filter_dims->w; in arm_convolve_1_x_n_s8() local
99 ker_end_idx[i] = MIN(kernel_x, input_x - est_input_x_idx); in arm_convolve_1_x_n_s8()
106 if ((ker_begin_idx[0] != 0) || (ker_end_idx[3] != kernel_x)) in arm_convolve_1_x_n_s8()
112 filter_data + (input_ch * kernel_x * i_out_ch) + in arm_convolve_1_x_n_s8()
118 filter_data + (input_ch * kernel_x * i_out_ch) + in arm_convolve_1_x_n_s8()
125 filter_data + (input_ch * kernel_x * i_out_ch) + in arm_convolve_1_x_n_s8()
132 filter_data + (input_ch * kernel_x * i_out_ch) + in arm_convolve_1_x_n_s8()
142 (void)arm_nn_mat_mul_core_4x_s8(kernel_x * input_ch, in arm_convolve_1_x_n_s8()
145 filter_data + (input_ch * kernel_x * i_out_ch), in arm_convolve_1_x_n_s8()
Darm_convolve_s8.c70 const uint16_t kernel_x = filter_dims->w; in arm_convolve_s8() local
97 const int32_t num_elem = kernel_x * kernel_y * input_ch; in arm_convolve_s8()
107 … for (int i_ker_x = i_out_x * stride_x - pad_x; i_ker_x < i_out_x * stride_x - pad_x + kernel_x; in arm_convolve_s8()
214 … for (i_ker_x = i_out_x * stride_x - pad_x; i_ker_x < i_out_x * stride_x - pad_x + kernel_x; in arm_convolve_s8()
235 if (two_column_buf == buffer_a + 2 * input_ch * kernel_y * kernel_x) in arm_convolve_s8()
245 input_ch * kernel_y * kernel_x, in arm_convolve_s8()
274 uint16_t col_count = (input_ch * kernel_y * kernel_x) >> 2; in arm_convolve_s8()
291 col_count = input_ch * kernel_y * kernel_x & 0x3; in arm_convolve_s8()
328 const int32_t ker_x_end = MIN(kernel_x, input_x - base_idx_x); in arm_convolve_s8()
340 filter_data[i_out_ch * input_ch * kernel_y * kernel_x + in arm_convolve_s8()
[all …]
Darm_depthwise_conv_u8_basic_ver1.c50 const int32_t kernel_x, in depthwise_conv_u8_mult_4() argument
86 … int32_t ker_idx = ker_h * (output_ch * kernel_x) + ker_w_start * output_ch + out_ch; in depthwise_conv_u8_mult_4()
89 for (int32_t ker_w = ker_w_start; ker_w < MIN(kernel_x, input_x - in_w); in depthwise_conv_u8_mult_4()
139 const int32_t kernel_x, in depthwise_conv_u8_generic() argument
176 const int ker_x_end = MIN(kernel_x, input_x - base_idx_x); in depthwise_conv_u8_generic()
186 … int32_t ker_idx_0 = (i_ker_y * kernel_x + i_ker_x) * (input_ch * ch_mult) + idx_out_ch; in depthwise_conv_u8_generic()
251 const uint16_t kernel_x, in arm_depthwise_conv_u8_basic_ver1() argument
284 kernel_x, in arm_depthwise_conv_u8_basic_ver1()
311 kernel_x, in arm_depthwise_conv_u8_basic_ver1()
Darm_depthwise_conv_s8_opt.c75 const int32_t kernel_x = filter_dims->w; in arm_depthwise_conv_s8_opt() local
98 const int32_t kernel_size = kernel_x * kernel_y; in arm_depthwise_conv_s8_opt()
107 for (int i_ker_x = base_idx_x; i_ker_x < base_idx_x + kernel_x; i_ker_x++) in arm_depthwise_conv_s8_opt()
235 memset(&col_buffer[index], 0, (kernel_x * input_ch) * ker_y_start * sizeof(q15_t)); in arm_depthwise_conv_s8_opt()
236 index += (kernel_x * input_ch) * ker_y_start; in arm_depthwise_conv_s8_opt()
243 for (int i_ker_x = 0; i_ker_x < kernel_x; i_ker_x++) in arm_depthwise_conv_s8_opt()
264 memset(&col_buffer[index], 0, (kernel_x * input_ch) * diff * sizeof(q15_t)); in arm_depthwise_conv_s8_opt()
280 uint16_t col_count = (kernel_x * kernel_y) / 2; in arm_depthwise_conv_s8_opt()
327 col_count = (kernel_x * kernel_y) & 0x1; in arm_depthwise_conv_s8_opt()
372 const uint16_t col_count = (kernel_x * kernel_y); in arm_depthwise_conv_s8_opt()
Darm_depthwise_conv_s8.c50 const int32_t kernel_x, in depthwise_conv_s8_mult_4() argument
85 … int32_t ker_idx = ker_h * (output_ch * kernel_x) + ker_w_start * output_ch + out_ch; in depthwise_conv_s8_mult_4()
88 for (int32_t ker_w = ker_w_start; ker_w < MIN(kernel_x, input_x - in_w); in depthwise_conv_s8_mult_4()
150 const uint16_t kernel_x, in depthwise_conv_s8_generic() argument
190 const int ker_x_end = MIN(kernel_x, input_x - base_idx_x); in depthwise_conv_s8_generic()
200 … int32_t ker_idx_0 = (i_ker_y * kernel_x + i_ker_x) * (input_ch * ch_mult) + idx_out_ch; in depthwise_conv_s8_generic()
/hal_nxp-latest/mcux/mcux-sdk/CMSIS/NN/Source/PoolingFunctions/
Darm_avgpool_s8.c90 const int32_t kernel_x = filter_dims->w; in arm_avgpool_s8() local
115 k_x_end = MIN(i_x * stride_x - pad_x + kernel_x, input_x); in arm_avgpool_s8()
258 const int32_t kernel_x = filter_dims->w; in arm_avgpool_s8() local
282 const int32_t kernel_x_end = MIN(kernel_x, input_x - idx_x); in arm_avgpool_s8()
338 … for (k_x = i_x * stride_x - pad_x; k_x < i_x * stride_x - pad_x + kernel_x; k_x++) in arm_avgpool_s8()
Darm_max_pool_s8.c177 const int32_t kernel_x = filter_dims->w; in arm_max_pool_s8() local
196 const int32_t kernel_x_end = MIN(kernel_x, input_x - base_idx_x); in arm_max_pool_s8()
/hal_nxp-latest/mcux/mcux-sdk/CMSIS/NN/Include/
Darm_nnfunctions.h1824 const uint16_t kernel_x,