Home
last modified time | relevance | path

Searched refs:check_grid_sampler_common (Results 1 – 6 of 6) sorted by relevance

/external/pytorch/aten/src/ATen/native/cudnn/
DGridSampler.cpp83 check_grid_sampler_common(input_t, grid_t); in cudnn_grid_sampler_forward()
133 check_grid_sampler_common(input_t, grid_t); in cudnn_grid_sampler_backward()
/external/pytorch/aten/src/ATen/native/
DGridSampler.cpp47 check_grid_sampler_common(input, grid); in grid_sampler_3d_cpu_impl()
212 check_grid_sampler_common(input, grid); in grid_sampler_3d_backward_cpu_impl()
454 check_grid_sampler_common(input, grid); in _grid_sampler_2d_cpu_quantized()
563 check_grid_sampler_common(input, grid); in _grid_sampler_2d_cpu_fallback()
719 check_grid_sampler_common(input, grid); in _grid_sampler_2d_cpu_fallback_backward()
924 check_grid_sampler_common(input, grid); in grid_sampler_2d_cpu()
968 check_grid_sampler_common(input, grid); in grid_sampler_3d_cpu()
984 check_grid_sampler_common(input, grid); in grid_sampler_2d_backward_cpu()
1033 check_grid_sampler_common(input, grid); in grid_sampler_3d_backward_cpu()
DGridSamplerUtils.h22 inline void check_grid_sampler_common( in check_grid_sampler_common() function
/external/pytorch/aten/src/ATen/native/mps/operations/
DGridSampler.mm24 check_grid_sampler_common(input, grid);
/external/pytorch/aten/src/ATen/native/cuda/
DGridSampler.cu755 check_grid_sampler_common(input, grid); in launch_grid_sampler_2d_forward_kernel()
799 check_grid_sampler_common(input, grid); in launch_grid_sampler_3d_forward_kernel()
846 check_grid_sampler_common(input, grid); in launch_grid_sampler_2d_backward_kernel()
909 check_grid_sampler_common(input, grid); in launch_grid_sampler_3d_backward_kernel()
/external/pytorch/torch/
D_meta_registrations.py4627 def check_grid_sampler_common(input: Tensor, grid: Tensor): function
4719 check_grid_sampler_common(input, grid)
4740 check_grid_sampler_common(input, grid)