xref: /aosp_15_r20/external/pytorch/torch/csrc/utils/python_dispatch.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1 #include <pybind11/pybind11.h>
2 #include <torch/csrc/utils/pybind.h>
3 
4 namespace torch::impl::dispatch {
5 
6 void initDispatchBindings(PyObject* module);
7 
8 void python_op_registration_trampoline_impl(
9     const c10::OperatorHandle& op,
10     c10::DispatchKey key,
11     c10::DispatchKeySet keyset,
12     torch::jit::Stack* stack,
13     bool with_keyset,
14     bool with_op);
15 
16 } // namespace torch::impl::dispatch
17