Searched refs:grid_sampler_compute_source_index_set_grad (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | GridSampler.cu | 367 …scalar_t ix = grid_sampler_compute_source_index_set_grad(x, inp_W, padding_mode, align_corners, &g… in grid_sampler_2d_backward_kernel()
368 …scalar_t iy = grid_sampler_compute_source_index_set_grad(y, inp_H, padding_mode, align_corners, &g… in grid_sampler_2d_backward_kernel()
582 …ix = grid_sampler_compute_source_index_set_grad(ix, inp_W, padding_mode, align_corners, &gix_mult); in grid_sampler_3d_backward_kernel()
583 …iy = grid_sampler_compute_source_index_set_grad(iy, inp_H, padding_mode, align_corners, &giy_mult); in grid_sampler_3d_backward_kernel()
584 …iz = grid_sampler_compute_source_index_set_grad(iz, inp_D, padding_mode, align_corners, &giz_mult); in grid_sampler_3d_backward_kernel()
|
| H A D | GridSampler.cuh | 190 scalar_t grid_sampler_compute_source_index_set_grad( in grid_sampler_compute_source_index_set_grad() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | GridSampler.cpp | 296 …ix = grid_sampler_compute_source_index_set_grad(ix, inp_W, padding_mode, align_corners, &gix_mult); in grid_sampler_3d_backward_cpu_impl()
297 …iy = grid_sampler_compute_source_index_set_grad(iy, inp_H, padding_mode, align_corners, &giy_mult); in grid_sampler_3d_backward_cpu_impl()
298 …iz = grid_sampler_compute_source_index_set_grad(iz, inp_D, padding_mode, align_corners, &giz_mult); in grid_sampler_3d_backward_cpu_impl()
782 …scalar_t ix = grid_sampler_compute_source_index_set_grad(x, inp_W, padding_mode, align_corners, &g… in _grid_sampler_2d_cpu_fallback_backward()
783 …scalar_t iy = grid_sampler_compute_source_index_set_grad(y, inp_H, padding_mode, align_corners, &g… in _grid_sampler_2d_cpu_fallback_backward()
|
| H A D | GridSampler.h | 178 static inline scalar_t grid_sampler_compute_source_index_set_grad( in grid_sampler_compute_source_index_set_grad() function
|