Searched refs:dispatchKeyToDeviceType (Results 1 – 6 of 6) sorted by relevance
744 inline c10::DeviceType dispatchKeyToDeviceType(DispatchKey dispatch_key) { in dispatchKeyToDeviceType() function774 .device(dispatchKeyToDeviceType(dispatch_key)); in dispatchKeyToTensorOptions()
71 .device(dispatchKeyToDeviceType(dispatch_key)) in THPVariable_pynew()
13 return c10::dispatchKeyToDeviceType(k); in sparseTensorSetToDeviceType()
586 dispatchKeyToDeviceType(dispatch_key) == device.value().type(), in check_legacy_ctor_device()588 dispatchKeyToDeviceType(dispatch_key), in check_legacy_ctor_device()
174 device_(c10::dispatchKeyToDeviceType(dispatch_key_), 0), in AOTIPythonKernelHolder()
1109 dispatchKeyToDeviceType(torch::tensors::get_default_dispatch_key()), in THPModule_getDefaultDevice()