Home
last modified time | relevance | path

Searched full:grid_sampler (Results 1 – 25 of 27) sorted by relevance

12

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DGridSamplerUtils.h21 // See NOTE [ grid_sampler Native Functions ].
31 "grid_sampler(): expected input to not be undefined"); in check_grid_sampler_common()
34 "grid_sampler(): expected grid to not be undefined"); in check_grid_sampler_common()
37 "grid_sampler(): expected input and grid to be on same device, but input " in check_grid_sampler_common()
41 "grid_sampler(): expected input and grid to have torch.strided layout, but " in check_grid_sampler_common()
45 "grid_sampler(): expected grid and input to have same batch size, but got " in check_grid_sampler_common()
49 "grid_sampler(): expected grid to have size ", input.dim() - 2, " in last " in check_grid_sampler_common()
54 "grid_sampler(): expected input to have non-empty spatial dimensions, " in check_grid_sampler_common()
60 // See NOTE [ grid_sampler Native Functions ].
67 "grid_sampler(): expected 4D input and grid with same number of " in check_grid_sampler_2d()
[all …]
H A DGridSampler.cpp45 // See NOTE [ grid_sampler Native Functions ]. in grid_sampler_3d_cpu_impl()
46 // Add checks here in case this is called instead of grid_sampler. in grid_sampler_3d_cpu_impl()
210 // See NOTE [ grid_sampler Native Functions ]. in grid_sampler_3d_backward_cpu_impl()
211 // Add checks here in case this is called instead of grid_sampler. in grid_sampler_3d_backward_cpu_impl()
452 // See NOTE [ grid_sampler Native Functions ]. in _grid_sampler_2d_cpu_quantized()
453 // Add checks here in case this is called instead of grid_sampler. in _grid_sampler_2d_cpu_quantized()
561 // See NOTE [ grid_sampler Native Functions ]. in _grid_sampler_2d_cpu_fallback()
562 // Add checks here in case this is called instead of grid_sampler. in _grid_sampler_2d_cpu_fallback()
717 // See NOTE [ grid_sampler Native Functions ]. in _grid_sampler_2d_cpu_fallback_backward()
718 // Add checks here in case this is called instead of grid_sampler. in _grid_sampler_2d_cpu_fallback_backward()
[all …]
H A DGridSampler.h40 // This is useful in the backward pass of grid_sampler.
63 // This is useful in the backward pass of grid_sampler.
109 // This is useful in the backward pass of grid_sampler.
176 // This is useful in the backward pass of grid_sampler.
H A DTensorProperties.cpp87 // cuDNN functions like grid_sampler returns CUDNN_STATUS_BAD_PARAM on empty in cudnn_is_acceptable()
H A Dnative_functions.yaml2855 # NOTE [ grid_sampler Native Functions ]
2856 # `grid_sampler` is _supposed to_ do all the shape checking and then dispatch to
2872 - func: grid_sampler(Tensor input, Tensor grid, int interpolation_mode, int padding_mode, bool alig…
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A DGridSampler.cpp81 // See NOTE [ grid_sampler Native Functions ]. in cudnn_grid_sampler_forward()
82 // Add checks here in case this is called instead of grid_sampler. in cudnn_grid_sampler_forward()
131 // See NOTE [ grid_sampler Native Functions ]. in cudnn_grid_sampler_backward()
132 // Add checks here in case this is called instead of grid_sampler. in cudnn_grid_sampler_backward()
/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_opset16.py43 # note (mkozuki): Why `grid_sampler` instead of `grid_sample`?
44 # Because `torch.nn.functional.grid_sample` calls `torch.grid_sampler`.
45 @_onnx_symbolic("aten::grid_sampler")
47 def grid_sampler( function
H A Dsymbolic_opset20.py47 @_onnx_symbolic("aten::grid_sampler")
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DGridSampler.cuh35 // This is useful in the backward pass of grid_sampler.
60 // This is useful in the backward pass of grid_sampler.
106 // This is useful in the backward pass of grid_sampler.
187 // This is useful in the backward pass of grid_sampler.
H A DGridSampler.cu753 // See NOTE [ grid_sampler Native Functions ]. in launch_grid_sampler_2d_forward_kernel()
754 // Add checks here in case this is called instead of grid_sampler. in launch_grid_sampler_2d_forward_kernel()
797 // See NOTE [ grid_sampler Native Functions ]. in launch_grid_sampler_3d_forward_kernel()
798 // Add checks here in case this is called instead of grid_sampler. in launch_grid_sampler_3d_forward_kernel()
844 // See NOTE [ grid_sampler Native Functions ]. in launch_grid_sampler_2d_backward_kernel()
845 // Add checks here in case this is called instead of grid_sampler. in launch_grid_sampler_2d_backward_kernel()
907 // See NOTE [ grid_sampler Native Functions ]. in launch_grid_sampler_3d_backward_kernel()
908 // Add checks here in case this is called instead of grid_sampler. in launch_grid_sampler_3d_backward_kernel()
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A Dautocast_mode.cpp315 KERNEL_MPS(grid_sampler, promote) in TORCH_LIBRARY_IMPL()
355 KERNEL_CPU(grid_sampler, fp32) in TORCH_LIBRARY_IMPL()
H A Dautocast_mode.h924 _(grid_sampler) \
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesModules.cpp355 VMAP_SUPPORT(grid_sampler_2d, GRID_SAMPLE_BATCH_RULE(grid_sampler)); in TORCH_LIBRARY_IMPL()
358 VMAP_SUPPORT(grid_sampler_3d, GRID_SAMPLE_BATCH_RULE(grid_sampler)); in TORCH_LIBRARY_IMPL()
H A DBatchRulesDecompositions.cpp145 OP_DECOMPOSE(grid_sampler); in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/
H A Dvision.h91 return torch::grid_sampler( in grid_sample()
/aosp_15_r20/external/pytorch/functorch/op_analysis/
H A Dpublic_api346 grid_sampler
H A Dannotated_ops129 grid_sampler, misc
/aosp_15_r20/external/pytorch/test/mobile/model_test/
H A Dcoverage.yaml264 - aten::grid_sampler
830 aten::grid_sampler: 3
H A Dmodel_ops.yaml167 aten::grid_sampler: 7
/aosp_15_r20/external/pytorch/torch/
H A D_meta_registrations.py4631 f"grid_sampler(): expected input and grid to be on same device, but input "
4638 f"grid_sampler(): expected input and grid to have torch.strided layout, but "
4645 f"grid_sampler(): expected grid and input to have same batch size, but got "
4652 f"grid_sampler(): expected grid to have size {input.ndim - 2} in last "
4661 f"grid_sampler(): expected input to have non-empty spatial dimensions, "
4677 f"grid_sampler(): expected 5D input and grid with same number of "
4687 lambda: "grid_sampler(): bicubic interpolation only supports 4D input",
H A Doverrides.py663 torch.grid_sampler: lambda input, grid, interpolation_mode, padding_mode, align_corners: -1,
/aosp_15_r20/external/pytorch/docs/source/
H A Damp.rst404 ``grid_sampler``,
H A Dconf.py1424 "grid_sampler",
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dautocast_test_lists.py191 ("grid_sampler", (torch.randn((2, 3, 33, 22), dtype=torch.float16, device=dev),
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dtrace_rules.py1855 "torch.grid_sampler",

12