xla

c3d57654 - [r2,3][SPMD] use UNKNOWN as default sharded data sharding type selectively (#6852)