Home
last modified time | relevance | path

Searched defs:iy_nw (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/aten/src/ATen/native/
DGridSampler.cpp512 int64_t iy_nw = static_cast<int64_t>(std::floor(iy)); in _grid_sampler_2d_cpu_quantized() local
614 int64_t iy_nw = static_cast<int64_t>(std::floor(iy)); in _grid_sampler_2d_cpu_fallback() local
674 scalar_t iy_nw = std::floor(iy); in _grid_sampler_2d_cpu_fallback() local
789 int64_t iy_nw = static_cast<int64_t>(std::floor(iy)); in _grid_sampler_2d_cpu_fallback_backward() local
864 scalar_t iy_nw = std::floor(iy); in _grid_sampler_2d_cpu_fallback_backward() local
/external/pytorch/aten/src/ATen/native/cuda/
DGridSampler.cu70 index_t iy_nw = static_cast<index_t>(::floor(iy)); in grid_sampler_2d_kernel() local
123 opmath_t iy_nw = std::floor(iy); in grid_sampler_2d_kernel() local
373 index_t iy_nw = static_cast<index_t>(std::floor(iy)); in grid_sampler_2d_backward_kernel() local
459 scalar_t iy_nw = std::floor(iy); in grid_sampler_2d_backward_kernel() local