Normalize layouts for Scatter.
Normalize layouts for Scatter.

Unfortunately this requires to temporarily use unsorted update_window_dims
attribute. ScatterSimplifier can turn this into valid HLO again.
