blob: 8e349095e054ee9796e209a85ed08982eb72ed9a [file] [log] [blame]
graph(%0 : Float(*, *)
%1 : Float(*, *)
%2 : Float(*, *)
%3 : Float(*, *)
%4 : Float(*, *)
%5 : Float(*)
%6 : Float(*)) {
%7 : Float(*, *) = aten::t(%3)
%8 : int = prim::Constant[value=1]()
%9 : Float(*, *) = aten::addmm(%5, %0, %7, %8, %8)
%10 : Float(*, *) = aten::t(%4)
%11 : Float(*, *) = aten::addmm(%6, %1, %10, %8, %8)
%12 : Dynamic[] = prim::ListConstruct(%9, %11)
%13 : Dynamic[] = aten::broadcast_tensors(%12)
%14 : Dynamic, %15 : Dynamic = prim::ListUnpack(%13)
%16 : Float(*, *) = prim::FusionGroup_0[device=0](%2, %14, %15)
return (%16);
}
with prim::FusionGroup_0 = graph(%0 : Float(*, *)
%1 : Dynamic
%2 : Dynamic) {
%3 : Float(*, *), %4 : Float(*, *), %5 : Float(*, *), %6 : Float(*, *) = prim::ConstantChunk[chunks=4, dim=1](%2)
%7 : Float(*, *), %8 : Float(*, *), %9 : Float(*, *), %10 : Float(*, *) = prim::ConstantChunk[chunks=4, dim=1](%1)
%11 : int = prim::Constant[value=1]()
%12 : Float(*, *) = aten::add(%7, %3, %11)
%13 : int = prim::Constant[value=1]()
%14 : Float(*, *) = aten::add(%8, %4, %13)
%15 : int = prim::Constant[value=1]()
%16 : Float(*, *) = aten::add(%9, %5, %15)
%17 : int = prim::Constant[value=1]()
%18 : Float(*, *) = aten::add(%10, %6, %17)
%19 : Float(*, *) = aten::sigmoid(%12)
%20 : Float(*, *) = aten::sigmoid(%14)
%21 : Float(*, *) = aten::tanh(%16)
%22 : Float(*, *) = aten::sigmoid(%18)
%23 : Float(*, *) = aten::mul(%20, %0)
%24 : Float(*, *) = aten::mul(%19, %21)
%25 : int = prim::Constant[value=1]()
%26 : Float(*, *) = aten::add(%23, %24, %25)
%27 : Float(*, *) = aten::tanh(%26)
%28 : Float(*, *) = aten::mul(%22, %27)
%29 : Float(*, *) = prim::FusedConcat[dim=0](%28, %26)
return (%29);
}