Searched defs:grad_grid (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | GridSampler.cpp | 56 auto grad_grid = at::empty_like(grid, LEGACY_CONTIGUOUS_MEMORY_FORMAT); in grid_sampler_2d_backward_cuda() local
75 auto grad_grid = at::empty_like(grid, LEGACY_CONTIGUOUS_MEMORY_FORMAT); in grid_sampler_3d_backward_cuda() local
|
| H A D | GridSampler.cu | 318 TensorInfo<scalar_t, index_t> grad_grid, // initialized to empty in grid_sampler_2d_backward_kernel()
524 TensorInfo<scalar_t, index_t> grad_grid, // initialized to empty in grid_sampler_3d_backward_kernel()
840 const TensorBase &grad_input, const TensorBase &grad_grid, in launch_grid_sampler_2d_backward_kernel()
903 const TensorBase &grad_input, const TensorBase &grad_grid, in launch_grid_sampler_3d_backward_kernel()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | AffineGridGenerator.cpp | 103 const Tensor& grad_grid, in affine_grid_generator_4D_backward()
118 const Tensor& grad_grid, in affine_grid_generator_5D_backward()
|
| H A D | GridSampler.cpp | 224 auto grad_grid = at::empty_like(grid, LEGACY_CONTIGUOUS_MEMORY_FORMAT); in grid_sampler_3d_backward_cpu_impl() local
727 auto grad_grid = at::empty_like(grid, LEGACY_CONTIGUOUS_MEMORY_FORMAT); in _grid_sampler_2d_cpu_fallback_backward() local
1018 auto grad_grid = at::empty_like(grid, LEGACY_CONTIGUOUS_MEMORY_FORMAT); in grid_sampler_2d_backward_cpu() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
| H A D | AffineGridGenerator.cpp | 104 TensorArg grad_grid{grad_grid_contig, "grad_grid", 1}; in cudnn_affine_grid_generator_backward() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesModules.cpp | 169 auto grad_grid = std::get<1>(bw_out); in grid_sample_backward_helper_out() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | GridSamplerKernel.cpp | 1211 const TensorBase &grad_grid, in grid_sampler_2d_backward_cpu_kernel_impl()
|