Home
last modified time | relevance | path

Searched defs:GetBackendDevice (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/lazy/backend/
H A Dbackend_device.cpp57 std::optional<BackendDevice> GetBackendDevice(at::ITensorListRef tensors) { in GetBackendDevice() function
66 std::optional<BackendDevice> GetBackendDevice(at::TensorList tensors) { in GetBackendDevice() function
70 std::optional<BackendDevice> GetBackendDevice(const at::Tensor& tensor) { in GetBackendDevice() function
77 std::optional<BackendDevice> GetBackendDevice( in GetBackendDevice() function
85 std::optional<BackendDevice> GetBackendDevice() { in GetBackendDevice() function
H A Dbackend_device.h89 std::optional<BackendDevice> GetBackendDevice( in GetBackendDevice() function
/aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/
H A Dts_backend_impl.cpp252 torch::lazy::BackendDevice TSBackendImpl::GetBackendDevice( in GetBackendDevice() function in torch::lazy::TSBackendImpl