Log In Sign Up
Models:
Richard Zick
RichardZick/
UGS-Net
0
Downloads: 1
[cf6a9e]: / gcn / csrc / vision.cpp

Download this file

7 lines (5 with data), 172 Bytes 

1
2
3
4
5
6
#include "GOF.h"



PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

  m.def("gof_forward", &GOF_forward, "GOF forward");

  m.def("gof_backward", &GOF_backward, "GOF backward");

}