[Mosaic GPU] Fix a minor bug in FragmentedArray.transfer_tiled
We don't need to apply rem to the first dimension, since it has the largest
stride. Not the last dimension (where the div does nothing and rem is the only
thing that matters).
This caused OOB memory accesses on some of our B200 tests.
PiperOrigin-RevId: 777540680
