%685 = stablehlo.slice %arg20 [8:12, 8:6136, 8:12280] : (tensor<20x6144x12288xf32>) -> tensor<4x6128x12272xf32> loc(#loc4553)
%1235 = "enzymexla.rotate"(%685) <{amount = 2 : si32, dimension = 2 : si32}> : (tensor<4x6128x12272xf32>) -> tensor<4x6128x12272xf32> loc(#loc3774)
%1635 = stablehlo.slice %arg20 [8:12, 8:6136, 10:11] : (tensor<20x6144x12288xf32>) -> tensor<4x6128x1xf32> loc(#loc2576)
%1636 = stablehlo.concatenate %1235, %1635, dim = 2 : (tensor<4x6128x12272xf32>, tensor<4x6128x1xf32>) -> tensor<4x6128x12273xf32> loc(#loc2576)
%22 = stablehlo.slice %15 [8:12, 8:6136, 12269:12270] : (tensor<20x6144x12272xf32>) -> tensor<4x6128x1xf32> loc(#loc3742)
%23 = stablehlo.slice %15 [8:12, 8:6136, 0:12272] : (tensor<20x6144x12272xf32>) -> tensor<4x6128x12272xf32> loc(#loc3742)
%648 = "enzymexla.rotate"(%23) <{amount = 12270 : si32, dimension = 2 : si32}> : (tensor<4x6128x12272xf32>) -> tensor<4x6128x12272xf32> loc(#loc3742)
%1847 = stablehlo.concatenate %22, %648, dim = 2 : (tensor<4x6128x1xf32>, tensor<4x6128x12272xf32>) -> tensor<4x6128x12273xf32> loc(#loc3603)
