Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/
DGridSampler.cpp511 int64_t ix_nw = static_cast<int64_t>(std::floor(ix)); in _grid_sampler_2d_cpu_quantized() local
613 int64_t ix_nw = static_cast<int64_t>(std::floor(ix)); in _grid_sampler_2d_cpu_fallback() local
673 scalar_t ix_nw = std::floor(ix); in _grid_sampler_2d_cpu_fallback() local
788 int64_t ix_nw = static_cast<int64_t>(std::floor(ix)); in _grid_sampler_2d_cpu_fallback_backward() local
863 scalar_t ix_nw = std::floor(ix); in _grid_sampler_2d_cpu_fallback_backward() local
/external/pytorch/aten/src/ATen/native/cuda/
DGridSampler.cu69 index_t ix_nw = static_cast<index_t>(::floor(ix)); in grid_sampler_2d_kernel() local
122 opmath_t ix_nw = std::floor(ix); in grid_sampler_2d_kernel() local
372 index_t ix_nw = static_cast<index_t>(std::floor(ix)); in grid_sampler_2d_backward_kernel() local
458 scalar_t ix_nw = std::floor(ix); in grid_sampler_2d_backward_kernel() local