Searched refs:rrefForkData (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/torch/csrc/distributed/rpc/ |
| D | py_rref.cpp | 18 py::tuple toPyTuple(const RRefForkData& rrefForkData) { in toPyTuple() argument
22 rrefForkData.ownerId_, in toPyTuple()
23 rrefForkData.rrefId_.createdOn_, in toPyTuple()
24 rrefForkData.rrefId_.localId_, in toPyTuple()
25 rrefForkData.forkId_.createdOn_, in toPyTuple()
26 rrefForkData.forkId_.localId_, in toPyTuple()
27 rrefForkData.parent_, in toPyTuple()
28 rrefForkData.typeStr_); in toPyTuple()
286 auto rrefForkData = ctx.prepareChildFork(rref_); in pickle() local
287 return toPyTuple(rrefForkData); in pickle()
[all …]
|
| D | rref_context.cpp | 305 const RRefForkData& rrefForkData, in getOrCreateRRef() argument
307 auto& ownerId = rrefForkData.ownerId_; in getOrCreateRRef()
308 auto& rrefId = rrefForkData.rrefId_; in getOrCreateRRef()
309 auto& forkId = rrefForkData.forkId_; in getOrCreateRRef()
437 auto rrefForkData = rref->fork(); in prepareChildFork() local
449 addForkOfOwner(rrefForkData.rrefId_, rrefForkData.forkId_); in prepareChildFork()
468 addPendingChild(rrefForkData.forkId_, rref); in prepareChildFork()
470 return rrefForkData; in prepareChildFork()
|
| /external/pytorch/torch/csrc/jit/serialization/ |
| D | pickler.cpp | 184 auto rrefForkData = ctx.prepareChildFork(rref); in pushRRef() local
186 pushInt(rrefForkData.ownerId_); in pushRRef()
187 pushInt(rrefForkData.rrefId_.createdOn_); in pushRRef()
188 pushInt(rrefForkData.rrefId_.localId_); in pushRRef()
189 pushInt(rrefForkData.forkId_.createdOn_); in pushRRef()
190 pushInt(rrefForkData.forkId_.localId_); in pushRRef()
191 pushInt(rrefForkData.parent_); in pushRRef()
192 pushString(rrefForkData.typeStr_); in pushRRef()
|
| D | unpickler.cpp | 1060 auto rrefForkData = distributed::rpc::RRefForkData( in rebuildRRef() local
1067 rref = ctx.getOrCreateRRef(rrefForkData, type.type_); in rebuildRRef()
1069 rrefForkData.forkId_, rrefForkData.parent_, rref); in rebuildRRef()
|