Home
last modified time | relevance | path

Searched defs:inp_sC (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DGridSampler.cpp64 int64_t inp_sC = input.stride(1); in grid_sampler_3d_cpu_impl() local
243 int64_t inp_sC = input.stride(1); in grid_sampler_3d_backward_cpu_impl() local
479 int64_t inp_sC = input.stride(1); in _grid_sampler_2d_cpu_quantized() local
581 int64_t inp_sC = input.stride(1); in _grid_sampler_2d_cpu_fallback() local
744 int64_t inp_sC = input.stride(1); in _grid_sampler_2d_cpu_fallback_backward() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DGridSampler.cu42 index_t inp_sC = input.strides[1]; in grid_sampler_2d_kernel() local
174 index_t inp_sC = input.strides[1]; in grid_sampler_3d_kernel() local
331 index_t inp_sC = input.strides[1]; in grid_sampler_2d_backward_kernel() local
539 index_t inp_sC = input.strides[1]; in grid_sampler_3d_backward_kernel() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DGridSamplerKernel.cpp470 const int64_t inp_sC; member
713 const int64_t inp_sC; member
821 const int64_t inp_sC; member