graph(%1 : Float(4, 4) | |
%2 : Float(4, 4)) { | |
%3 : Float(4, 4) = add[alpha={1}](%1, %2), uses = [%4.i0]; | |
%5 : Float(4!, 2), %6 : Float(4!, 2) = split[split_size=2, dim=1](%3), uses = [[%7.i0], [%7.i1]]; | |
%7 : Float(4, 2) = mul(%5, %6), uses = [%0.i0]; | |
return (%7); | |
} |