Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A DModule.h4 PyObject* THCPModule_getDevice_wrap(PyObject* self);
H A DModule.cpp127 PyObject* THCPModule_getDevice_wrap(PyObject* self, PyObject* noargs) { in THCPModule_getDevice_wrap() function
1784 {"_cuda_getDevice", THCPModule_getDevice_wrap, METH_NOARGS, nullptr},