Searched defs:torch_module (Results 1 – 7 of 7) sorted by relevance
307 auto torch_module = THPObjectPtr(PyImport_ImportModule("torch")); in set_default_storage_type() local393 auto torch_module = THPObjectPtr(PyImport_ImportModule("torch")); in py_bind_tensor_types() local
22 auto torch_module = THPObjectPtr(PyImport_ImportModule("torch")); in initializeLayouts() local
11 auto torch_module = THPObjectPtr(PyImport_ImportModule("torch")); in initializeDtypes() local
18 auto torch_module = THPObjectPtr(PyImport_ImportModule("torch")); in initializeQSchemes() local
29 auto torch_module = THPObjectPtr(PyImport_ImportModule("torch")); in initializeMemoryFormats() local
158 py::object torch_module = py::module::import("torch"); in THPDevice_reduce() local
262 py::object torch_module = py::module::import("torch"); in THPGenerator_reduce() local