a b/gcn/csrc/cuda/vision.h 

  1
  
#pragma once





  
  

  2
  
#include <torch/extension.h>





  
  

  3
  





  
  

  4
  





  
  

  5
  
at::Tensor GOF_forward_cuda(const at::Tensor& weight, 





  
  

  6
  
                            const at::Tensor& gaborFilterBank);





  
  

  7
  





  
  

  8
  
at::Tensor GOF_backward_cuda(const at::Tensor& grad_output,





  
  

  9
  
                             const at::Tensor& gaborFilterBank);