Home
last modified time | relevance | path

Searched refs:GetDefaultDeviceType (Results 1 – 5 of 5) sorted by relevance

/external/pytorch/torch/csrc/lazy/backend/
Dbackend_device.cpp14 : type_(getBackend()->GetDefaultDeviceType()), in BackendDevice()
49 return BackendDevice(getBackend()->GetDefaultDeviceType(), ordinal); in atenDeviceToBackendDevice()
Dbackend_interface.h103 virtual std::shared_ptr<BackendDeviceType> GetDefaultDeviceType() const = 0;
/external/pytorch/torch/csrc/lazy/ts_backend/
Dts_backend_impl.cpp147 std::shared_ptr<torch::lazy::BackendDeviceType> GetDefaultDeviceType() in GetDefaultDeviceType() function in torch::lazy::TSBackendImpl
257 return torch::lazy::BackendDevice(GetDefaultDeviceType(), device.index()); in GetBackendDevice()
264 return (c10::DeviceType)GetDefaultDeviceType()->type; in EagerFallbackDeviceType()
/external/pytorch/torch/csrc/lazy/python/
Dinit.cpp32 getBackend()->GetDefaultDeviceType(); in GetDeviceOrCurrent()
207 return getBackend()->GetDefaultDeviceType()->toString(); in initLazyBindings()
/external/pytorch/test/cpp/lazy/
Dtest_backend_device.cpp60 auto default_type = getBackend()->GetDefaultDeviceType(); in TEST()