blob: c22a28bcf44f616b2fd130734e30ca3dae2461d5 [file] [log] [blame]
graph(%x : Float(*, *)) {
%1 : Float(*, *) = prim::FusionGroup_0(%x)
return (%1);
}
with prim::FusionGroup_0 = graph(%0 : Float(*, *)) {
%1 : Float(*, *), %2 : Float(*, *), %3 : Float(*, *) = prim::ConstantChunk[chunks=3, dim=1](%0)
%4 : Float(*, *) = aten::mul(%1, %2)
%5 : int = prim::Constant[value=1]()
%6 : Float(*, *) = aten::add(%4, %3, %5)
return (%6);
}