Searched defs:fromWorkerId (Results 1 – 5 of 5) sorted by relevance
48 ScriptRRefFetchCall(worker_id_t fromWorkerId, const RRefId& rrefId) in ScriptRRefFetchCall()52 inline worker_id_t fromWorkerId() const { in fromWorkerId() function66 PythonRRefFetchCall(worker_id_t fromWorkerId, const RRefId& rrefId) in PythonRRefFetchCall()
319 auto fromWorkerId = rpcWithAutograd.fromWorkerId(); in processForwardAutogradReq() local
17 worker_id_t fromWorkerId, in RpcWithAutograd()35 worker_id_t fromWorkerId, in RpcWithAutograd()169 rpc::worker_id_t RpcWithAutograd::fromWorkerId() const { in fromWorkerId() function in torch::distributed::autograd::RpcWithAutograd
17 rpc::worker_id_t fromWorkerId, in RecvRpcBackward()
54 rpc::worker_id_t fromWorkerId, in addRecvRpcBackward()