xla

70109e15 - [SPMD] Use sharding annotation from XLATensor data as ground truth. (#4390)