Skip to content
Snippets Groups Projects
Commit d7c395ac authored by Houssem ROUIS's avatar Houssem ROUIS
Browse files

include new operators headers

parent 504077b9
No related branches found
No related tags found
2 merge requests!59Improvements and fixes,!47Vit operators
...@@ -28,10 +28,13 @@ ...@@ -28,10 +28,13 @@
#include "aidge/operator/Add.hpp" #include "aidge/operator/Add.hpp"
#include "aidge/operator/AvgPooling.hpp" #include "aidge/operator/AvgPooling.hpp"
#include "aidge/operator/BatchNorm.hpp" #include "aidge/operator/BatchNorm.hpp"
#include "aidge/operator/Concat.hpp"
#include "aidge/operator/Conv.hpp" #include "aidge/operator/Conv.hpp"
#include "aidge/operator/ConvDepthWise.hpp" #include "aidge/operator/ConvDepthWise.hpp"
#include "aidge/operator/Div.hpp" #include "aidge/operator/Div.hpp"
#include "aidge/operator/Erf.hpp"
#include "aidge/operator/FC.hpp" #include "aidge/operator/FC.hpp"
#include "aidge/operator/Gather.hpp"
#include "aidge/operator/GenericOperator.hpp" #include "aidge/operator/GenericOperator.hpp"
#include "aidge/operator/MatMul.hpp" #include "aidge/operator/MatMul.hpp"
#include "aidge/operator/MaxPooling.hpp" #include "aidge/operator/MaxPooling.hpp"
...@@ -42,11 +45,15 @@ ...@@ -42,11 +45,15 @@
#include "aidge/operator/Pad.hpp" #include "aidge/operator/Pad.hpp"
#include "aidge/operator/Producer.hpp" #include "aidge/operator/Producer.hpp"
#include "aidge/operator/Pow.hpp" #include "aidge/operator/Pow.hpp"
#include "aidge/operator/ReduceMean.hpp"
#include "aidge/operator/ReLU.hpp" #include "aidge/operator/ReLU.hpp"
#include "aidge/operator/Reshape.hpp"
#include "aidge/operator/Scaling.hpp" #include "aidge/operator/Scaling.hpp"
#include "aidge/operator/Slice.hpp"
#include "aidge/operator/Softmax.hpp" #include "aidge/operator/Softmax.hpp"
#include "aidge/operator/Sqrt.hpp" #include "aidge/operator/Sqrt.hpp"
#include "aidge/operator/Sub.hpp" #include "aidge/operator/Sub.hpp"
#include "aidge/operator/Transpose.hpp"
#include "aidge/scheduler/Scheduler.hpp" #include "aidge/scheduler/Scheduler.hpp"
#include "aidge/utils/Attributes.hpp" #include "aidge/utils/Attributes.hpp"
#include "aidge/utils/StaticAttributes.hpp" #include "aidge/utils/StaticAttributes.hpp"
......
...@@ -23,10 +23,13 @@ void init_Operator(py::module&); ...@@ -23,10 +23,13 @@ void init_Operator(py::module&);
void init_Add(py::module&); void init_Add(py::module&);
void init_AvgPooling(py::module&); void init_AvgPooling(py::module&);
void init_BatchNorm(py::module&); void init_BatchNorm(py::module&);
void init_Concat(py::module&);
void init_Conv(py::module&); void init_Conv(py::module&);
void init_ConvDepthWise(py::module&); void init_ConvDepthWise(py::module&);
void init_Div(py::module&); void init_Div(py::module&);
void init_Erf(py::module&);
void init_FC(py::module&); void init_FC(py::module&);
void init_Gather(py::module&);
void init_GenericOperator(py::module&); void init_GenericOperator(py::module&);
void init_LeakyReLU(py::module&); void init_LeakyReLU(py::module&);
void init_MatMul(py::module&); void init_MatMul(py::module&);
...@@ -35,10 +38,14 @@ void init_MetaOperatorDefs(py::module&); ...@@ -35,10 +38,14 @@ void init_MetaOperatorDefs(py::module&);
void init_Mul(py::module&); void init_Mul(py::module&);
void init_Producer(py::module&); void init_Producer(py::module&);
void init_Pow(py::module&); void init_Pow(py::module&);
void init_ReduceMean(py::module&);
void init_ReLU(py::module&); void init_ReLU(py::module&);
void init_Reshape(py::module&);
void init_Slice(py::module&);
void init_Softmax(py::module&); void init_Softmax(py::module&);
void init_Sqrt(py::module&); void init_Sqrt(py::module&);
void init_Sub(py::module&); void init_Sub(py::module&);
void init_Transpose(py::module&);
void init_Identity(py::module&); void init_Identity(py::module&);
void init_Node(py::module&); void init_Node(py::module&);
...@@ -69,10 +76,13 @@ void init_Aidge(py::module& m){ ...@@ -69,10 +76,13 @@ void init_Aidge(py::module& m){
init_Add(m); init_Add(m);
init_AvgPooling(m); init_AvgPooling(m);
init_BatchNorm(m); init_BatchNorm(m);
init_Concat(m);
init_Conv(m); init_Conv(m);
init_ConvDepthWise(m); init_ConvDepthWise(m);
init_Div(m); init_Div(m);
init_Erf(m);
init_FC(m); init_FC(m);
init_Gather(m);
init_GenericOperator(m); init_GenericOperator(m);
init_LeakyReLU(m); init_LeakyReLU(m);
init_MatMul(m); init_MatMul(m);
...@@ -80,10 +90,14 @@ void init_Aidge(py::module& m){ ...@@ -80,10 +90,14 @@ void init_Aidge(py::module& m){
init_MetaOperatorDefs(m); init_MetaOperatorDefs(m);
init_Mul(m); init_Mul(m);
init_Pow(m); init_Pow(m);
init_ReduceMean(m);
init_ReLU(m); init_ReLU(m);
init_Reshape(m);
init_Slice(m);
init_Softmax(m); init_Softmax(m);
init_Sqrt(m); init_Sqrt(m);
init_Sub(m); init_Sub(m);
init_Transpose(m);
init_Identity(m); init_Identity(m);
init_Producer(m); init_Producer(m);
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment