Searched defs:getDeviceMap (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/rpc/ | ||
| H A D | rpc_agent.cpp | 313 DeviceMap RpcAgent::getDeviceMap(const WorkerInfo& /* unused */) const { in getDeviceMap() function in torch::distributed::rpc::RpcAgent |
| H A D | tensorpipe_agent.cpp | 1433 DeviceMap TensorPipeAgent::getDeviceMap(const WorkerInfo& dst) const { in getDeviceMap() function in torch::distributed::rpc::TensorPipeAgent |