Searched refs:PythonRpcHandler (Results 1 – 8 of 8) sorted by relevance
45 return PythonRpcHandler::getInstance().jitCompilationUnit()->get_type(name); in resolveType()69 void PythonRpcHandler::init() { in init()98 PythonRpcHandler::PythonRpcHandler() : initialized_(false) {} in PythonRpcHandler() function in torch::distributed::rpc::PythonRpcHandler100 void PythonRpcHandler::cleanup() { in cleanup()118 PythonRpcHandler& PythonRpcHandler::getInstance() { in getInstance()129 static PythonRpcHandler* handler = new PythonRpcHandler(); in getInstance()134 std::shared_ptr<torch::jit::CompilationUnit> PythonRpcHandler::139 py::object PythonRpcHandler::runPythonUdf(const py::object& pythonUdf) { in runPythonUdf()149 SerializedPyObj PythonRpcHandler::serialize(const py::object& obj) { in serialize()156 py::object PythonRpcHandler::deserialize(const SerializedPyObj& serializedObj) { in deserialize()[all …]
16 class PYBIND11_EXPORT PythonRpcHandler {30 static PythonRpcHandler& getInstance();83 PythonRpcHandler(const PythonRpcHandler&) = delete;84 PythonRpcHandler& operator=(const PythonRpcHandler&) = delete;85 PythonRpcHandler(PythonRpcHandler&&) = delete;86 PythonRpcHandler& operator=(PythonRpcHandler&&) = delete;90 PythonRpcHandler();91 ~PythonRpcHandler() = default;
196 auto& pythonRpcHandler = PythonRpcHandler::getInstance(); in toHere()223 auto& rpcHandler = PythonRpcHandler::getInstance(); in localValue()250 auto& pythonRpcHandler = PythonRpcHandler::getInstance(); in createRRefProxy()274 auto& pythonRpcHandler = PythonRpcHandler::getInstance(); in getRRefType()294 PythonRpcHandler::getInstance().parseTypeFromStr(rrefForkData.typeStr_); in unpickle()
88 auto& pythonRpcHandler = PythonRpcHandler::getInstance(); in serializePyObject()110 auto& pythonRpcHandler = PythonRpcHandler::getInstance(); in runPythonFunction()311 future = PythonRpcHandler::getInstance() in runJitFunction()
43 auto& pythonRpcHandler = PythonRpcHandler::getInstance(); in toPyIValue()250 auto functionSchema = PythonRpcHandler::getInstance() in pyRpcTorchscript()400 auto functionSchema = PythonRpcHandler::getInstance() in pyRemoteTorchscript()
662 auto typePtr = PythonRpcHandler::getInstance().parseTypeFromStr( in rpc_init()665 PythonRpcHandler::getInstance().jitCompilationUnit(), in rpc_init()700 []() { PythonRpcHandler::getInstance().cleanup(); }, in rpc_init()
11 auto& pythonRpcHandler = PythonRpcHandler::getInstance(); in UnpickledPythonCall()
277 friend class PythonRpcHandler; variable