From a570e3c01925626039351c264e9fb7671ab6bfb2 Mon Sep 17 00:00:00 2001
From: Jerome Hue <jerome.hue@cea.fr>
Date: Mon, 9 Dec 2024 12:30:13 +0100
Subject: [PATCH] feat: Add python bindings for Leaky 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 5f173068a..90edfe791 100644
--- a/python_binding/operator/pybind_MetaOperatorDefs.cpp
+++ b/python_binding/operator/pybind_MetaOperatorDefs.cpp
@@ -179,6 +179,14 @@ void declare_LSTMOp(py::module &m) {
py::arg("seq_length"));
}
+void declare_LeakyOp(py::module &m) {
+ m.def("Leaky", &Leaky,
+ py::arg("nb_timesteps"),
+ py::arg("threshold"),
+ py::arg("beta"),
+ py::arg("name") = "");
+}
+
void init_MetaOperatorDefs(py::module &m) {
declare_PaddedConvOp<1>(m);
declare_PaddedConvOp<2>(m);
@@ -193,6 +201,7 @@ void init_MetaOperatorDefs(py::module &m) {
declare_PaddedMaxPoolingOp<2>(m);
// declare_PaddedMaxPoolingOp<3>(m);
declare_LSTMOp(m);
+ declare_LeakyOp(m);
py::class_<MetaOperator_Op, std::shared_ptr<MetaOperator_Op>, OperatorTensor>(m, "MetaOperatorOp", py::multiple_inheritance())
.def(py::init<const char *, const std::shared_ptr<GraphView>&, const std::vector<InputCategory>&>(),
--
GitLab