Function PYBIND11_MODULE

op/upfirdn2d.cpp:29–31

}

PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
  m.def("upfirdn2d", &upfirdn2d, "upfirdn2d (CUDA)");
}