Home
last modified time | relevance | path

Searched refs:THPDevice_New (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/
H A DDevice.cpp20 PyObject* THPDevice_New(const at::Device& device) { in THPDevice_New() function
65 return THPDevice_New(device); in THPDevice_pynew()
84 return THPDevice_New(device); in THPDevice_pynew()
H A DDevice.h19 TORCH_API PyObject* THPDevice_New(const at::Device& device);
H A DEvent.cpp90 return THPDevice_New(self->event.device()); in THPEvent_get_device()
H A DStream.cpp130 return THPDevice_New(c10::Device( in THPStream_get_device()
H A DGenerator.cpp249 return THPDevice_New(self->cdata.device()); in THPGenerator_get_device()
H A DStorage.cpp693 return THPDevice_New(THPStorage_Unpack(self).device()); in THPStorage_device()
/aosp_15_r20/external/pytorch/tools/autograd/templates/
H A Dpython_nn_functions.cpp57 PyTuple_SET_ITEM(tuple.get(), 0, THPDevice_New(*device)); in THPVariable__parse_to()
/aosp_15_r20/external/pytorch/torch/csrc/xpu/
H A DEvent.cpp67 return THPDevice_New(device.value()); in THXPEvent_get_device()
/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A DEvent.cpp121 return THPDevice_New(device.value()); in THCPEvent_get_device()
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dpybind.h189 return handle(THPDevice_New(src));
/aosp_15_r20/external/pytorch/torch/csrc/jit/python/
H A Dpython_ir.cpp920 PyObject* thp_device = THPDevice_New(device.value()); in initPythonIRBindings()
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dpython_function.cpp219 PyObject* device(THPDevice_New(input_info.device)); in defer_to_dynamo()
H A Dpython_variable.cpp1568 return THPDevice_New(THPVariable_Unpack(self).device()); in THPVariable_device()