File: ops.def

package info (click to toggle)
pytorch 1.7.1-7
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 80,340 kB
  • sloc: cpp: 670,830; python: 343,991; ansic: 67,845; asm: 5,503; sh: 2,924; java: 2,888; xml: 266; makefile: 244; ruby: 148; yacc: 144; objc: 51; lex: 44
file content (80 lines) | stat: -rw-r--r-- 969 bytes parent folder | download | duplicates (2)
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