-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathcnn-polar-lag0
68 lines (68 loc) · 2.48 KB
/
cnn-polar-lag0
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
digraph {
graph [size="12,12"]
node [align=left fontname=monospace fontsize=10 height=0.2 ranksep=0.1 shape=box style=filled]
140468548634960 [label="
(1, 2)" fillcolor=darkolivegreen1]
140468820287104 [label=AddmmBackward0]
140468820286288 -> 140468820287104
140473764243392 [label="out.bias
(2)" fillcolor=lightblue]
140473764243392 -> 140468820286288
140468820286288 [label=AccumulateGrad]
140468820286096 -> 140468820287104
140468820286096 [label=ViewBackward0]
140468826906528 -> 140468820286096
140468826906528 [label=ReluBackward0]
140468834153760 -> 140468826906528
140468834153760 [label=MkldnnConvolutionBackward0]
140468833352240 -> 140468834153760
140468833352240 [label=MaxPool2DWithIndicesBackward0]
140468833349744 -> 140468833352240
140468833349744 [label=ReluBackward0]
140468833352048 -> 140468833349744
140468833352048 [label=MkldnnConvolutionBackward0]
140468833351808 -> 140468833352048
140468833351808 [label=MaxPool2DWithIndicesBackward0]
140468833352432 -> 140468833351808
140468833352432 [label=ReluBackward0]
140468833352528 -> 140468833352432
140468833352528 [label=MkldnnConvolutionBackward0]
140468826889376 -> 140468833352528
140468828173312 [label="conv1.0.weight
(16, 1, 5, 5)" fillcolor=lightblue]
140468828173312 -> 140468826889376
140468826889376 [label=AccumulateGrad]
140468826889136 -> 140468833352528
140468828173392 [label="conv1.0.bias
(16)" fillcolor=lightblue]
140468828173392 -> 140468826889136
140468826889136 [label=AccumulateGrad]
140468833352192 -> 140468833352048
140468828173232 [label="conv2.0.weight
(32, 16, 5, 5)" fillcolor=lightblue]
140468828173232 -> 140468833352192
140468833352192 [label=AccumulateGrad]
140468833352144 -> 140468833352048
140468828173472 [label="conv2.0.bias
(32)" fillcolor=lightblue]
140468828173472 -> 140468833352144
140468833352144 [label=AccumulateGrad]
140468833352576 -> 140468834153760
140468822092912 [label="conv3.0.weight
(64, 32, 5, 5)" fillcolor=lightblue]
140468822092912 -> 140468833352576
140468833352576 [label=AccumulateGrad]
140468833351568 -> 140468834153760
140468822090992 [label="conv3.0.bias
(64)" fillcolor=lightblue]
140468822090992 -> 140468833351568
140468833351568 [label=AccumulateGrad]
140468826905904 -> 140468820287104
140468826905904 [label=TBackward0]
140468826903264 -> 140468826905904
140468822092432 [label="out.weight
(2, 3136)" fillcolor=lightblue]
140468822092432 -> 140468826903264
140468826903264 [label=AccumulateGrad]
140468820287104 -> 140468548634960
}