xla

fa059c47 - Revert "Apply sharding to all compatible input tensors (#4872)" (#4911)