Searched defs:rrefId (Results 1 – 11 of 11) sorted by relevance
| /external/pytorch/torch/csrc/distributed/rpc/ |
| D | rref_proto.h | 16 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()
|
| D | rref_context.cpp | 44 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 …]
|
| D | rref_impl.cpp | 43 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()
|
| D | unpickled_python_remote_call.cpp | 9 const at::IValue& rrefId, in UnpickledPythonRemoteCall()
16 const RRefId& UnpickledPythonRemoteCall::rrefId() const { in rrefId() function in torch::distributed::rpc::UnpickledPythonRemoteCall
|
| D | rref_proto.cpp | 43 const RRefId& RRefMessageBase::rrefId() { in rrefId() function in torch::distributed::rpc::RRefMessageBase
|
| D | rref_impl.h | 215 inline const RRefId& rrefId() const { in rrefId() function
|
| D | py_rref.cpp | 41 const RRefId& rrefId = RRefId( in fromPyTuple() local
|
| D | python_functions.cpp | 113 const IValue& rrefId, in sendPythonRemoteCall()
|
| D | request_callback_no_python.cpp | 166 const RRefId& rrefId, in assignOwnerRRef()
|
| /external/pytorch/torch/csrc/distributed/autograd/rpc_messages/ |
| D | rref_backward_req.cpp | 13 const rpc::RRefId& rrefId, in RRefBackwardReq()
57 rpc::RRefId rrefId = rpc::RRefId::fromIValue(tupleElements[0]); in fromMessage() local
|
| /external/pytorch/torch/csrc/jit/serialization/ |
| D | unpickler.cpp | 1050 const auto& rrefId = distributed::rpc::RRefId( in rebuildRRef() local
|