Function GetNearestInputIndex

oneflow/user/kernels/upsample_kernel.h:29–34

}

OF_DEVICE_FUNC static int64_t GetNearestInputIndex(const int64_t out_dim_idx, const double scale,
                                                   const int64_t in_dim_size) {
  int64_t index = static_cast<int64_t>(floorf(out_dim_idx * scale));
  index = index > in_dim_size - 1 ? in_dim_size - 1 : index;
  return index;
}

OF_DEVICE_FUNC double GetAreaPixelScale(const int64_t input_size, const int64_t output_size,
                                        bool align_corners, const double scale) {