xla

a10c4504 - Add more sharding unit tests (#5055)

Add more sharding unit tests (#5055) * Add more sharding unit tests * Remove implicit and explicit sharding tests in favor of multiple operations test