xla
Enable `padded` option for even tensor sharding
#4156
Merged

Loading