-
Notifications
You must be signed in to change notification settings - Fork 5
/
Copy pathconv_cuda.go
42 lines (33 loc) · 885 Bytes
/
conv_cuda.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
// +build cuda
package golgi
import (
"github.com/chewxy/hm"
G "gorgonia.org/gorgonia"
"gorgonia.org/tensor"
)
// Conv represents a convolution layer
type Conv struct{}
// Model will return the gorgonia.Nodes associated with this convolution layer
func (l *Conv) Model() G.Nodes {
panic("not implemented")
}
// Fwd runs the equation forwards
func (l *Conv) Fwd(x G.Input) G.Result {
panic("not implemented")
}
// Type will return the hm.Type of the convolution layer
func (l *Conv) Type() hm.Type {
panic("not implemented")
}
// Shape will return the tensor.Shape of the convolution layer
func (l *Conv) Shape() tensor.Shape {
panic("not implemented")
}
// Name will return the name of the convolution layer
func (l *Conv) Name() string {
panic("not implemented")
}
// Describe will describe a convolution layer
func (l *Conv) Describe() {
panic("not implemented")
}