Searched refs:grid_sampler_unnormalize_set_grad (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | GridSampler.h | 42 static inline scalar_t grid_sampler_unnormalize_set_grad(scalar_t coord, int64_t size, in grid_sampler_unnormalize_set_grad() function
185 coord = grid_sampler_unnormalize_set_grad(coord, size, align_corners, grad_in); in grid_sampler_compute_source_index_set_grad()
|
| H A D | GridSampler.cpp | 860 ix = grid_sampler_unnormalize_set_grad(x, inp_W, align_corners, &gix_mult); in _grid_sampler_2d_cpu_fallback_backward()
861 iy = grid_sampler_unnormalize_set_grad(y, inp_H, align_corners, &giy_mult); in _grid_sampler_2d_cpu_fallback_backward()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | GridSampler.cuh | 38 scalar_t grid_sampler_unnormalize_set_grad(scalar_t coord, int size, in grid_sampler_unnormalize_set_grad() function
197 coord = grid_sampler_unnormalize_set_grad(coord, size, align_corners, grad_in); in grid_sampler_compute_source_index_set_grad()
|
| H A D | GridSampler.cu | 455 ix = grid_sampler_unnormalize_set_grad(x, inp_W, align_corners, &gix_mult); in grid_sampler_2d_backward_kernel()
456 iy = grid_sampler_unnormalize_set_grad(y, inp_H, align_corners, &giy_mult); in grid_sampler_2d_backward_kernel()
|