| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/operations/ |
| H A D | GridSampler.mm | 10 #include <ATen/ops/grid_sampler_2d.h>
131 TORCH_WARN_ONCE("MPS: grid_sampler_2d op is supported natively starting from macOS 13.2. ",
134 …return at::grid_sampler_2d(input.to("cpu"), grid.to("cpu"), interpolation_mode, padding_mode, alig…
|
| /aosp_15_r20/external/pytorch/test/expect/ |
| H A D | HasDecompTest.test_aten_core_operators.expect | 265 aten::grid_sampler_2d
266 aten::grid_sampler_2d.out
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | ts_native_functions.yaml | 56 - grid_sampler_2d
|
| H A D | GridSampler.cpp | 23 #include <ATen/ops/grid_sampler_2d.h>
1063 return at::grid_sampler_2d( in grid_sampler()
|
| H A D | native_functions.yaml | 2857 # one of `cudnn_grid_sampler`, `grid_sampler_2d`, or `grid_sampler_3d`, each of
2864 # implementation detail of grid_sampler_2d and is only exposed here for testing
2874 - func: grid_sampler_2d(Tensor input, Tensor grid, int interpolation_mode, int padding_mode, bool a…
2879 autogen: grid_sampler_2d.out
|
| /aosp_15_r20/external/pytorch/functorch/op_analysis/ |
| H A D | public_api | 347 grid_sampler_2d
|
| H A D | annotated_ops | 130 grid_sampler_2d, misc
|
| /aosp_15_r20/external/executorch/kernels/aten/ |
| H A D | functions.yaml | 188 - op: grid_sampler_2d.out
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | decomposition.py | 682 @register_decomposition([aten.grid_sampler_2d])
684 def grid_sampler_2d( function
|
| /aosp_15_r20/external/pytorch/test/edge/ |
| H A D | selected_operators.yaml | 176 aten::grid_sampler_2d.out:
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_ops.py | 1776 ), # NYI: forward AD for grid_sampler_2d
1777 xfail("grid_sampler_2d", ""), # NYI: forward AD for grid_sampler_2d
1996 "grid_sampler_2d"
|
| /aosp_15_r20/external/pytorch/docs/source/ |
| H A D | amp.rst | 405 ``grid_sampler_2d``,
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_decomp.py | 262 (torch.float32, torch.ops.aten.grid_sampler_2d.default): (7e-6, 3e-5),
484 skip("grid_sampler_2d"), # slow: fails with --timeout=360 secs
|
| H A D | test_mps.py | 154 'grid_sampler_2d': [torch.float32],
167 'grid_sampler_2d': [torch.float32],
634 'grid_sampler_2d': [torch.float32], # Unsupported Border padding mode
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | autocast_mode.cpp | 367 KERNEL_CPU(grid_sampler_2d, fp32) in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesModules.cpp | 355 VMAP_SUPPORT(grid_sampler_2d, GRID_SAMPLE_BATCH_RULE(grid_sampler)); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/torchgen/static_runtime/ |
| H A D | generator.py | 95 "grid_sampler_2d",
|
| /aosp_15_r20/external/pytorch/test/distributed/_tensor/ |
| H A D | test_dtensor_ops.py | 178 xfail("grid_sampler_2d"),
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_torchinductor_opinfo.py | 470 ("grid_sampler_2d", "cuda"): {f16},
|
| H A D | test_torchinductor.py | 5837 aten.grid_sampler_2d(a, b, 0, 0, True),
5838 aten.grid_sampler_2d(a, b, 0, 1, False),
|
| /aosp_15_r20/external/pytorch/torch/_decomp/ |
| H A D | decompositions.py | 4083 …# This method is a copy of grid_sampler_2d implementation and introduced with additional arg _expa…
4241 @register_decomposition(aten.grid_sampler_2d)
4244 def grid_sampler_2d( function
|
| /aosp_15_r20/external/pytorch/benchmarks/dynamo/microbenchmarks/operator_inp_logs/torchbench_train/ |
| H A D | Super_SloMo_training.txt | 118 Operator: aten.grid_sampler_2d.default
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | trace_rules.py | 1853 "torch.grid_sampler_2d",
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | overrides.py | 664 … torch.grid_sampler_2d: lambda input, grid, interpolation_mode, padding_mode, align_corners: -1,
|
| /aosp_15_r20/external/pytorch/tools/autograd/ |
| H A D | derivatives.yaml | 760 - name: grid_sampler_2d(Tensor input, Tensor grid, int interpolation_mode, int padding_mode, bool a…
|