Function PYBIND11_MODULE

alt_cuda_corr/correlation.cpp:51–54

PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
  m.def("forward", &corr_forward, "CORR forward");
  m.def("backward", &corr_backward, "CORR backward");
}