diff --git a/aidge_core/aidge_export_aidge/templates/pybind.jinja b/aidge_core/aidge_export_aidge/templates/pybind.jinja new file mode 100644 index 0000000000000000000000000000000000000000..9b48cc97f242813cc33b77bfc028e85c08b0cec7 --- /dev/null +++ b/aidge_core/aidge_export_aidge/templates/pybind.jinja @@ -0,0 +1,14 @@ +#include <pybind11/pybind11.h> + +#include "dnn.hpp" + +namespace py = pybind11; + + +void init_{{name}}(py::module& m){ + m.def("generate_model", generateModel); +} + +PYBIND11_MODULE({{name}}, m) { + init_{{name}}(m); +}