From d7c395acc3f3f0c83ac5802c318a64b5e64feff4 Mon Sep 17 00:00:00 2001
From: hrouis <houssemeddine.rouis92@gmail.com>
Date: Thu, 23 Nov 2023 15:28:42 +0100
Subject: [PATCH] include new operators headers
---
include/aidge/aidge.hpp | 7 +++++++
python_binding/pybind_core.cpp | 14 ++++++++++++++
2 files changed, 21 insertions(+)
diff --git a/include/aidge/aidge.hpp b/include/aidge/aidge.hpp
index 65dc7f70c..82aecf163 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 9c8d8c61d..518da5200 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);
--
GitLab