Deprecate API

Author: ghpvnist

torch_xla/core/xla_model.py

@@ -118,7 +118,7 @@ def master_print(*args: Any,
     print(*args, file=fd, flush=flush)
 
 
-@deprecated("Use torch_xla.device instead")
+@deprecated("Use torch.device('xla') instead")
 def xla_device(n: Optional[int] = None,
                devkind: Optional[str] = None) -> torch.device:
   """Returns a given instance of an XLA device.

torch_xla/torch_xla.py

@@ -4,6 +4,7 @@
 import functools
 import uuid
 from typing import Any, Callable, List, Optional, Tuple
+from typing_extensions import deprecated
 import weakref
 
 import torch
@@ -16,6 +17,7 @@
 import torch_xla.utils.utils as xu
 
 
+@deprecated("Use torch.device('xla') instead")
 def device(index: int = None) -> torch.device:
   """Returns a given instance of an XLA device.