• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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