diff --git a/python_binding/operator/pybind_MetaOperatorDefs.cpp b/python_binding/operator/pybind_MetaOperatorDefs.cpp index 5f173068af0f1140830d458979ec924c38ade078..90edfe791a5a1c87fc18b961eded32358e407c5f 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>&>(),