xla
Apply sharding to all compatible input tensors
#4919
Merged

Loading