Searched refs:PythonFunctionGuard (Results 1 – 2 of 2) sorted by relevance
80 struct VISIBILITY_HIDDEN PythonFunctionGuard { struct81 explicit PythonFunctionGuard(py::function func) : func_(std::move(func)) {} in PythonFunctionGuard() argument83 ~PythonFunctionGuard() { in ~PythonFunctionGuard() argument154 auto pf = std::make_shared<PythonFunctionGuard>(std::move(cb)); in then()190 auto pf = std::make_shared<PythonFunctionGuard>(std::move(cb)); in add_done_callback()194 const std::shared_ptr<PythonFunctionGuard>& pf) { in add_done_callback()266 pyfg_ = std::make_shared<torch::jit::PythonFunctionGuard>(std::move(pf)); in PythonAwaitWrapper()305 std::shared_ptr<torch::jit::PythonFunctionGuard> pyfg_;
2067 std::make_shared<torch::jit::PythonFunctionGuard>( in initJITBindings()