xla

c1db8759 - [SPMD] use UNKNOWN as default sharded data sharding type only for auto-sharding (#6846)