Home
last modified time | relevance | path

Searched refs:privateuse1_backend_name (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/c10/core/
DDeviceType.cpp123 static std::string privateuse1_backend_name; variable
134 name_registered ? privateuse1_backend_name : "privateuseone"; in get_privateuse1_backend()
145 privateuse1_backend_name == backend_name, in register_privateuse1_backend()
147 privateuse1_backend_name); in register_privateuse1_backend()
156 privateuse1_backend_name = backend_name; in register_privateuse1_backend()
/external/pytorch/torch/testing/_internal/
Dcommon_device_type.py752 privateuse1_backend_name = torch._C._get_privateuse1_backend_name()
754 ["privateuse1" if x == privateuse1_backend_name else x for x in except_for]
759 ["privateuse1" if x == privateuse1_backend_name else x for x in only_for]
Dcommon_utils.py1332 privateuse1_backend_name = torch._C._get_privateuse1_backend_name()
1333 privateuse1_backend_module = getattr(torch, privateuse1_backend_name, None)
/external/pytorch/torch/
Doverrides.py1511 privateuse1_backend_name = (
1514 if hasattr(Tensor, privateuse1_backend_name):
1515 ret[getattr(Tensor, privateuse1_backend_name)] = (