Home
last modified time | relevance | path

Searched defs:workerInfo (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/rpc/
H A Dinit.cpp81 auto workerInfo = in rpc_init() local
113 [](const WorkerInfo& workerInfo) { in rpc_init()
120 [](const WorkerInfo& workerInfo) { in rpc_init()
H A Drpc_agent.cpp11 static auto workerInfo = torch::class_<WorkerInfo>("dist_rpc", "WorkerInfo") in RegisterWorkerInfoOnce() local
331 std::ostream& operator<<(std::ostream& os, const WorkerInfo& workerInfo) { in operator <<()
H A Dtensorpipe_agent.cpp1220 const WorkerInfo& workerInfo, in updateGroupMembership()
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dregister_distributed_ops.cpp22 distributed::rpc::RegisterWorkerInfoOnce workerInfo{}; variable