Searched defs:ix_nearest (Results 1 – 2 of 2) sorted by relevance
104 index_t ix_nearest = static_cast<index_t>(std::nearbyint(ix)); in grid_sampler_2d_kernel() local286 index_t ix_nearest = static_cast<index_t>(std::nearbyint(ix)); in grid_sampler_3d_kernel() local434 index_t ix_nearest = static_cast<index_t>(std::nearbyint(ix)); in grid_sampler_2d_backward_kernel() local723 auto ix_nearest = static_cast<index_t>(std::nearbyint(ix)); in grid_sampler_3d_backward_kernel() local
180 int64_t ix_nearest = static_cast<int64_t>(std::nearbyint(ix)); in grid_sampler_3d_cpu_impl() local421 int64_t ix_nearest = static_cast<int64_t>(std::nearbyint(ix)); in grid_sampler_3d_backward_cpu_impl() local652 int64_t ix_nearest = static_cast<int64_t>(std::nearbyint(ix)); in _grid_sampler_2d_cpu_fallback() local847 int64_t ix_nearest = static_cast<int64_t>(std::nearbyint(ix)); in _grid_sampler_2d_cpu_fallback_backward() local