[XLA] Support transposing iota tile assignment with tile dimensions combining multiple reshape dimensions.
[XLA] Support transposing iota tile assignment with tile dimensions combining multiple reshape dimensions.
