xla

[SPMD] Test sharding_spec with clear_sharding, deepcopy

#4144

Merged