Searched refs:getDeviceMap (Results 1 – 6 of 6) sorted by relevance
155 agent.getDeviceMap(dst)); in sendMessageWithAutograd()
162 &RpcAgent::getDeviceMap, in rpc_init()634 TensorPipeAgent::getDeviceMap, in rpc_init()
251 virtual DeviceMap getDeviceMap(const WorkerInfo& dst) const;
313 DeviceMap RpcAgent::getDeviceMap(const WorkerInfo& /* unused */) const { in getDeviceMap() function in torch::distributed::rpc::RpcAgent
208 DeviceMap getDeviceMap(const WorkerInfo& dest) const override;
1433 DeviceMap TensorPipeAgent::getDeviceMap(const WorkerInfo& dst) const { in getDeviceMap() function in torch::distributed::rpc::TensorPipeAgent