Home
last modified time | relevance | path

Searched defs:w_k (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/aten/src/ATen/native/cuda/
Dvol2col.cuh177 uint64_t w_k = (w_im - w_col * stride_w); in vol2im_kernel() local
Dim2col.cuh143 int64_t w_k = (w_im - w_col * stride_width); in col2im_device() local