slice of extend #1238
Description
%1076 = "enzymexla.extend"(%1047) <{dimension = 0 : i64, lhs = 1 : i64, rhs = 0 : i64}> : (tensor<4x1520x3056xf64>) -> tensor<5x1520x3056xf64> loc(#loc4184)
%1077 = stablehlo.slice %1076 [0:3, 0:1520, 0:3056] : (tensor<5x1520x3056xf64>) -> tensor<3x1520x3056xf64> loc(#loc4184)
%2494 = stablehlo.slice %1076 [0:4, 0:1520, 0:3056] : (tensor<5x1520x3056xf64>) -> tensor<4x1520x3056xf64> loc(#loc3406)
we can slice 1047 up to [0:3] here before extending