diff --git a/include/aidge/aidge.hpp b/include/aidge/aidge.hpp index 65dc7f70cde21e522fee29ced4552e6801f0b923..82aecf163d477072d9290574350daa47d506bca4 100644 --- a/include/aidge/aidge.hpp +++ b/include/aidge/aidge.hpp @@ -28,10 +28,13 @@ #include "aidge/operator/Add.hpp" #include "aidge/operator/AvgPooling.hpp" #include "aidge/operator/BatchNorm.hpp" +#include "aidge/operator/Concat.hpp" #include "aidge/operator/Conv.hpp" #include "aidge/operator/ConvDepthWise.hpp" #include "aidge/operator/Div.hpp" +#include "aidge/operator/Erf.hpp" #include "aidge/operator/FC.hpp" +#include "aidge/operator/Gather.hpp" #include "aidge/operator/GenericOperator.hpp" #include "aidge/operator/MatMul.hpp" #include "aidge/operator/MaxPooling.hpp" @@ -42,11 +45,15 @@ #include "aidge/operator/Pad.hpp" #include "aidge/operator/Producer.hpp" #include "aidge/operator/Pow.hpp" +#include "aidge/operator/ReduceMean.hpp" #include "aidge/operator/ReLU.hpp" +#include "aidge/operator/Reshape.hpp" #include "aidge/operator/Scaling.hpp" +#include "aidge/operator/Slice.hpp" #include "aidge/operator/Softmax.hpp" #include "aidge/operator/Sqrt.hpp" #include "aidge/operator/Sub.hpp" +#include "aidge/operator/Transpose.hpp" #include "aidge/scheduler/Scheduler.hpp" #include "aidge/utils/Attributes.hpp" #include "aidge/utils/StaticAttributes.hpp" diff --git a/python_binding/pybind_core.cpp b/python_binding/pybind_core.cpp index 9c8d8c61daf836c52b06b90b1c88577172fe22e5..518da520043f09ec057f6f4673332733af8e92bd 100644 --- a/python_binding/pybind_core.cpp +++ b/python_binding/pybind_core.cpp @@ -23,10 +23,13 @@ void init_Operator(py::module&); void init_Add(py::module&); void init_AvgPooling(py::module&); void init_BatchNorm(py::module&); +void init_Concat(py::module&); void init_Conv(py::module&); void init_ConvDepthWise(py::module&); void init_Div(py::module&); +void init_Erf(py::module&); void init_FC(py::module&); +void init_Gather(py::module&); void init_GenericOperator(py::module&); void init_LeakyReLU(py::module&); void init_MatMul(py::module&); @@ -35,10 +38,14 @@ void init_MetaOperatorDefs(py::module&); void init_Mul(py::module&); void init_Producer(py::module&); void init_Pow(py::module&); +void init_ReduceMean(py::module&); void init_ReLU(py::module&); +void init_Reshape(py::module&); +void init_Slice(py::module&); void init_Softmax(py::module&); void init_Sqrt(py::module&); void init_Sub(py::module&); +void init_Transpose(py::module&); void init_Identity(py::module&); void init_Node(py::module&); @@ -69,10 +76,13 @@ void init_Aidge(py::module& m){ init_Add(m); init_AvgPooling(m); init_BatchNorm(m); + init_Concat(m); init_Conv(m); init_ConvDepthWise(m); init_Div(m); + init_Erf(m); init_FC(m); + init_Gather(m); init_GenericOperator(m); init_LeakyReLU(m); init_MatMul(m); @@ -80,10 +90,14 @@ void init_Aidge(py::module& m){ init_MetaOperatorDefs(m); init_Mul(m); init_Pow(m); + init_ReduceMean(m); init_ReLU(m); + init_Reshape(m); + init_Slice(m); init_Softmax(m); init_Sqrt(m); init_Sub(m); + init_Transpose(m); init_Identity(m); init_Producer(m);