blob: da7a7bbe2b4680e755ad52b19c04663a4f67e722 [file] [log] [blame]
graph(%x : Float(*, *),
%scale : Float(*),
%shift : Float(*)):
%3 : Float(*, *) = prim::FusionGroup_0(%shift, %x, %scale)
return (%3)
with prim::FusionGroup_0 = graph(%0 : Float(*),
%1 : Float(*, *),
%2 : Float(*)):
%3 : int = prim::Constant[value=1]()
%4 : Float(*, *) = aten::mul(%1, %2)
%5 : Float(*, *) = aten::add(%4, %0, %3)
return (%5)