Searched refs:scriptRemoteCall (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/rpc/ |
| H A D | request_callback_impl.cpp | 198 auto& scriptRemoteCall = static_cast<ScriptRemoteCall&>(rpc); in processScriptRemoteCall() local
201 if (scriptRemoteCall.hasOp()) { in processScriptRemoteCall()
203 *scriptRemoteCall.op(), scriptRemoteCall.stackRef(), streams); in processScriptRemoteCall()
206 scriptRemoteCall.qualifiedName(), in processScriptRemoteCall()
207 scriptRemoteCall.stackRef(), in processScriptRemoteCall()
209 scriptRemoteCall.isAsyncExecution()); in processScriptRemoteCall()
213 scriptRemoteCall.retRRefId(), scriptRemoteCall.retForkId(), future); in processScriptRemoteCall()
|
| H A D | torchscript_functions.cpp | 105 auto scriptRemoteCall = std::make_unique<ScriptRemoteCall>( in remoteTorchscript() local
115 std::move(*scriptRemoteCall).toMessage(), in remoteTorchscript()
132 auto scriptRemoteCall = std::make_unique<ScriptRemoteCall>( in remoteTorchscript() local
142 std::move(*scriptRemoteCall).toMessage(), in remoteTorchscript()
|
| H A D | request_callback_no_python.cpp | 206 auto& scriptRemoteCall = static_cast<ScriptRemoteCall&>(rpc); in processScriptRemoteCall() local
209 scriptRemoteCall.hasOp(), "ScriptRemoteCall needs to have an op!"); in processScriptRemoteCall()
211 *scriptRemoteCall.op(), scriptRemoteCall.stackRef(), streams); in processScriptRemoteCall()
214 scriptRemoteCall.retRRefId(), scriptRemoteCall.retForkId(), future); in processScriptRemoteCall()
|
| H A D | python_functions.cpp | 294 auto scriptRemoteCall = std::make_unique<ScriptRemoteCall>( in pyRemoteBuiltin() local
300 std::move(*scriptRemoteCall).toMessage(), in pyRemoteBuiltin()
316 auto scriptRemoteCall = std::make_unique<ScriptRemoteCall>( in pyRemoteBuiltin() local
321 std::move(*scriptRemoteCall).toMessage(), in pyRemoteBuiltin()
|
| /aosp_15_r20/external/pytorch/test/cpp/rpc/ |
| H A D | e2e_test_base.h | 79 ScriptRemoteCall scriptRemoteCall( in createRemoteRRef()
84 std::move(scriptRemoteCall).toMessage(), in createRemoteRRef()
|