| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | python_saved_variable_hooks.cpp | 65 c10::SafePyObject(pack_hook.release().ptr(), getPyInterpreter()), in push_hooks()
66 c10::SafePyObject(unpack_hook.release().ptr(), getPyInterpreter())); in push_hooks()
72 pack_hook.ptr(getPyInterpreter()) != nullptr && in pop_hooks()
73 unpack_hook.ptr(getPyInterpreter()) != nullptr); in pop_hooks()
|
| H A D | init.cpp | 74 c10::impl::PythonDispatcherTLS::set_state(getPyInterpreter()); in EnablePythonDispatcher()
1086 std::make_shared<c10::SafePyObject>(arg, getPyInterpreter())); in push_on_torch_function_stack()
1097 auto* r = mode->ptr(getPyInterpreter()); in pop_torch_function_stack()
1115 auto* r = mode->ptr(getPyInterpreter()); in get_function_stack_at()
1145 arg, getPyInterpreter()), in push_on_torch_dispatch_stack()
1150 arg, getPyInterpreter())); in push_on_torch_dispatch_stack()
1174 r = mode->ptr(getPyInterpreter()); in pop_torch_dispatch_stack()
1177 r = mode->ptr(getPyInterpreter()); in pop_torch_dispatch_stack()
1196 auto* r = mode->ptr(getPyInterpreter()); in get_dispatch_stack_at()
1215 mode, getPyInterpreter()), in set_dispatch_mode()
[all …]
|
| H A D | python_hook.cpp | 182 c10::SafePyObject(value, getPyInterpreter()), in compiled_args()
192 args.add_pre_hook(c10::SafePyObject(value, getPyInterpreter())); in compiled_args()
201 args.add_post_hook(c10::SafePyObject(value, getPyInterpreter())); in compiled_args()
236 c10::SafePyObject hook_obj(value, getPyInterpreter()); in compiled_args()
|
| H A D | python_variable.cpp | 254 c10::impl::GPUTrace::set_trace(getPyInterpreter()); in activateGPUTrace()
272 getPyInterpreter(), /*ignore_hermetic_tls=*/false); in THPVariable_Wrap()
349 getPyInterpreter(), /*ignore_hermetic_tls=*/false) != in isResurrectable()
376 getPyInterpreter(), in THPVariable_tryResurrect()
457 getPyInterpreter(), /*ignore_hermetic_tls=*/false) == in THPVariable_clear()
2019 getPyInterpreter(), /*ignore_hermetic_tls=*/false); in THPVariable_NewWithVar()
2102 getPyInterpreter(), obj, status); in THPVariable_NewWithVar()
|
| H A D | python_function.cpp | 344 args.add_backward(c10::SafePyObject(obj, getPyInterpreter())); in compiled_args()
351 c10::SafePyObject(bw_state, getPyInterpreter())); in compiled_args()
|
| /aosp_15_r20/external/pytorch/torch/csrc/ |
| H A D | Storage.cpp | 46 getPyInterpreter(), /*ignore_hermetic_tls=*/false); in THPStorage_NewWithStorage()
82 getPyInterpreter(), obj, status); in THPStorage_NewWithStorage()
106 !pyobj_slot->check_interpreter(getPyInterpreter())) { in THPStorage_Wrap()
113 getPyInterpreter(), /*ignore_hermetic_tls=*/false); in THPStorage_Wrap()
156 getPyInterpreter(), /*ignore_hermetic_tls=*/true) != in THPStorage_isPreservable()
175 getPyInterpreter(), in THPStorage_tryPreserve()
|
| H A D | PyInterpreter.cpp | 583 getPyInterpreter(), /*ignore_hermetic_tls=*/false); in set_tensor_attr_with_capsule()
611 tensor->pyobj_slot()->check_pyobj(getPyInterpreter()); in get_set_cached_attr()
948 ss << getPyInterpreter(); in name()
957 return op.getPythonOp(getPyInterpreter(), [&]() -> PyObject* { in getTorchApiFunction()
982 c10::impl::PyInterpreter* getPyInterpreter() { in getPyInterpreter() function
|
| H A D | PyInterpreter.h | 12 TORCH_PYTHON_API c10::impl::PyInterpreter* getPyInterpreter();
|
| H A D | Module.cpp | 393 getPyInterpreter(), /*ignore_hermetic_tls=*/false); in THPModule_swap_tensor_impl()
396 getPyInterpreter(), /*ignore_hermetic_tls=*/false); in THPModule_swap_tensor_impl()
407 getPyInterpreter(), a_, c10::impl::PyInterpreterStatus::TAGGED_BY_US); in THPModule_swap_tensor_impl()
409 getPyInterpreter(), b_, c10::impl::PyInterpreterStatus::TAGGED_BY_US); in THPModule_swap_tensor_impl()
2090 std::make_shared<c10::SafePyObject>(arg.ptr(), getPyInterpreter())); in initModule()
2096 auto obj = safe_pyobject->ptr(getPyInterpreter()); in initModule()
|
| H A D | Device.cpp | 185 mode.release().ptr(), getPyInterpreter())); in THPDevice_enter()
|
| /aosp_15_r20/external/pytorch/torch/csrc/utils/ |
| H A D | python_dispatch.cpp | 120 : func_(func.release().ptr(), getPyInterpreter()), in PythonKernelHolder()
195 py::reinterpret_borrow<py::object>(func_.ptr(getPyInterpreter())); in operator ()()
441 func.release().ptr(), getPyInterpreter())); in initDispatchBindings()
899 std::make_unique<c10::SafePyObject>(obj, getPyInterpreter()); in initDispatchBindings()
1004 auto* pyobj = func->ptr(getPyInterpreter()); in python_op_registration_trampoline_impl()
|
| H A D | python_symnode.h | 37 pyobj.release().ptr(), getPyInterpreter()); in PythonSymNodeImpl()
307 return py::handle(pyobj_->ptr(getPyInterpreter())); in getPyObj()
|
| H A D | python_arg_parser.cpp | 378 tf_g->get_cur_mode()->ptr(getPyInterpreter())); in dispatch_on_mode()
382 td_g->get_cur_mode()->ptr(getPyInterpreter())); in dispatch_on_mode()
|
| /aosp_15_r20/external/pytorch/functorch/csrc/dim/ |
| H A D | python_variable_simple.h | 49 TORCH_PYTHON_API c10::impl::PyInterpreter* getPyInterpreter();
|
| H A D | dim.cpp | 1116 …t->unsafeGetTensorImpl()->pyobj_slot()->check_pyobj(getPyInterpreter(), /*ignore_hermetic_tls=*/fa… in handle_from_tensor()
|
| /aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_eager/ |
| H A D | kernel_holder.cpp | 175 pyinterpreter_(getPyInterpreter()) { in AOTIPythonKernelHolder()
|
| /aosp_15_r20/external/pytorch/torch/csrc/dynamo/ |
| H A D | guards.cpp | 2553 PyTypeObject* mode_type = Py_TYPE(mode->ptr(getPyInterpreter())); in check_nopybind()
|