Searched defs:deviceMap (Results 1 – 5 of 5) sorted by relevance
21 rpc::DeviceMap deviceMap) in RpcWithAutograd()41 rpc::DeviceMap deviceMap) in RpcWithAutograd()63 c10::Dict<std::string, std::string> deviceMap; in toMessageImpl() local116 rpc::DeviceMap deviceMap; in fromMessage() local173 const rpc::DeviceMap& RpcWithAutograd::deviceMap() { in deviceMap() function in torch::distributed::autograd::RpcWithAutograd
55 const rpc::DeviceMap& deviceMap) { in addRecvRpcBackward()106 const rpc::DeviceMap& deviceMap) { in getMessageWithAutograd()
98 std::map<int, DevPtr> deviceMap; variable
18 rpc::DeviceMap deviceMap) in RecvRpcBackward()
40 const DeviceMap& deviceMap, in getDevicesForTensors()782 const DeviceMap& deviceMap) { in send()