Searched refs:GetBackendDevice (Results 1 – 5 of 5) sorted by relevance
76 TORCH_API std::optional<BackendDevice> GetBackendDevice(78 TORCH_API std::optional<BackendDevice> GetBackendDevice(80 TORCH_API std::optional<BackendDevice> GetBackendDevice(82 TORCH_API std::optional<BackendDevice> GetBackendDevice(86 TORCH_API std::optional<BackendDevice> GetBackendDevice();89 std::optional<BackendDevice> GetBackendDevice( in GetBackendDevice() function92 auto optional_device = GetBackendDevice(tensor); in GetBackendDevice()96 return GetBackendDevice(forward_tensors...); in GetBackendDevice()
57 std::optional<BackendDevice> GetBackendDevice(at::ITensorListRef tensors) { in GetBackendDevice() function66 std::optional<BackendDevice> GetBackendDevice(at::TensorList tensors) { in GetBackendDevice() function67 return GetBackendDevice(at::ITensorListRef(tensors)); in GetBackendDevice()70 std::optional<BackendDevice> GetBackendDevice(const at::Tensor& tensor) { in GetBackendDevice() function77 std::optional<BackendDevice> GetBackendDevice( in GetBackendDevice() function85 std::optional<BackendDevice> GetBackendDevice() { in GetBackendDevice() function
129 virtual BackendDevice GetBackendDevice(c10::Device device) const = 0;
169 torch::lazy::BackendDevice GetBackendDevice(247 devices.emplace_back(GetBackendDevice(c10::Device(c10::kCPU, 0))); in GetBackendDevices()248 devices.emplace_back(GetBackendDevice(c10::Device(c10::kCUDA, 0))); in GetBackendDevices()252 torch::lazy::BackendDevice TSBackendImpl::GetBackendDevice( in GetBackendDevice() function in torch::lazy::TSBackendImpl
127 EXPECT_FALSE(GetBackendDevice(tensor)); in TEST()134 EXPECT_FALSE(GetBackendDevice(tensor1, tensor2)); in TEST()