xla

[SPMD] Use sharding annotation from XLATensor data as ground truth.

#4390

Merged