Searched defs:setBoundDeviceId (Results 1 – 2 of 2) sorted by relevance
/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ | ||
H A D | Backend.hpp | 389 void setBoundDeviceId(std::optional<at::Device> device) { in setBoundDeviceId() function in c10d::Backend |
H A D | ProcessGroup.hpp | 710 void setBoundDeviceId(std::optional<at::Device> device) { in setBoundDeviceId() function in c10d::ProcessGroup |