Home
last modified time | relevance | path

Searched refs:rename_privateuse1_backend (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_cpp_extensions_open_device_registration.py114 torch.utils.rename_privateuse1_backend("foo")
147 torch.utils.rename_privateuse1_backend("foo")
153 torch.utils.rename_privateuse1_backend("dev")
544 torch.utils.rename_privateuse1_backend("foo")
H A Dtest_utils.py1066 torch.utils.rename_privateuse1_backend("foo")
1068 torch.utils.rename_privateuse1_backend("dummmy")
/aosp_15_r20/external/pytorch/test/cpp_extensions/open_registration_extension/pytorch_openreg/
H A D__init__.py19 torch.utils.rename_privateuse1_backend("openreg")
/aosp_15_r20/external/pytorch/torch/utils/
H A D__init__.py17 rename_privateuse1_backend,
H A Dbackend_registration.py18 def rename_privateuse1_backend(backend_name: str) -> None: function
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_extension_backend.py107 torch.utils.rename_privateuse1_backend("extension_device")
/aosp_15_r20/external/pytorch/docs/source/
H A Dutils.rst10 rename_privateuse1_backend