Searched defs:iy_nearest (Results 1 – 2 of 2) sorted by relevance
105 index_t iy_nearest = static_cast<index_t>(std::nearbyint(iy)); in grid_sampler_2d_kernel() local287 index_t iy_nearest = static_cast<index_t>(std::nearbyint(iy)); in grid_sampler_3d_kernel() local435 index_t iy_nearest = static_cast<index_t>(std::nearbyint(iy)); in grid_sampler_2d_backward_kernel() local724 auto iy_nearest = static_cast<index_t>(std::nearbyint(iy)); in grid_sampler_3d_backward_kernel() local
181 int64_t iy_nearest = static_cast<int64_t>(std::nearbyint(iy)); in grid_sampler_3d_cpu_impl() local422 int64_t iy_nearest = static_cast<int64_t>(std::nearbyint(iy)); in grid_sampler_3d_backward_cpu_impl() local653 int64_t iy_nearest = static_cast<int64_t>(std::nearbyint(iy)); in _grid_sampler_2d_cpu_fallback() local848 int64_t iy_nearest = static_cast<int64_t>(std::nearbyint(iy)); in _grid_sampler_2d_cpu_fallback_backward() local