Searched refs:grid_sampler_unnormalize (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | GridSampler.h | 26 static inline scalar_t grid_sampler_unnormalize(scalar_t coord, int64_t size, in grid_sampler_unnormalize() function
168 coord = grid_sampler_unnormalize(coord, size, align_corners); in grid_sampler_compute_source_index()
|
| H A D | GridSampler.cpp | 670 ix = grid_sampler_unnormalize(x, inp_W, align_corners); in _grid_sampler_2d_cpu_fallback()
671 iy = grid_sampler_unnormalize(y, inp_H, align_corners); in _grid_sampler_2d_cpu_fallback()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | GridSampler.cuh | 22 scalar_t grid_sampler_unnormalize(scalar_t coord, int size, bool align_corners) { in grid_sampler_unnormalize() function
179 coord = grid_sampler_unnormalize(coord, size, align_corners); in grid_sampler_compute_source_index()
|
| H A D | GridSampler.cu | 119 ix = grid_sampler_unnormalize(x, inp_W, align_corners); in grid_sampler_2d_kernel()
120 iy = grid_sampler_unnormalize(y, inp_H, align_corners); in grid_sampler_2d_kernel()
|