| 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); |
| } |