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