Home
last modified time | relevance | path

Searched defs:rpcAgent (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/rpc/
H A Dinit.cpp132 auto rpcAgent = in rpc_init() local
655 [](const std::shared_ptr<RpcAgent>& rpcAgent) { in rpc_init()
H A Drpc_agent.cpp275 void RpcAgent::setCurrentRpcAgent(std::shared_ptr<RpcAgent> rpcAgent) { in setCurrentRpcAgent()
H A Dpy_rref.cpp355 auto rpcAgent = rpc::RpcAgent::getCurrentRpcAgent(); in backward() local
/aosp_15_r20/external/pytorch/torch/csrc/distributed/autograd/functions/
H A Drecvrpc_backward.cpp52 auto rpcAgent = rpc::RpcAgent::getCurrentRpcAgent(); in apply() local
/aosp_15_r20/external/pytorch/test/cpp/rpc/
H A De2e_test_base.h163 std::shared_ptr<RpcAgent> rpcAgent; variable