Searched refs:backendToDeviceType (Results 1 – 4 of 4) sorted by relevance
42 return backendToDeviceType(backend_); in device_type()46 return backendToDeviceType(backend_) == kCUDA; in is_cuda()
220 inline DeviceType backendToDeviceType(Backend b) { in backendToDeviceType() function
460 return at::Device(c10::backendToDeviceType(default_backend)); in get_default_device()
237 …return to(options().device(backendToDeviceType(b)).layout(layout_from_backend(b)), /*non_blocking*… in toBackend()