Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DGridSampler.cu344 index_t gInp_sC; in grid_sampler_2d_backward_kernel() local
555 int64_t gInp_sC = 0; in grid_sampler_3d_backward_kernel() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DGridSampler.cpp258 int64_t gInp_sC = 0; in grid_sampler_3d_backward_cpu_impl() local
756 int64_t gInp_sC = grad_input.stride(1); in _grid_sampler_2d_cpu_fallback_backward() local