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

fix python binding of FC_Op

parent e95ac474
No related branches found
No related tags found
1 merge request!350Draft: Fix import tests
...@@ -29,7 +29,14 @@ void declare_FC(py::module &m) { ...@@ -29,7 +29,14 @@ void declare_FC(py::module &m) {
:param type : The type of the Fully Connected operation. :param type : The type of the Fully Connected operation.
:type type : :py:class:`str` :type type : :py:class:`str`
)mydelimiter") )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_inputs_name", &FC_Op::getInputsName)
.def_static("get_outputs_name", &FC_Op::getOutputsName) .def_static("get_outputs_name", &FC_Op::getOutputsName)
.def_readonly_static("Type", &FC_Op::Type) .def_readonly_static("Type", &FC_Op::Type)
......
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