xla

mark_sharding over a replicated tensor is allowed.

#5513

Merged