xla

Migrate `torch_xla.device()` to `torch.device('xla')`

#9253

Open