blob: 7d0f36fcbca9f9a1ebb076947f9f2a8404b7b4be [file] [log] [blame]
graph(%0 : Float(*, *)
%1 : Float(*, *)) {
%2 : Float(*, *) = prim::FusionGroup_0[device=0](%0, %1)
return (%2);
}
with prim::FusionGroup_0 = graph(%0 : Float(*, *)
%1 : Float(*, *)) {
%2 : int = prim::Constant[value=1]()
%3 : Float(*, *) = aten::add(%0, %1, %2)
%4 : Float(*, *) = aten::mul(%0, %1)
%5 : Float(*, *) = prim::FusedConcat[dim=0](%3, %4)
return (%5);
}