Searched refs:PythonFutureWrapper (Results 1 – 8 of 8) sorted by relevance
2031 py::class_<PythonFutureWrapper, std::shared_ptr<PythonFutureWrapper>>( in initJITBindings()2034 return std::make_shared<PythonFutureWrapper>( in initJITBindings()2041 &PythonFutureWrapper::done) in initJITBindings()2044 &PythonFutureWrapper::value, in initJITBindings()2048 &PythonFutureWrapper::wait, in initJITBindings()2052 &PythonFutureWrapper::then, in initJITBindings()2056 &PythonFutureWrapper::add_done_callback, in initJITBindings()2061 &PythonFutureWrapper::markCompleted) in initJITBindings()2065 [](PythonFutureWrapper& self, py::function unwrapFunc) { in initJITBindings()2080 [](const PythonFutureWrapper& /* unused */) { in initJITBindings() argument[all …]
104 struct VISIBILITY_HIDDEN PythonFutureWrapper struct105 : std::enable_shared_from_this<PythonFutureWrapper> {108 explicit PythonFutureWrapper( argument113 explicit PythonFutureWrapper(const PythonFutureWrapper&) = delete; argument114 PythonFutureWrapper& operator=(const PythonFutureWrapper&) = delete; argument149 std::shared_ptr<PythonFutureWrapper> then(py::function cb) { in then() argument156 return std::make_shared<jit::PythonFutureWrapper>(fut->then( in then()235 std::shared_ptr<PythonFutureWrapper> getPtr() { in getPtr() argument
533 return obj.cast<std::shared_ptr<PythonFutureWrapper>>()->fut; in toIValue()720 return py::cast(std::make_shared<PythonFutureWrapper>(ivalue.toFuture())); in toPyObject()
457 return std::make_shared<jit::PythonFutureWrapper>( in rpc_init()469 return std::make_shared<jit::PythonFutureWrapper>( in rpc_init()480 const std::shared_ptr<jit::PythonFutureWrapper>& in rpc_init()710 return std::make_shared<jit::PythonFutureWrapper>( in rpc_init()722 return std::make_shared<jit::PythonFutureWrapper>(pyRpcPythonUdf( in rpc_init()739 return std::make_shared<jit::PythonFutureWrapper>(pyRpcTorchscript( in rpc_init()
135 return result.cast<jit::PythonFutureWrapper&>().fut; in runPythonFunction()
28 return py_fut.cast<std::shared_ptr<torch::jit::PythonFutureWrapper>>()->fut; in runHook()
581 const std::vector<std::shared_ptr<jit::PythonFutureWrapper>>& in c10d_init()647 -> std::shared_ptr<jit::PythonFutureWrapper> { in c10d_init()650 return std::make_shared<jit::PythonFutureWrapper>(fut); in c10d_init()656 -> std::shared_ptr<jit::PythonFutureWrapper> { in c10d_init()659 return std::make_shared<jit::PythonFutureWrapper>(fut); in c10d_init()2967 [](::c10d::Work& work) -> std::shared_ptr<jit::PythonFutureWrapper> { in c10d_init()2968 return std::make_shared<jit::PythonFutureWrapper>(work.getFuture()); in c10d_init()3191 [](const std::shared_ptr<jit::PythonFutureWrapper>& future) { in c10d_init()
38 o.cast<std::shared_ptr<torch::jit::PythonFutureWrapper>>(); in getFuture()