Searched refs:torch_api_function (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/utils/ |
| H A D | python_arg_parser.cpp | 233 py::object torch_api_function = in handle_torch_function() local
236 torch_api_function.ptr() != nullptr, "torch API function must exist"); in handle_torch_function()
243 torch_api_function.ptr(), in handle_torch_function()
264 PyObject* torch_api_function, in maybe_get_registered_torch_dispatch_rule() argument
274 py::reinterpret_borrow<py::object>(torch_api_function), in maybe_get_registered_torch_dispatch_rule()
284 PyObject* torch_api_function, in dispatch_on_subclass() argument
320 torch_api_function, py::reinterpret_borrow<py::object>(arg)); in dispatch_on_subclass()
327 torch_api_function, in dispatch_on_subclass()
343 torch_api_function, in dispatch_on_subclass()
365 PyObject* torch_api_function, in dispatch_on_mode() argument
[all …]
|
| H A D | python_arg_parser.h | 1217 PyObject* torch_api_function,
|
| /aosp_15_r20/external/pytorch/torch/csrc/ |
| H A D | PyInterpreter.cpp | 183 PyObject* torch_api_function, in torchDispatchFromTensorImpl() argument
187 if (torch_api_function == nullptr) {
225 torch_api_function,
970 py::handle torch_api_function = in getTorchApiFunction() local
973 return torch_api_function.attr("default").ptr(); in getTorchApiFunction()
975 return torch_api_function.attr(overload_name.c_str()).ptr(); in getTorchApiFunction()
|
| /aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_eager/ |
| H A D | kernel_holder.cpp | 482 py::handle torch_api_function = py::module::import("torch") in produce_aoti_kernel_lib() local
487 return torch_api_function.attr("default").ptr(); in produce_aoti_kernel_lib()
489 return torch_api_function.attr(overload_name.c_str()).ptr(); in produce_aoti_kernel_lib()
|