Home
last modified time | relevance | path

Searched defs:rrefId (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/rpc/
H A Drref_context.cpp44 const RRefId& rrefId) { in finishCreatingOwnerRRef()
155 const RRefId& rrefId = entry.first; in checkRRefLeaks() local
184 const auto rrefId = genGloballyUniqueId(); in createUserRRef() local
191 const RRefId& rrefId, in createUserRRef()
213 const RRefId& rrefId, in delUser()
277 auto rrefId = it.first; in delAllUsersAndUnforkedOwners() local
283 for (auto& rrefId : unforkedOwners) { in delAllUsersAndUnforkedOwners() local
308 auto& rrefId = rrefForkData.rrefId_; in getOrCreateRRef() local
318 const RRefId& rrefId, in getOrCreateOwnerRRef()
384 const RRefId& rrefId, in getOwnerRRef()
[all …]
H A Drref_proto.h16 RRefMessageBase(const RRefId& rrefId, MessageType type) in RRefMessageBase()
30 ForkMessageBase(const RRefId& rrefId, const ForkId& forkId, MessageType type) in ForkMessageBase()
48 ScriptRRefFetchCall(worker_id_t fromWorkerId, const RRefId& rrefId) in ScriptRRefFetchCall()
66 PythonRRefFetchCall(worker_id_t fromWorkerId, const RRefId& rrefId) in PythonRRefFetchCall()
114 RRefUserDelete(const RRefId& rrefId, const ForkId& forkId) in RRefUserDelete()
122 RemoteRet(const RRefId& rrefId, const ForkId& forkId) in RemoteRet()
146 RRefForkRequest(const RRefId& rrefId, const ForkId& forkId) in RRefForkRequest()
H A Drref_impl.cpp43 const RRefId& rrefId, in RRefForkData()
55 RRef::RRef(worker_id_t ownerId, const RRefId& rrefId, TypePtr type) in RRef()
92 const RRefId& rrefId, in UserRRef()
242 const RRefId& rrefId, in OwnerRRef()
254 const RRefId& rrefId, in OwnerRRef()
H A Dunpickled_python_remote_call.cpp9 const at::IValue& rrefId, in UnpickledPythonRemoteCall()
16 const RRefId& UnpickledPythonRemoteCall::rrefId() const { in rrefId() function in torch::distributed::rpc::UnpickledPythonRemoteCall
H A Drref_impl.h215 inline const RRefId& rrefId() const { in rrefId() function
H A Drref_proto.cpp43 const RRefId& RRefMessageBase::rrefId() { in rrefId() function in torch::distributed::rpc::RRefMessageBase
H A Dpy_rref.cpp41 const RRefId& rrefId = RRefId( in fromPyTuple() local
H A Dpython_functions.cpp113 const IValue& rrefId, in sendPythonRemoteCall()
H A Drequest_callback_no_python.cpp166 const RRefId& rrefId, in assignOwnerRRef()
/aosp_15_r20/external/pytorch/torch/csrc/distributed/autograd/rpc_messages/
H A Drref_backward_req.cpp13 const rpc::RRefId& rrefId, in RRefBackwardReq()
57 rpc::RRefId rrefId = rpc::RRefId::fromIValue(tupleElements[0]); in fromMessage() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/
H A Dunpickler.cpp1050 const auto& rrefId = distributed::rpc::RRefId( in rebuildRRef() local