Searched defs:torch_api_function (Results 1 – 3 of 3) sorted by relevance
233 py::object torch_api_function = in handle_torch_function() local264 PyObject* torch_api_function, in maybe_get_registered_torch_dispatch_rule()284 PyObject* torch_api_function, in dispatch_on_subclass()365 PyObject* torch_api_function, in dispatch_on_mode()450 PyObject* torch_api_function, in handle_torch_function_no_python_arg_parser()601 py::object torch_api_function = PyObject_FastGetAttrString( in handle_torch_function() local
183 PyObject* torch_api_function, in torchDispatchFromTensorImpl()970 py::handle torch_api_function = in getTorchApiFunction() local
482 py::handle torch_api_function = py::module::import("torch") in produce_aoti_kernel_lib() local