Searched refs:out_buff (Results 1 – 3 of 3) sorted by relevance
77 int32_t out_buff[4]; in depthwise_conv_u8_mult_4() local79 out_buff[0] = 0; in depthwise_conv_u8_mult_4()80 out_buff[1] = 0; in depthwise_conv_u8_mult_4()81 out_buff[2] = 0; in depthwise_conv_u8_mult_4()82 out_buff[3] = 0; in depthwise_conv_u8_mult_4()93 … out_buff[0] += in_val * (kernel[ker_idx + 0 + mult_tile] + filter_offset); in depthwise_conv_u8_mult_4()94 … out_buff[1] += in_val * (kernel[ker_idx + 1 + mult_tile] + filter_offset); in depthwise_conv_u8_mult_4()95 … out_buff[2] += in_val * (kernel[ker_idx + 2 + mult_tile] + filter_offset); in depthwise_conv_u8_mult_4()96 … out_buff[3] += in_val * (kernel[ker_idx + 3 + mult_tile] + filter_offset); in depthwise_conv_u8_mult_4()102 out_buff[0] += bias[out_ch + 0 + mult_tile]; in depthwise_conv_u8_mult_4()[all …]
76 int32_t out_buff[4]; in depthwise_conv_s8_mult_4() local78 out_buff[0] = bias[out_ch + 0 + mult_tile]; in depthwise_conv_s8_mult_4()79 out_buff[1] = bias[out_ch + 1 + mult_tile]; in depthwise_conv_s8_mult_4()80 out_buff[2] = bias[out_ch + 2 + mult_tile]; in depthwise_conv_s8_mult_4()81 out_buff[3] = bias[out_ch + 3 + mult_tile]; in depthwise_conv_s8_mult_4()92 out_buff[0] += in_val * kernel[ker_idx + 0 + mult_tile]; in depthwise_conv_s8_mult_4()93 out_buff[1] += in_val * kernel[ker_idx + 1 + mult_tile]; in depthwise_conv_s8_mult_4()94 out_buff[2] += in_val * kernel[ker_idx + 2 + mult_tile]; in depthwise_conv_s8_mult_4()95 out_buff[3] += in_val * kernel[ker_idx + 3 + mult_tile]; in depthwise_conv_s8_mult_4()100 int32x4_t res = vldrwq_s32(out_buff); in depthwise_conv_s8_mult_4()[all …]
175 int32_t out_buff = bias[in_ch]; in arm_depthwise_conv_3x3_s8() local184 out_buff += (*(input_ptr) + input_offset) * *(kernel_ptr); in arm_depthwise_conv_3x3_s8()187 out_buff += (*(input_ptr + input_ch) + input_offset) * *(kernel_ptr + input_ch); in arm_depthwise_conv_3x3_s8()191 … out_buff += (*(input_ptr + (input_ch << 1)) + input_offset) * *(kernel_ptr + (input_ch << 1)); in arm_depthwise_conv_3x3_s8()198 out_buff = arm_nn_requantize(out_buff, output_mult[in_ch], output_shift[in_ch]); in arm_depthwise_conv_3x3_s8()199 out_buff += output_offset; in arm_depthwise_conv_3x3_s8()200 out_buff = MIN(MAX(out_buff, output_activation_min), output_activation_max); in arm_depthwise_conv_3x3_s8()201 output[out_idx++] = (int8_t)out_buff; in arm_depthwise_conv_3x3_s8()
