Searched refs:col_count (Results 1 – 4 of 4) sorted by relevance
83 int32_t col_count = num_col_a / COL_PER_LOOP; in arm_nn_mat_mult_kernel_s8_s16() local85 while (col_count) in arm_nn_mat_mult_kernel_s8_s16()113 col_count--; in arm_nn_mat_mult_kernel_s8_s16()117 col_count = (num_col_a & (COL_PER_LOOP - 1)) - 1; in arm_nn_mat_mult_kernel_s8_s16()118 while (col_count >= 0) in arm_nn_mat_mult_kernel_s8_s16()120 const int32_t b0 = ip_b0_s16[col_count]; in arm_nn_mat_mult_kernel_s8_s16()121 const int32_t b1 = ip_b1_s16[col_count]; in arm_nn_mat_mult_kernel_s8_s16()123 ch_0_out_n += b0 * ip_a0_s8[col_count]; in arm_nn_mat_mult_kernel_s8_s16()124 ch_1_out_n += b0 * ip_a1_s8[col_count]; in arm_nn_mat_mult_kernel_s8_s16()125 ch_2_out_n += b0 * ip_a2_s8[col_count]; in arm_nn_mat_mult_kernel_s8_s16()[all …]
78 uint16_t col_count = num_col_a / 4; in arm_nn_mat_mult_kernel_s8_s16_reordered() local80 while (col_count) in arm_nn_mat_mult_kernel_s8_s16_reordered()102 col_count--; in arm_nn_mat_mult_kernel_s8_s16_reordered()148 int32_t col_count = num_col_a / 4; in arm_nn_mat_mult_kernel_s8_s16_reordered() local149 while (col_count) in arm_nn_mat_mult_kernel_s8_s16_reordered()166 col_count--; in arm_nn_mat_mult_kernel_s8_s16_reordered()
280 uint16_t col_count = (kernel_x * kernel_y) / 2; in arm_depthwise_conv_s8_opt() local285 while (col_count) in arm_depthwise_conv_s8_opt()324 col_count--; in arm_depthwise_conv_s8_opt()327 col_count = (kernel_x * kernel_y) & 0x1; in arm_depthwise_conv_s8_opt()328 while (col_count) in arm_depthwise_conv_s8_opt()338 col_count--; in arm_depthwise_conv_s8_opt()372 const uint16_t col_count = (kernel_x * kernel_y); in arm_depthwise_conv_s8_opt() local375 for (int i = 0; i < col_count; i++) in arm_depthwise_conv_s8_opt()
274 uint16_t col_count = (input_ch * kernel_y * kernel_x) >> 2; in arm_convolve_s8()276 while (col_count) in arm_convolve_s8()288 col_count--; in arm_convolve_s8()291 col_count = input_ch * kernel_y * kernel_x & 0x3; in arm_convolve_s8()292 while (col_count) in arm_convolve_s8()297 col_count--; in arm_convolve_s8()
