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

Switch to unified view

a b/gcn/csrc/vision.cpp 

  1
  
#include "GOF.h"





  
  

  2
  





  
  

  3
  
PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {





  
  

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





  
  

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





  
  

  6
  
}