From 352bdfb68527d896772ffe1b2ce236626505c6c9 Mon Sep 17 00:00:00 2001
From: cmoineau <cyril.moineau@cea.fr>
Date: Wed, 15 Nov 2023 08:27:07 +0000
Subject: [PATCH] Add binding for MetaOperator
---
python_binding/operator/pybind_MetaOperatorDefs.cpp | 9 +++++++++
1 file changed, 9 insertions(+)
diff --git a/python_binding/operator/pybind_MetaOperatorDefs.cpp b/python_binding/operator/pybind_MetaOperatorDefs.cpp
index aa9f3c50e..3fc0f89c0 100644
--- a/python_binding/operator/pybind_MetaOperatorDefs.cpp
+++ b/python_binding/operator/pybind_MetaOperatorDefs.cpp
@@ -121,6 +121,15 @@ void init_MetaOperatorDefs(py::module &m) {
declare_PaddedMaxPoolingOp<2>(m);
declare_PaddedMaxPoolingOp<3>(m);
+ py::class_<MetaOperator_Op, std::shared_ptr<MetaOperator_Op>, Operator>(m, "MetaOperator_Op", py::multiple_inheritance());
+
+ m.def("meta_operator", &MetaOperator,
+ py::arg("type"),
+ py::arg("graph"),
+ py::arg("name") = "",
+ py::arg("input_nodes") = std::vector<NodePtr>(),
+ py::arg("output_nodes") = std::vector<NodePtr>()
+ );
}
} // namespace Aidge
--
GitLab