[XLA] Add argument and return shardings to while body and cond based on result shardings if present.
[XLA] Add argument and return shardings to while body and cond based on result shardings if present.

This is needed in case propagation runs before this conversion, in which case the best indicator for the argument and return shardings are the shardings of the while results.
