Searched refs:kernel_col (Results 1 – 5 of 5) sorted by relevance
43 int kernel_row, kernel_col; in RollingIm2ColPackDwUnitFp32() local52 for (kernel_col = 0; kernel_col < kernel_w; kernel_col++) { in RollingIm2ColPackDwUnitFp32()53 int input_col = -pad_left + kernel_col * dilation_w + input_w; in RollingIm2ColPackDwUnitFp32()93 int kernel_row, kernel_col; in rolling_im2col_hwc() local102 for (kernel_col = 0; kernel_col < kernel_w; kernel_col++) { in rolling_im2col_hwc()103 int input_col = -pad_left + kernel_col * dilation_w + input_w; in rolling_im2col_hwc()122 for (kernel_col = 0; kernel_col < kernel_w; kernel_col++) { in rolling_im2col_hwc()123 int input_col = -pad_left + kernel_col * dilation_w + input_w; in rolling_im2col_hwc()163 int channel, kernel_row, kernel_col, output_rows, output_col; in rolling_im2row_hwc() local166 for (kernel_col = 0; kernel_col < kernel_w; kernel_col++) { in rolling_im2row_hwc()[all …]
56 int kernel_col = k % k_w; in ConvDwInputGrad() local58 … int input_col = -conv_param->pad_l_ + kernel_col * conv_param->dilation_w_ + col_stride_offset; in ConvDwInputGrad()88 int kernel_col = k % k_w; in ConvDwInputGrad() local90 … int input_col = -conv_param->pad_l_ + kernel_col * conv_param->dilation_w_ + col_stride_offset; in ConvDwInputGrad()
42 int kernel_row, kernel_col; in RollingIm2ColPackDwUnitFp16() local51 for (kernel_col = 0; kernel_col < kernel_w; kernel_col++) { in RollingIm2ColPackDwUnitFp16()52 int input_col = -pad_left + kernel_col * dilation_w + input_w; in RollingIm2ColPackDwUnitFp16()92 int kernel_row, kernel_col; in RollingIm2ColPackUnitFp16() local101 for (kernel_col = 0; kernel_col < kernel_w; kernel_col++) { in RollingIm2ColPackUnitFp16()102 int input_col = -pad_left + kernel_col * dilation_w + input_w; in RollingIm2ColPackUnitFp16()121 for (kernel_col = 0; kernel_col < kernel_w; kernel_col++) { in RollingIm2ColPackUnitFp16()122 int input_col = -pad_left + kernel_col * dilation_w + input_w; in RollingIm2ColPackUnitFp16()158 int kernel_row, kernel_col; in RollingCol2ImPackUnitFp16() local168 for (kernel_col = 0; kernel_col < kernel_w; kernel_col++) { in RollingCol2ImPackUnitFp16()[all …]
54 int kernel_col = k % k_w; in ConvDwInputGrad16() local56 int input_col = kernel_col * conv_param->dilation_w_ + col_stride_offset; in ConvDwInputGrad16()160 int kernel_col = k % k_w; in ConvDwInputGrad8() local162 int input_col = kernel_col * conv_param->dilation_w_ + col_stride_offset; in ConvDwInputGrad8()243 int kernel_col = k % k_w; in ConvDwInputGrad4() local245 int input_col = kernel_col * conv_param->dilation_w_ + col_stride_offset; in ConvDwInputGrad4()319 int kernel_col = k % k_w; in ConvDwInputGradFp16() local321 int input_col = kernel_col * conv_param->dilation_w_ + col_stride_offset; in ConvDwInputGradFp16()
363 for (std::size_t kernel_col = 0; kernel_col < kernel.size(); ++kernel_col) in convolve_2d_impl() local365 flip_ker_col = kernel.size() - 1 - kernel_col; // column index of flipped kernel in convolve_2d_impl()