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

Download this file

10 lines (6 with data), 285 Bytes 

1
2
3
4
5
6
7
8
9
#pragma once

#include <torch/extension.h>





at::Tensor GOF_forward_cpu(const at::Tensor& weight, 

                           const at::Tensor& gaborFilterBank);



at::Tensor GOF_backward_cpu(const at::Tensor& grad_output,

                            const at::Tensor& gaborFilterBank);