1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
graph(%0 : Float(4, 4)
%1 : Float(4, 4)) {
%2 : Float(4!, 2), %3 : Float(4!, 2) = aten::chunk[chunks=2, dim=1](%0)
%4 : Float(4!, 2), %5 : Float(4!, 2) = aten::chunk[chunks=2, dim=1](%1)
%6 : Float(4, 2) = prim::FusionGroup_0[device=0](%2, %4, %3, %5)
return (%6);
}
with prim::FusionGroup_0 = graph(%3 : Float(4!, 2)
%4 : Float(4!, 2)
%6 : Float(4!, 2)
%7 : Float(4!, 2)) {
%8 : Float(4, 2) = aten::add[alpha={1}](%6, %7)
%5 : Float(4, 2) = aten::add[alpha={1}](%3, %4)
%2 : Float(4, 2) = aten::mul(%5, %8)
return (%2);
}
|
