Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/lazy/backend/
Dbackend_device.cpp53 c10::Device backendDeviceToAtenDevice(const BackendDevice& device) { in backendDeviceToAtenDevice() function
Dbackend_device.h72 TORCH_API c10::Device backendDeviceToAtenDevice(const BackendDevice& device);
/external/pytorch/test/cpp/lazy/
Dtest_backend_device.cpp117 auto device = backendDeviceToAtenDevice(BackendDevice()); in TEST()
Dtest_lazy_ops_util.cpp90 torch::Device torch_device = torch::lazy::backendDeviceToAtenDevice(device); in ForEachDevice()
/external/pytorch/torch/csrc/lazy/core/
Dtensor_impl.cpp86 backendDeviceToAtenDevice(tensor.GetDevice())),