/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/Device.h (500B)
#pragma once
#include
#include
#include
// NOLINTNEXTLINE(cppcoreguidelines-pro-type-member-init)
struct TORCH_API THPDevice {
PyObject_HEAD
at::Device device;
};
TORCH_API extern PyTypeObject THPDeviceType;
inline bool THPDevice_Check(PyObject *obj) {
return Py_TYPE(obj) == &THPDeviceType;
}
TORCH_API PyObject * THPDevice_New(const at::Device& device);
TORCH_API void THPDevice_init(PyObject *module);