Searched refs:kernel_row (Results 1 – 5 of 5) sorted by relevance
43 int kernel_row, kernel_col; in RollingIm2ColPackDwUnitFp32() local50 for (kernel_row = 0; kernel_row < kernel_h; kernel_row++) { in RollingIm2ColPackDwUnitFp32()51 int input_row = -pad_up + kernel_row * dilation_h + input_h; in RollingIm2ColPackDwUnitFp32()93 int kernel_row, kernel_col; in rolling_im2col_hwc() local100 for (kernel_row = 0; kernel_row < kernel_h; kernel_row++) { in rolling_im2col_hwc()101 int input_row = -pad_up + kernel_row * dilation_h + input_h; in rolling_im2col_hwc()120 for (kernel_row = 0; kernel_row < kernel_h; kernel_row++) { in rolling_im2col_hwc()121 int input_row = -pad_up + kernel_row * dilation_h + input_h; in rolling_im2col_hwc()163 int channel, kernel_row, kernel_col, output_rows, output_col; in rolling_im2row_hwc() local165 for (kernel_row = 0; kernel_row < kernel_h; kernel_row++) { in rolling_im2row_hwc()[all …]
55 int kernel_row = k / k_w; in ConvDwInputGrad() local57 … int input_row = -conv_param->pad_u_ + kernel_row * conv_param->dilation_h_ + row_stride_offset; in ConvDwInputGrad()87 int kernel_row = k / k_w; in ConvDwInputGrad() local89 … int input_row = -conv_param->pad_u_ + kernel_row * conv_param->dilation_h_ + row_stride_offset; in ConvDwInputGrad()
42 int kernel_row, kernel_col; in RollingIm2ColPackDwUnitFp16() local49 for (kernel_row = 0; kernel_row < kernel_h; kernel_row++) { in RollingIm2ColPackDwUnitFp16()50 int input_row = -pad_up + kernel_row * dilation_h + input_h; in RollingIm2ColPackDwUnitFp16()92 int kernel_row, kernel_col; in RollingIm2ColPackUnitFp16() local99 for (kernel_row = 0; kernel_row < kernel_h; kernel_row++) { in RollingIm2ColPackUnitFp16()100 int input_row = -pad_up + kernel_row * dilation_h + input_h; in RollingIm2ColPackUnitFp16()119 for (kernel_row = 0; kernel_row < kernel_h; kernel_row++) { in RollingIm2ColPackUnitFp16()120 int input_row = -pad_up + kernel_row * dilation_h + input_h; in RollingIm2ColPackUnitFp16()158 int kernel_row, kernel_col; in RollingCol2ImPackUnitFp16() local166 for (kernel_row = 0; kernel_row < kernel_h; kernel_row++) { in RollingCol2ImPackUnitFp16()[all …]
53 int kernel_row = k / k_w; in ConvDwInputGrad16() local55 int input_row = kernel_row * conv_param->dilation_h_ + row_stride_offset; in ConvDwInputGrad16()159 int kernel_row = k / k_w; in ConvDwInputGrad8() local161 int input_row = kernel_row * conv_param->dilation_h_ + row_stride_offset; in ConvDwInputGrad8()242 int kernel_row = k / k_w; in ConvDwInputGrad4() local244 int input_row = kernel_row * conv_param->dilation_h_ + row_stride_offset; in ConvDwInputGrad4()318 int kernel_row = k / k_w; in ConvDwInputGradFp16() local320 int input_row = kernel_row * conv_param->dilation_h_ + row_stride_offset; in ConvDwInputGradFp16()
359 for (std::size_t kernel_row = 0; kernel_row < kernel.size(); ++kernel_row) in convolve_2d_impl() local361 flip_ker_row = kernel.size() - 1 - kernel_row; // row index of flipped kernel in convolve_2d_impl()