xla

40127959 - Enable xla_sharding_test.py except the virtual device related ones (#4424)