blob: 7f6b3f1c8b1b9c2dc81a1b115ceed4ad016f81e4 [file] [log] [blame]
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(6, 20) = prim::FusionGroup_0[device=0](%2, %16, %20, %15, %19, %14, %18, %13, %17)
return (%21);
}
with prim::FusionGroup_0 = graph(%14 : Float(3, 20)
%24 : Float(3!, 20)
%25 : Float(3!, 20)
%27 : Float(3!, 20)
%28 : Float(3!, 20)
%30 : Float(3!, 20)
%31 : Float(3!, 20)
%33 : Float(3!, 20)
%34 : Float(3!, 20)) {
%35 : Float(3, 20) = aten::add[alpha={1}](%33, %34)
%32 : Float(3, 20) = aten::add[alpha={1}](%30, %31)
%29 : Float(3, 20) = aten::add[alpha={1}](%27, %28)
%26 : Float(3, 20) = aten::add[alpha={1}](%24, %25)
%23 : Float(3, 20) = aten::sigmoid(%35)
%21 : Float(3, 20) = aten::sigmoid(%32)
%19 : Float(3, 20) = aten::tanh(%29)
%17 : Float(3, 20) = aten::sigmoid(%26)
%15 : Float(3, 20) = aten::mul(%21, %14)
%12 : Float(3, 20) = aten::mul(%23, %19)
%9 : Float(3, 20) = aten::add[alpha={1}](%15, %12)
%6 : Float(3, 20) = aten::tanh(%9)
%5 : Float(3, 20) = aten::mul(%17, %6)
%2 : Float(6, 20) = aten::cat[dim=0](%5, %9)
return (%2);
}