Incorrect IR generated

See the IR: https://pastebin.com/wtd7brkX

```julia
/tmp/jl_CDw1qNavoJ.mlir:671:34: error: use of undeclared SSA value name
    %212 = "stablehlo.transpose"(%292) <{permutation = array<i64: 0>}> : (tensor<2xui64>) -> tensor<2xui64>
```
