Home
last modified time | relevance | path

Searched refs:kernel_y (Results 1 – 7 of 7) sorted by relevance

/hal_nxp-latest/mcux/mcux-sdk/CMSIS/NN/Source/ConvolutionFunctions/
Darm_convolve_s8.c71 const uint16_t kernel_y = filter_dims->h; in arm_convolve_s8() local
97 const int32_t num_elem = kernel_x * kernel_y * input_ch; in arm_convolve_s8()
104 … for (int i_ker_y = i_out_y * stride_y - pad_y; i_ker_y < i_out_y * stride_y - pad_y + kernel_y; in arm_convolve_s8()
212 … (i_ker_y = i_out_y * stride_y - pad_y; i_ker_y < i_out_y * stride_y - pad_y + kernel_y; i_ker_y++) 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()
327 const int32_t ker_y_end = MIN(kernel_y, input_y - base_idx_y); in arm_convolve_s8()
340 filter_data[i_out_ch * input_ch * kernel_y * kernel_x + in arm_convolve_s8()
Darm_depthwise_conv_u8_basic_ver1.c51 const int32_t kernel_y, in depthwise_conv_u8_mult_4() argument
84 … for (int32_t ker_h = ker_h_start; ker_h < MIN(kernel_y, input_y - in_h); ++ker_h) in depthwise_conv_u8_mult_4()
140 const int32_t kernel_y, in depthwise_conv_u8_generic() argument
175 const int ker_y_end = MIN(kernel_y, input_y - base_idx_y); in depthwise_conv_u8_generic()
252 const uint16_t kernel_y, in arm_depthwise_conv_u8_basic_ver1() argument
285 kernel_y, in arm_depthwise_conv_u8_basic_ver1()
312 kernel_y, in arm_depthwise_conv_u8_basic_ver1()
Darm_depthwise_conv_s8_opt.c76 const int32_t kernel_y = filter_dims->h; in arm_depthwise_conv_s8_opt() local
98 const int32_t kernel_size = kernel_x * kernel_y; in arm_depthwise_conv_s8_opt()
105 for (int i_ker_y = base_idx_y; i_ker_y < base_idx_y + kernel_y; i_ker_y++) in arm_depthwise_conv_s8_opt()
230 const int ker_y_end = MIN(kernel_y, input_y - base_idx_y); in arm_depthwise_conv_s8_opt()
261 const int diff = kernel_y - ker_y_end; 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.c51 const int32_t kernel_y, in depthwise_conv_s8_mult_4() argument
83 … for (int32_t ker_h = ker_h_start; ker_h < MIN(kernel_y, input_y - in_h); ++ker_h) in depthwise_conv_s8_mult_4()
151 const uint16_t kernel_y, in depthwise_conv_s8_generic() argument
189 const int ker_y_end = MIN(kernel_y, input_y - base_idx_y); in depthwise_conv_s8_generic()
/hal_nxp-latest/mcux/mcux-sdk/CMSIS/NN/Source/PoolingFunctions/
Darm_avgpool_s8.c89 const int32_t kernel_y = filter_dims->h; in arm_avgpool_s8() local
112 k_y_end = MIN(i_y * stride_y - pad_y + kernel_y, input_y); in arm_avgpool_s8()
257 const int32_t kernel_y = filter_dims->h; in arm_avgpool_s8() local
281 const int32_t kernel_y_end = MIN(kernel_y, input_y - idx_y); in arm_avgpool_s8()
336 for (k_y = i_y * stride_y - pad_y; k_y < i_y * stride_y - pad_y + kernel_y; k_y++) in arm_avgpool_s8()
Darm_max_pool_s8.c176 const int32_t kernel_y = filter_dims->h; in arm_max_pool_s8() local
195 const int32_t kernel_y_end = MIN(kernel_y, input_y - base_idx_y); in arm_max_pool_s8()
/hal_nxp-latest/mcux/mcux-sdk/CMSIS/NN/Include/
Darm_nnfunctions.h1825 const uint16_t kernel_y,