xla

[r2,3][SPMD] use UNKNOWN as default sharded data sharding type selectively

#6852

Merged