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

Download this file

9 lines (6 with data), 288 Bytes 

1
2
3
4
5
6
7
8
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);