[XLA:MSA] Avoid sorting colocations.
[XLA:MSA] Avoid sorting colocations.

Maintaining a map from AliasedOffset* to an index in a vector lets us avoid having to sort by just constructing a colocation_vector which is stable in iteration order.
