Home
last modified time | relevance | path

Searched refs:grid_sampler_compute_source_index (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DGridSampler.cpp95 ix = grid_sampler_compute_source_index(ix, inp_W, padding_mode, align_corners); in grid_sampler_3d_cpu_impl()
96 iy = grid_sampler_compute_source_index(iy, inp_H, padding_mode, align_corners); in grid_sampler_3d_cpu_impl()
97 iz = grid_sampler_compute_source_index(iz, inp_D, padding_mode, align_corners); in grid_sampler_3d_cpu_impl()
504 float ix = grid_sampler_compute_source_index( in _grid_sampler_2d_cpu_quantized()
506 float iy = grid_sampler_compute_source_index( in _grid_sampler_2d_cpu_quantized()
607 scalar_t ix = grid_sampler_compute_source_index(x, inp_W, padding_mode, align_corners); in _grid_sampler_2d_cpu_fallback()
608 scalar_t iy = grid_sampler_compute_source_index(y, inp_H, padding_mode, align_corners); in _grid_sampler_2d_cpu_fallback()
H A DGridSampler.h163 static inline scalar_t grid_sampler_compute_source_index( in grid_sampler_compute_source_index() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DGridSampler.cu64 opmath_t ix = grid_sampler_compute_source_index(x, inp_W, padding_mode, align_corners); in grid_sampler_2d_kernel()
65 opmath_t iy = grid_sampler_compute_source_index(y, inp_H, padding_mode, align_corners); in grid_sampler_2d_kernel()
201 opmath_t ix = grid_sampler_compute_source_index(x, inp_W, padding_mode, align_corners); in grid_sampler_3d_kernel()
202 opmath_t iy = grid_sampler_compute_source_index(y, inp_H, padding_mode, align_corners); in grid_sampler_3d_kernel()
203 opmath_t iz = grid_sampler_compute_source_index(z, inp_D, padding_mode, align_corners); in grid_sampler_3d_kernel()
H A DGridSampler.cuh174 scalar_t grid_sampler_compute_source_index( in grid_sampler_compute_source_index() function