1 | #include <pybind11/pybind11.h> |
---|---|
2 | #include <torch/csrc/utils/pybind.h> |
3 | |
4 | namespace torch { |
5 | namespace impl { |
6 | namespace dispatch { |
7 | |
8 | void initDispatchBindings(PyObject* module); |
9 | |
10 | void python_op_registration_trampoline_impl( |
11 | const c10::OperatorHandle& op, |
12 | c10::DispatchKey key, |
13 | torch::jit::Stack* stack); |
14 | |
15 | } // namespace dispatch |
16 | } // namespace impl |
17 | } // namespace torch |
18 |