diff --git a/aidge/_CPU/include/aidge_cpu.hpp b/aidge/_CPU/include/aidge_cpu.hpp new file mode 100644 index 0000000000000000000000000000000000000000..4b0b1ca50c78ab4354f36c5fcdfb5b369fd0656a --- /dev/null +++ b/aidge/_CPU/include/aidge_cpu.hpp @@ -0,0 +1,6 @@ +#include "data/TensorImpl.hpp" +#include "operator/AddImpl.hpp" +#include "operator/ConvImpl.hpp" +#include "operator/FCImpl.hpp" +#include "operator/ProducerImpl.hpp" +#include "operator/ReLUImpl.hpp" \ No newline at end of file diff --git a/aidge/_CPU/python_binding/pybind_cpu.cpp b/aidge/_CPU/python_binding/pybind_cpu.cpp index 6b5b7cd16a4cac9b418984dc798caed5595373aa..f8267163bd85a4be51015ad0331f448f6ff3a52e 100644 --- a/aidge/_CPU/python_binding/pybind_cpu.cpp +++ b/aidge/_CPU/python_binding/pybind_cpu.cpp @@ -1,7 +1,6 @@ #include <pybind11/pybind11.h> // Need to call this header to register every impl -#include "data/TensorImpl.hpp" -#include "operator/AddImpl.hpp" +#include "aidge_cpu.hpp" namespace py = pybind11;