search

NVIDIA / flownet2-pytorch

Pytorch implementation of FlowNet 2.0: Evolution of Optical Flow Estimation with Deep Networks
Other
3.08k stars 738 forks source link

networks/resample2d_package/resample2d_cuda.cc #262

Open 1359347500cwc opened 2 years ago

1359347500cwc commented 2 years ago

include <ATen/ATen.h>

include <torch/torch.h>

include "resample2d_kernel.cuh"

can't include these files