Home
last modified time | relevance | path

Searched defs:jitFuture (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/rpc/
H A Drref_context.cpp15 const JitFuture& jitFuture, in confirmPendingUser()
43 const JitFuture& jitFuture, in finishCreatingOwnerRRef()
107 void RRefContext::handleException(const JitFuture& jitFuture) { in handleException()
115 void RRefContext::handleExceptionSilent(const JitFuture& jitFuture) { in handleExceptionSilent()
222 auto jitFuture = agent_->sendWithRetries( in delUser() local
512 auto jitFuture = agent_->sendWithRetries( in notifyOwnerAndParentOfFork() local
520 auto jitFuture = agent_->sendWithRetries( in notifyOwnerAndParentOfFork() local
710 auto jitFuture = agent_->sendWithRetries( in finishForkRequest() local
H A Drpc_agent.cpp104 auto jitFuture = send(to, std::move(message)); in sendWithRetries() local
154 c10::intrusive_ptr<JitFuture> jitFuture; in retryExpiredRpcs() local
186 auto jitFuture = it.first; in retryExpiredRpcs() local
211 JitFuture& jitFuture, in rpcRetryCallback()
H A Dtorchscript_functions.cpp42 auto jitFuture = autograd::sendMessageWithAutograd( in rpcTorchscript() local
112 auto jitFuture = torch::distributed::autograd::sendMessageWithAutograd( in remoteTorchscript() local
139 auto jitFuture = torch::distributed::autograd::sendMessageWithAutograd( in remoteTorchscript() local
H A Dpython_functions.cpp297 auto jitFuture = sendMessageWithAutograd( in pyRemoteBuiltin() local
318 auto jitFuture = sendMessageWithAutograd( in pyRemoteBuiltin() local
349 auto jitFuture = sendPythonRemoteCall( in pyRemotePythonUdf() local
369 auto jitFuture = sendPythonRemoteCall( in pyRemotePythonUdf() local
H A Drref_impl.cpp71 void RRef::handleError(RPCErrorType errorType, const JitFuture& jitFuture) { in handleError()
178 auto jitFuture = autograd::sendMessageWithAutograd( in toHere() local
H A Drequest_callback_impl.cpp174 [](JitFuture& jitFuture) { in processScriptCall()
H A Dtensorpipe_agent.h307 c10::intrusive_ptr<JitFuture> jitFuture; member
H A Dutils.cpp60 RPCErrorType getRPCErrorType(const JitFuture& jitFuture) { in getRPCErrorType()
/aosp_15_r20/external/pytorch/torch/csrc/distributed/autograd/functions/
H A Drecvrpc_backward.cpp53 auto jitFuture = rpcAgent->send( in apply() local
/aosp_15_r20/external/pytorch/test/cpp/rpc/
H A De2e_test_base.h81 auto jitFuture = autograd::sendMessageWithAutograd( in createRemoteRRef() local
/aosp_15_r20/external/pytorch/torch/csrc/distributed/rpc/testing/
H A Dfaulty_tensorpipe_agent.cpp88 auto jitFuture = c10::make_intrusive<JitFuture>(at::AnyClassType::get()); in send() local
/aosp_15_r20/external/pytorch/torch/csrc/distributed/autograd/context/
H A Dcontext.cpp136 const c10::intrusive_ptr<rpc::JitFuture>& jitFuture) { in addOutstandingRpc()