Searched refs:py_compiler (Results 1 – 2 of 2) sorted by relevance
/aosp_15_r20/external/pytorch/torch/csrc/dynamo/ |
H A D | python_compiled_autograd.cpp | 549 ClosingTHPObjectPtr py_compiler( in _compiled_autograd_impl() local 553 py_compiler, *cache, compiler_call, output_edges.size()); in _compiled_autograd_impl() 574 py_compiler, in _compiled_autograd_impl() 598 py_compiler.get(), "pre_hook", "Oi", pyinputs.get(), hook)); in _compiled_autograd_impl() 607 PyObject_GetAttrString(py_compiler.get(), "set_node_origin")); in _compiled_autograd_impl() 617 SwapSavedVariables saved(compiler_call, state, py_compiler.get(), call); in _compiled_autograd_impl() 637 py_compiler.get(), in _compiled_autograd_impl() 657 PyObject* res = check(call_end_capture(py_compiler, state.outputs)); in _compiled_autograd_impl()
|
H A D | compiled_autograd.h | 776 : compiler(c), state(s), py_compiler(p), curr_node_call(n) {} in SwapSavedVariables() 779 return py_compiler; in get_py_compiler() 832 PyObject* py_compiler; variable
|