[Pallas/Mosaic GPU] Disable inference of sequential axis shapes
They should just be specified in the grid, so we don't need to do this. It's
also incorrect, because it's not guaranteed that each input is sliced in the
same dimension by the sequential axis.
PiperOrigin-RevId: 679114626
