blob: 27719ce9a2ee47ab02959c376f4d760561302c6a [file] [log] [blame]
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);
}