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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
macro - CONV_ATTRS
- KernelShape : vector<int>
- Pads : vector<int> : {0, 0}
- Strides : vector<int> : {1, 1}
- Group : int : 1
- Dilations: vector<int> : {1, 1}
endmacro
macro - POOL_ATTRS
- KernelShape : vector<int>
- Pads : vector<int> : {0, 0}
- Strides : vector<int> : {1, 1}
endmacro
Relu
Conv
- CONV_ATTRS
ConvRelu : Conv
- CONV_ATTRS
ConvTranspose
- CONV_ATTRS
AveragePool
- POOL_ATTRS
AveragePoolRelu : AveragePool
- POOL_ATTRS
MaxPool
- POOL_ATTRS
MaxPoolRelu : MaxPool
- POOL_ATTRS
Sum
SumRelu : Sum
Send
- Destination : string
Receive
- Source : string
BatchNormalization
- Epsilon : float : 1e-5f
- Momentum : float : 0.9f
- Spatial : bool : true
- IsTest : bool : false
Clip
- Min : float
- Max : float
FC
- Axis : int : 1
- AxisW : int : 1
GivenTensorFill
Concat
- Axis : int : -1
- AddAxis : bool : false
Softmax
ChannelShuffle
Add
- Broadcast : int : 0
Reshape
Flatten
CopyToOpenCL
CopyFromOpenCL
NCHW2NHWC
NHWC2NCHW
Declare
Export
|
