forked from pytorch/pytorch
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathTestFuser.test_lstm_traced_cpu.expect
42 lines (42 loc) · 1.73 KB
/
TestFuser.test_lstm_traced_cpu.expect
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
graph(%0 : Float(3, 10)
%1 : Float(3, 20)
%2 : Float(3, 20)
%3 : Float(80, 10)
%4 : Float(80, 20)
%5 : Float(80)
%6 : Float(80)) {
%7 : Float(10!, 80!) = aten::t(%3)
%8 : Float(3!, 80) = aten::expand[size=[3, 80], implicit=0](%5)
%9 : Float(3, 80) = aten::addmm[alpha={1}, beta={1}](%8, %0, %7)
%10 : Float(20!, 80!) = aten::t(%4)
%11 : Float(3!, 80) = aten::expand[size=[3, 80], implicit=0](%6)
%12 : Float(3, 80) = aten::addmm[alpha={1}, beta={1}](%11, %1, %10)
%13 : Float(3!, 20), %14 : Float(3!, 20), %15 : Float(3!, 20), %16 : Float(3!, 20) = aten::chunk[chunks=4, dim=1](%9)
%17 : Float(3!, 20), %18 : Float(3!, 20), %19 : Float(3!, 20), %20 : Float(3!, 20) = aten::chunk[chunks=4, dim=1](%12)
%21 : Float(3, 20), %22 : Float(3, 20) = prim::FusionGroup_0[device=-1](%2, %16, %20, %15, %19, %14, %18, %13, %17)
return (%21, %22);
}
with prim::FusionGroup_0 = graph(%12 : Float(3, 20)
%22 : Float(3!, 20)
%23 : Float(3!, 20)
%25 : Float(3!, 20)
%26 : Float(3!, 20)
%28 : Float(3!, 20)
%29 : Float(3!, 20)
%31 : Float(3!, 20)
%32 : Float(3!, 20)) {
%33 : Float(3, 20) = aten::add[alpha={1}](%31, %32)
%30 : Float(3, 20) = aten::add[alpha={1}](%28, %29)
%27 : Float(3, 20) = aten::add[alpha={1}](%25, %26)
%24 : Float(3, 20) = aten::add[alpha={1}](%22, %23)
%21 : Float(3, 20) = aten::sigmoid(%33)
%19 : Float(3, 20) = aten::sigmoid(%30)
%17 : Float(3, 20) = aten::tanh(%27)
%15 : Float(3, 20) = aten::sigmoid(%24)
%13 : Float(3, 20) = aten::mul(%19, %12)
%10 : Float(3, 20) = aten::mul(%21, %17)
%7 : Float(3, 20) = aten::add[alpha={1}](%13, %10)
%4 : Float(3, 20) = aten::tanh(%7)
%2 : Float(3, 20) = aten::mul(%15, %4)
return (%2, %7);
}