--- a
+++ b/gcn/csrc/cuda/vision.h
@@ -0,0 +1,9 @@
+#pragma once
+#include <torch/extension.h>
+
+
+at::Tensor GOF_forward_cuda(const at::Tensor& weight, 
+                            const at::Tensor& gaborFilterBank);
+
+at::Tensor GOF_backward_cuda(const at::Tensor& grad_output,
+                             const at::Tensor& gaborFilterBank);
\ No newline at end of file