From aab35aabd0e2eeebbab30efe757a8727e4b11dc2 Mon Sep 17 00:00:00 2001
From: hrouis <houssemeddine.rouis92@gmail.com>
Date: Tue, 18 Feb 2025 15:55:20 +0100
Subject: [PATCH] fix python binding of FC_Op
---
python_binding/operator/pybind_FC.cpp | 9 ++++++++-
1 file changed, 8 insertions(+), 1 deletion(-)
diff --git a/python_binding/operator/pybind_FC.cpp b/python_binding/operator/pybind_FC.cpp
index 447d0b29a..40433eb51 100644
--- a/python_binding/operator/pybind_FC.cpp
+++ b/python_binding/operator/pybind_FC.cpp
@@ -29,7 +29,14 @@ void declare_FC(py::module &m) {
:param type : The type of the Fully Connected operation.
:type type : :py:class:`str`
)mydelimiter")
- .def(py::init<>())
+ .def(py::init<float,
+ float,
+ bool,
+ bool>(),
+ py::arg("alpha")=1.0,
+ py::arg("beta")=1.0,
+ py::arg("transA")=false,
+ py::arg("transB")=false)
.def_static("get_inputs_name", &FC_Op::getInputsName)
.def_static("get_outputs_name", &FC_Op::getOutputsName)
.def_readonly_static("Type", &FC_Op::Type)
--
GitLab