Home
last modified time | relevance | path

Searched defs:interpolation_mode (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DGridSampler.cpp20 int64_t interpolation_mode, int64_t padding_mode, in grid_sampler_2d_cuda()
32 int64_t interpolation_mode, int64_t padding_mode, in grid_sampler_3d_cuda()
46 const Tensor& grid, int64_t interpolation_mode, int64_t padding_mode, in grid_sampler_2d_backward_cuda()
65 const Tensor& grid, int64_t interpolation_mode, int64_t padding_mode, in grid_sampler_3d_backward_cuda()
H A DGridSampler.cu31 const GridSamplerInterpolation interpolation_mode, in grid_sampler_2d_kernel()
161 const GridSamplerInterpolation interpolation_mode, in grid_sampler_3d_kernel()
319 const GridSamplerInterpolation interpolation_mode, in grid_sampler_2d_backward_kernel()
525 const GridSamplerInterpolation interpolation_mode, in grid_sampler_3d_backward_kernel()
752 int64_t interpolation_mode, int64_t padding_mode, bool align_corners) { in launch_grid_sampler_2d_forward_kernel()
796 int64_t interpolation_mode, int64_t padding_mode, bool align_corners) { in launch_grid_sampler_3d_forward_kernel()
842 const TensorBase &grid, int64_t interpolation_mode, int64_t padding_mode, in launch_grid_sampler_2d_backward_kernel()
905 const TensorBase& grid, int64_t interpolation_mode, int64_t padding_mode, in launch_grid_sampler_3d_backward_kernel()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DGridSampler.cpp42 GridSamplerInterpolation interpolation_mode, in grid_sampler_3d_cpu_impl()
207 GridSamplerInterpolation interpolation_mode, in grid_sampler_3d_backward_cpu_impl()
457 auto interpolation_mode = in _grid_sampler_2d_cpu_quantized() local
566 auto interpolation_mode = static_cast<GridSamplerInterpolation>(interpolation_mode_); in _grid_sampler_2d_cpu_fallback() local
722 const auto interpolation_mode = static_cast<GridSamplerInterpolation>(interpolation_mode_); in _grid_sampler_2d_cpu_fallback_backward() local
920 int64_t interpolation_mode, int64_t padding_mode, in grid_sampler_2d_cpu()
964 int64_t interpolation_mode, int64_t padding_mode, in grid_sampler_3d_cpu()
980 int64_t interpolation_mode, int64_t padding_mode, bool align_corners, in grid_sampler_2d_backward_cpu()
1029 int64_t interpolation_mode, int64_t padding_mode, bool align_corners, in grid_sampler_3d_backward_cpu()
1049 int64_t interpolation_mode, in grid_sampler()
H A DGridSamplerUtils.h76 int64_t interpolation_mode in check_grid_sampler_3d()
/aosp_15_r20/external/mesa3d/src/microsoft/compiler/
H A Ddxil_signature.h78 uint8_t interpolation_mode; // DXIL::InterpolationMode or D3D10_SB_INTERPOLATION_MODE member
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DGridSamplerKernel.cpp1146 int64_t interpolation_mode, int64_t padding_mode, bool align_corners) { in grid_sampler_2d_cpu_kernel_impl()
1215 int64_t interpolation_mode, in grid_sampler_2d_backward_cpu_kernel_impl()
/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Dshape_inference.cpp864 int64_t interpolation_mode, in compute_shape_grid_sampler_2d()
879 int64_t interpolation_mode, in compute_shape_grid_sampler_2d_backward()
/aosp_15_r20/external/pytorch/torch/
H A D_meta_registrations.py4696 interpolation_mode, argument
4715 interpolation_mode, argument
4735 interpolation_mode, argument