-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathLSTMFullGenerator
54 lines (54 loc) · 1.83 KB
/
LSTMFullGenerator
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
digraph {
graph [size="12,12"]
node [align=left fontname=monospace fontsize=10 height=0.2 ranksep=0.1 shape=box style=filled]
2180725950656 [label="
(1, 576, 15)" fillcolor=darkolivegreen1]
2180745598080 [label=ViewBackward0]
2180745595584 -> 2180745598080
2180745595584 [label=TanhBackward0]
2180745596640 -> 2180745595584
2180745596640 [label=AddmmBackward0]
2180745599712 -> 2180745596640
2180725960416 [label="linear.0.bias
(15)" fillcolor=lightblue]
2180725960416 -> 2180745599712
2180745599712 [label=AccumulateGrad]
2180745599664 -> 2180745596640
2180745599664 [label=ViewBackward0]
2180745599856 -> 2180745599664
2180745599856 [label=TransposeBackward0]
2180745600000 -> 2180745599856
2180745600000 [label=MkldnnRnnLayerBackward0]
2180745600096 -> 2180745600000
2180720154480 [label="lstm.weight_ih_l0
(1024, 15)" fillcolor=lightblue]
2180720154480 -> 2180745600096
2180745600096 [label=AccumulateGrad]
2180745600048 -> 2180745600000
2180745651584 [label="lstm.weight_hh_l0
(1024, 256)" fillcolor=lightblue]
2180745651584 -> 2180745600048
2180745600048 [label=AccumulateGrad]
2180745599904 -> 2180745600000
2180745650544 [label="lstm.bias_ih_l0
(1024)" fillcolor=lightblue]
2180745650544 -> 2180745599904
2180745599904 [label=AccumulateGrad]
2180745600144 -> 2180745600000
2180745651664 [label="lstm.bias_hh_l0
(1024)" fillcolor=lightblue]
2180745651664 -> 2180745600144
2180745600144 [label=AccumulateGrad]
2180745598656 -> 2180745596640
2180745598656 [label=TBackward0]
2180745600192 -> 2180745598656
2180725956576 [label="linear.0.weight
(15, 256)" fillcolor=lightblue]
2180725956576 -> 2180745600192
2180745600192 [label=AccumulateGrad]
2180745598080 -> 2180725950656
2180725960096 [label="
(576, 15)" fillcolor=darkolivegreen3]
2180745595584 -> 2180725960096
2180725960096 -> 2180725950656 [style=dotted]
}