Searched refs:kernel_extent_h (Results 1 – 2 of 2) sorted by relevance
159 auto kernel_extent_h = (kernel_h - 1) * dilation_h + 1; in vol2im_kernel() local166 (h_im < kernel_extent_h) ? 0 : (h_im - kernel_extent_h) / stride_h + 1; in vol2im_kernel()
128 int64_t kernel_extent_h = (kernel_h - 1) * dilation_height + 1; in col2im_device() local134 const int64_t h_col_start = (h_im < kernel_extent_h) in col2im_device()136 : (h_im - kernel_extent_h) / stride_height + 1; in col2im_device()