| #pragma once | |
| #include <torch/csrc/python_headers.h> | |
| #include <pybind11/pybind11.h> | |
| #include <pybind11/stl.h> | |
| #include <torch/csrc/autograd/python_function.h> | |
| #include <torch/csrc/autograd/python_cpp_function.h> | |
| namespace py = pybind11; | |
| namespace pybind11 { namespace detail { | |
| }} // namespace pybind11::detail |