| graph(%0 : Float(*, *) | |
| %1 : Float(*) | |
| %2 : Float(*)) { | |
| %3 : Float(*, *) = prim::FusionGroup_0[device=0](%2, %0, %1) | |
| return (%3); | |
| } | |
| with prim::FusionGroup_0 = graph(%0 : Float(*) | |
| %1 : Float(*, *) | |
| %2 : Float(*)) { | |
| %3 : Float(*, *) = aten::mul(%1, %2) | |
| %4 : int = prim::Constant[value=1]() | |
| %5 : Float(*, *) = aten::add(%3, %0, %4) | |
| return (%5); | |
| } |