xla

62629e18 - on cpu same config will become replicated, dont't check actual op sharding type