Searched defs:iz_nearest (Results 1 – 2 of 2) sorted by relevance
288 index_t iz_nearest = static_cast<index_t>(std::nearbyint(iz)); in grid_sampler_3d_kernel() local725 auto iz_nearest = static_cast<index_t>(std::nearbyint(iz)); in grid_sampler_3d_backward_kernel() local
182 int64_t iz_nearest = static_cast<int64_t>(std::nearbyint(iz)); in grid_sampler_3d_cpu_impl() local423 int64_t iz_nearest = static_cast<int64_t>(std::nearbyint(iz)); in grid_sampler_3d_backward_cpu_impl() local