graph(%hx : Float(*, *) | |
%cx : Float(*, *)) { | |
%2 : Float(*, *) = prim::FusionGroup_0(%hx, %cx) | |
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); | |
} |