Lines Matching refs:output_ch
80 const int32_t output_ch = output_dims->c; in arm_transpose_conv_s8() local
112 … arm_memset_s8((int8_t *)img_buf_ptr, 0, output_x * output_y * output_ch * sizeof(int32_t)); in arm_transpose_conv_s8()
120 memcpy(img_data, bias_data, output_ch * sizeof(int32_t)); in arm_transpose_conv_s8()
121 img_data += output_ch; in arm_transpose_conv_s8()
130 for (int i_output_ch = 0; i_output_ch < output_ch; i_output_ch++) in arm_transpose_conv_s8()
139 output_ch); in arm_transpose_conv_s8()
155 int32_t *dst_data = img_data + (i_pad_y * output_x + i_pad_x) * output_ch; in arm_transpose_conv_s8()
163 for (int i_output_ch = 0; i_output_ch < output_ch; i_output_ch++) in arm_transpose_conv_s8()
168 dst_data += output_ch; in arm_transpose_conv_s8()
169 col_buf += output_ch; in arm_transpose_conv_s8()
171 dst_data += (output_x - kernel_x) * output_ch; in arm_transpose_conv_s8()
181 for (int32_t i_channel_rmdr = output_ch; i_channel_rmdr > 0; i_channel_rmdr -= 4) in arm_transpose_conv_s8()
195 output_data_ptr += output_ch; in arm_transpose_conv_s8()
198 for (; i_output_ch < output_ch; i_output_ch++) in arm_transpose_conv_s8()
208 img_data += output_ch; in arm_transpose_conv_s8()