Home
last modified time | relevance | path

Searched refs:PythonAwaitWrapper (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/python/
H A Dpybind_utils.h252 struct VISIBILITY_HIDDEN PythonAwaitWrapper struct
253 : std::enable_shared_from_this<PythonAwaitWrapper> {
254 explicit PythonAwaitWrapper(c10::intrusive_ptr<c10::ivalue::Await> aw) in PythonAwaitWrapper() argument
256 explicit PythonAwaitWrapper(py::handle input) { in PythonAwaitWrapper() function
264 explicit PythonAwaitWrapper(py::function pf, py::tuple args) in PythonAwaitWrapper() argument
276 explicit PythonAwaitWrapper(const PythonAwaitWrapper&) = delete;
277 PythonAwaitWrapper& operator=(const PythonAwaitWrapper&) = delete; argument
309 std::shared_ptr<PythonAwaitWrapper> getPtr() { in getPtr() argument
482 auto awptr = input.cast<std::shared_ptr<PythonAwaitWrapper>>(); in tryToInferType()
H A Dinit.cpp2097 py::class_<PythonAwaitWrapper, std::shared_ptr<PythonAwaitWrapper>>( in initJITBindings()
2101 &PythonAwaitWrapper::wait, in initJITBindings()
2103 .def("fn", &PythonAwaitWrapper::fn) in initJITBindings()
2104 .def("args", &PythonAwaitWrapper::args) in initJITBindings()
2105 .def("type", &PythonAwaitWrapper::type) in initJITBindings()
2106 .def("is_nowait", &PythonAwaitWrapper::is_nowait) in initJITBindings()
2109 [](PythonAwaitWrapper& self, const std::string& name) -> py::object { in initJITBindings()
2117 [](const PythonAwaitWrapper& /* unused */) { in initJITBindings() argument
2160 return std::make_shared<PythonAwaitWrapper>( in initJITBindings()
2164 return std::make_shared<PythonAwaitWrapper>(input); in initJITBindings()
[all …]
H A Dpybind_utils.cpp536 return obj.cast<std::shared_ptr<PythonAwaitWrapper>>()->aw_; in toIValue()
722 return py::cast(std::make_shared<PythonAwaitWrapper>(ivalue.toAwait())); in toPyObject()