Searched defs:forkId (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/rpc/ |
| H A D | rref_context.cpp | 156 for (const auto& forkId : entry.second) { in checkRRefLeaks() local
185 const auto forkId = genGloballyUniqueId(); in createUserRRef() local
192 const ForkId& forkId, in createUserRRef()
214 const ForkId& forkId) { in delUser()
309 auto& forkId = rrefForkData.forkId_; in getOrCreateRRef() local
474 const ForkId& forkId, in notifyOwnerAndParentOfFork()
537 const ForkId& forkId, in addPendingChild()
551 void RRefContext::delPendingChild(const ForkId& forkId) { in delPendingChild()
584 const ForkId& forkId, in addPendingUser()
611 void RRefContext::delPendingUser(const ForkId& forkId) { in delPendingUser()
[all …]
|
| H A D | rref_proto.h | 30 ForkMessageBase(const RRefId& rrefId, const ForkId& forkId, MessageType type) in ForkMessageBase()
114 RRefUserDelete(const RRefId& rrefId, const ForkId& forkId) in RRefUserDelete()
122 RemoteRet(const RRefId& rrefId, const ForkId& forkId) in RemoteRet()
132 explicit RRefChildAccept(const ForkId& forkId) : forkId_(forkId) {} in RRefChildAccept()
146 RRefForkRequest(const RRefId& rrefId, const ForkId& forkId) in RRefForkRequest()
|
| H A D | rref_impl.cpp | 44 const ForkId& forkId, in RRefForkData()
93 const ForkId& forkId, in UserRRef()
129 const ForkId& UserRRef::forkId() const { in forkId() function in torch::distributed::rpc::UserRRef
|
| H A D | unpickled_python_remote_call.cpp | 10 const at::IValue& forkId, in UnpickledPythonRemoteCall()
20 const ForkId& UnpickledPythonRemoteCall::forkId() const { in forkId() function in torch::distributed::rpc::UnpickledPythonRemoteCall
|
| H A D | rref_proto.cpp | 49 const ForkId& ForkMessageBase::forkId() { in forkId() function in torch::distributed::rpc::ForkMessageBase
151 const ForkId& RRefChildAccept::forkId() const { in forkId() function in torch::distributed::rpc::RRefChildAccept
|
| H A D | py_rref.cpp | 44 const RRefId& forkId = RRefId( in fromPyTuple() local
|
| H A D | python_functions.cpp | 114 const IValue& forkId, in sendPythonRemoteCall()
|
| H A D | request_callback_no_python.cpp | 167 const RRefId& forkId, in assignOwnerRRef()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/ |
| H A D | unpickler.cpp | 1053 const auto& forkId = distributed::rpc::RRefId( in rebuildRRef() local
|