UGS-Net / Git / Diff of /gcn/csrc/vision.cpp

Models:

RichardZick/

UGS-Net

Downloads: 1

Diff of /gcn/csrc/vision.cpp [000000] .. [f77492]

Switch to side-by-side view

--- a
+++ b/gcn/csrc/vision.cpp
@@ -0,0 +1,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");
+}