Searched defs:torch_device (Results 1 – 2 of 2) sorted by relevance
90 torch::Device torch_device = torch::lazy::backendDeviceToAtenDevice(device); in ForEachDevice() local
159 py::object torch_device = torch_module.attr("device"); in THPDevice_reduce() local