test/expect/TestScript.test_milstm_fusion_cuda-backward.expect

graph(%0 : Float(*, *)
      %1 : Float(*, *)
      %2 : Dynamic
      %3 : Dynamic
      %4 : Dynamic
      %5 : Dynamic
      %6 : Dynamic
      %7 : Dynamic
      %8 : Dynamic
      %9 : Dynamic
      %10 : Dynamic
      %11 : Dynamic
      %12 : Float(*, *)
      %13 : Float(*, *)
      %14 : Float(*)
      %15 : Float(*)
      %16 : Float(*)
      %17 : Float(*, *)
      %18 : Float(*, *)
      %Wx : Float(*, *)
      %20 : Float(*, *)
      %Uz : Float(*, *)
      %22 : Float(*, *)
      %ingate : Float(*, *)
      %forgetgate : Float(*, *)
      %cellgate : Float(*, *)
      %outgate : Float(*, *)
      %27 : Float(*, *)) {
  %28 : Float(*, *), %29 : Float(*, *) = prim::FusionGroup_0[device=0](%ingate, %forgetgate, %cellgate, %outgate, %17, %1, %27, %0)
  %30 : Float(*, *), %31 : Float(*, *), %32 : Float(*, *), %33 : Float(*, *), %34 : Float(*, *), %35 : Float(*, *) = prim::FusionGroup_1[device=0](%14, %15, %Wx, %28, %Uz, %22, %16)
  %36 : Float(*, *) = aten::t(%20)
  %37 : Float(*, *) = aten::mm(%32, %36)
  %38 : Float(*, *) = aten::t(%13)
  %39 : Float(*, *) = aten::mm(%38, %32)
  %40 : Float(*, *) = aten::t(%39)
  %41 : Float(*, *) = aten::t(%18)
  %42 : Float(*, *) = aten::mm(%30, %41)
  %43 : Float(*, *) = aten::t(%12)
  %44 : Float(*, *) = aten::mm(%43, %30)
  %45 : Float(*, *) = aten::t(%44)
  return (%45, %42, %40, %37, %31, %33, %34, %35, %29);
}
with prim::FusionGroup_0 = graph(%0 : Float(*, *)
      %1 : Float(*, *)
      %2 : Float(*, *)
      %3 : Float(*, *)
      %4 : Float(*, *)
      %5 : Float(*, *)
      %6 : Float(*, *)
      %7 : Float(*, *)) {
  %8 : Float(*, *) = aten::mul(%5, %3)
  %9 : Float(*, *) = aten::mul(%6, %6)
  %10 : Float(*, *) = aten::neg(%9)
  %11 : int = prim::Constant[value=1]()
  %12 : int = prim::Constant[value=1]()
  %13 : Float(*, *) = aten::add(%10, %12, %12)
  %14 : Float(*, *) = aten::mul(%8, %13)
  %15 : int = prim::Constant[value=1]()
  %16 : int = prim::Constant[value=1]()
  %17 : Float(*, *) = aten::add(%7, %14, %16)
  %18 : Float(*, *) = aten::mul(%17, %1)
  %19 : Float(*, *) = aten::mul(%5, %6)
  %20 : int = prim::Constant[value=1]()
  %21 : Float(*, *) = aten::mul(%17, %20)
  %22 : Float(*, *) = aten::mul(%21, %2)
  %23 : Float(*, *) = aten::mul(%21, %0)
  %24 : Float(*, *) = aten::mul(%17, %4)
  %25 : Float(*, *) = aten::neg(%3)
  %26 : int = prim::Constant[value=1]()
  %27 : Float(*, *) = aten::add(%25, %26, %26)
  %28 : Float(*, *) = aten::mul(%19, %3)
  %29 : Float(*, *) = aten::mul(%28, %27)
  %30 : Float(*, *) = aten::mul(%2, %2)
  %31 : Float(*, *) = aten::neg(%30)
  %32 : int = prim::Constant[value=1]()
  %33 : Float(*, *) = aten::add(%31, %32, %32)
  %34 : Float(*, *) = aten::mul(%23, %33)
  %35 : Float(*, *) = aten::neg(%1)
  %36 : int = prim::Constant[value=1]()
  %37 : Float(*, *) = aten::add(%35, %36, %36)
  %38 : Float(*, *) = aten::mul(%24, %1)
  %39 : Float(*, *) = aten::mul(%38, %37)
  %40 : Float(*, *) = aten::neg(%0)
  %41 : int = prim::Constant[value=1]()
  %42 : Float(*, *) = aten::add(%40, %41, %41)
  %43 : Float(*, *) = aten::mul(%22, %0)
  %44 : Float(*, *) = aten::mul(%43, %42)
  %45 : Float(*, *) = prim::FusedConcat[dim=1](%44, %39, %34, %29)
  return (%45, %18);
}
with prim::FusionGroup_1 = graph(%0 : Float(*)
      %1 : Float(*)
      %2 : Float(*, *)
      %3 : Float(*, *)
      %4 : Float(*, *)
      %5 : Float(*, *)
      %6 : Float(*)) {
  %7 : int = prim::Constant[value=1]()
  %8 : int = prim::Constant[value=1]()
  %9 : int = prim::Constant[value=1]()
  %10 : int = prim::Constant[value=1]()
  %11 : Float(*, *) = aten::mul(%3, %10)
  %12 : Float(*, *) = aten::mul(%11, %4)
  %13 : Float(*, *) = aten::mul(%11, %2)
  %14 : Float(*, *) = aten::mul(%11, %6)
  %15 : Float(*, *) = aten::mul(%3, %5)
  %16 : int = prim::Constant[value=1]()
  %17 : int = prim::Constant[value=1]()
  %18 : Float(*, *) = aten::add(%14, %15, %17)
  %19 : Float(*, *) = aten::mul(%3, %4)
  %20 : Float(*, *) = aten::mul(%19, %2)
  %21 : Float(*, *) = aten::mul(%11, %1)
  %22 : Float(*, *) = aten::mul(%19, %0)
  %23 : int = prim::Constant[value=1]()
  %24 : Float(*, *) = aten::add(%21, %22, %23)
  return (%24, %20, %18, %13, %12, %11);
}