Home
last modified time | relevance | path

Searched refs:THPModule_getDefaultDevice (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/
H A DModule.cpp1106 PyObject* THPModule_getDefaultDevice(PyObject* _unused, PyObject* arg) { in THPModule_getDefaultDevice() function
1519 {"_get_default_device", THPModule_getDefaultDevice, METH_NOARGS, nullptr},
/aosp_15_r20/external/pytorch/torch/_C/
H A D__init__.pyi.in1211 def _get_default_device() -> str: ... # THPModule_getDefaultDevice