Searched refs:getAccelerator (Results 1 – 9 of 9) sorted by relevance
23 TORCH_API std::optional<c10::DeviceType> getAccelerator(bool checked = false);
5 std::optional<c10::DeviceType> getAccelerator(bool checked) { in getAccelerator() function
65 : at::getAccelerator(true).value();
2105 auto device_type = at::getAccelerator(); in initModule()2117 auto device_type = at::getAccelerator(); in initModule()2126 auto device_type = at::getAccelerator(); in initModule()2137 auto device_type = at::getAccelerator(); in initModule()2149 auto device_type = at::getAccelerator(); in initModule()2162 at::getAccelerator(check.value_or(false)) in initModule()
40 device = at::Device(at::getAccelerator(false).value_or(at::kCPU)); in THPEvent_pynew()
42 auto default_accelerator = at::getAccelerator(false); in THPStream_pynew()
243 auto opt_device_type = at::getAccelerator(); in stream()
1545 const auto accelerator = at::getAccelerator(true).value(); in stash_current_streams()
810 at::getAccelerator(true).value(), in deviceFromLong()