nearest_kernel.cu

#include <ATen/ATen.h>

#define THREADS 1024

at::Tensor nearest_cuda(at::Tensor x, at::Tensor y, at::Tensor batch_x,
                        at::Tensor batch_y) {
  return batch_x;
}