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
|