Searched refs:i_out_ch (Results 1 – 2 of 2) sorted by relevance
58 for (int i_out_ch = 0; i_out_ch < output_ch; i_out_ch++) in arm_nn_mat_mult_s8() local61 const int8_t *ip_r0 = input_row + (i_out_ch * row_len); in arm_nn_mat_mult_s8()107 res = vaddq_n_s32(res, bias[i_out_ch]); in arm_nn_mat_mult_s8()109 res = arm_requantize_mve(res, output_mult[i_out_ch], output_shift[i_out_ch]); in arm_nn_mat_mult_s8()116 vstrbq_scatter_offset_s32(&out[i_out_ch], scatter_offset, res); in arm_nn_mat_mult_s8()124 for (int i_out_ch = 0; i_out_ch < output_ch; i_out_ch++) in arm_nn_mat_mult_s8() local128 const int8_t *ip_r0 = input_row + (i_out_ch * row_len); in arm_nn_mat_mult_s8()150 acc_0 += bias[i_out_ch]; in arm_nn_mat_mult_s8()152 acc_0 = arm_nn_requantize(acc_0, output_mult[i_out_ch], output_shift[i_out_ch]); in arm_nn_mat_mult_s8()156 out[i_out_ch] = (int8_t)acc_0; in arm_nn_mat_mult_s8()
91 for (int32_t i_out_ch = 0; i_out_ch < output_ch; i_out_ch++) in arm_convolve_s16() local93 const int32_t reduced_multiplier = REDUCE_MULTIPLIER(output_mult[i_out_ch]); in arm_convolve_s16()120 filter_data[i_out_ch * input_ch * kernel_y * kernel_x + in arm_convolve_s16()128 conv_out_acc += bias_data[i_out_ch]; in arm_convolve_s16()131 …int32_t conv_out = arm_nn_requantize_s64(conv_out_acc, reduced_multiplier, output_shift[i_out_ch]); in arm_convolve_s16()134 … output_data[i_out_ch + (i_out_y * output_x + i_out_x) * output_ch] = (int16_t)conv_out; in arm_convolve_s16()
