Home
last modified time | relevance | path

Searched defs:fromWorkerId (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/rpc/
H A Drref_proto.h48 ScriptRRefFetchCall(worker_id_t fromWorkerId, const RRefId& rrefId) in ScriptRRefFetchCall()
52 inline worker_id_t fromWorkerId() const { in fromWorkerId() function
66 PythonRRefFetchCall(worker_id_t fromWorkerId, const RRefId& rrefId) in PythonRRefFetchCall()
H A Drequest_callback_no_python.cpp319 auto fromWorkerId = rpcWithAutograd.fromWorkerId(); in processForwardAutogradReq() local
/aosp_15_r20/external/pytorch/torch/csrc/distributed/autograd/rpc_messages/
H A Drpc_with_autograd.cpp17 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
/aosp_15_r20/external/pytorch/torch/csrc/distributed/autograd/functions/
H A Drecvrpc_backward.cpp17 rpc::worker_id_t fromWorkerId, in RecvRpcBackward()
/aosp_15_r20/external/pytorch/torch/csrc/distributed/autograd/
H A Dutils.cpp54 rpc::worker_id_t fromWorkerId, in addRecvRpcBackward()