Searched defs:device_lazy_init (Results 1 – 2 of 2) sorted by relevance
20 void device_lazy_init(at::DeviceType device_type) { in device_lazy_init() function
129 torch::utils::device_lazy_init(at::kCUDA); in THCPModule_getDevice_wrap() member in THCPModule_getDevice_wrap::torch::utils