blob: e0d9f25e7948bd6272c13a8d3d9fc5a85363aad7 [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]](%5)
%9 : Float(3, 80) = aten::addmm[beta={1}, alpha={1}](%8, %0, %7)
%10 : Float(20!, 80!) = aten::t(%4)
%11 : Float(3!, 80) = aten::expand[size=[3, 80]](%6)
%12 : Float(3, 80) = aten::addmm[beta={1}, alpha={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(%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);
}