Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DGridSampler.cuh252 const index_t NC_offset, in safe_add_2d()
268 const index_t NC_offset, in safe_add_3d()
286 const index_t NC_offset, in add_value_bounded()
H A DGridSampler.cu389 index_t NC_offset = n * gInp_sN; in grid_sampler_2d_backward_kernel() local
439 index_t NC_offset = n * gInp_sN; in grid_sampler_2d_backward_kernel() local
478 index_t NC_offset = n * gInp_sN; in grid_sampler_2d_backward_kernel() local
634 index_t NC_offset; in grid_sampler_3d_backward_kernel() local
729 index_t NC_offset = n * gInp_sN; in grid_sampler_3d_backward_kernel() local