Searched refs:pythonUdf (Results 1 – 5 of 5) sorted by relevance
139 py::object PythonRpcHandler::runPythonUdf(const py::object& pythonUdf) { in runPythonUdf() argument146 return pyRunFunction_(pythonUdf); in runPythonUdf()
185 runPythonFunction(upc.pythonUdf(), streams, upc.isAsyncExecution()); in processPythonCall()221 runPythonFunction(uprc.pythonUdf(), streams, uprc.isAsyncExecution()); in processPythonRemoteCall()
26 const py::object& pythonUdf() const;
30 const py::object& UnpickledPythonCall::pythonUdf() const { in pythonUdf() function in torch::distributed::rpc::UnpickledPythonCall
33 py::object runPythonUdf(const py::object& pythonUdf);