diff --git a/python_binding/operator/pybind_GlobalAveragePooling.cpp b/python_binding/operator/pybind_GlobalAveragePooling.cpp index 3f5cd4d05a4ff87475214c9bc646dc507be3d55c..26b36fa800db554891066104b15966fdff858081 100644 --- a/python_binding/operator/pybind_GlobalAveragePooling.cpp +++ b/python_binding/operator/pybind_GlobalAveragePooling.cpp @@ -18,11 +18,13 @@ namespace py = pybind11; namespace Aidge { -void init_GlobalAveragePooling(py::module& m) { - py::class_<GlobalAveragePooling_Op, std::shared_ptr<GlobalAveragePooling_Op>, OperatorTensor, Attributes>(m, "GlobalAveragePooling", py::multiple_inheritance()) - .def("get_inputs_name", &GlobalAveragePooling_Op::getInputsName) - .def("get_outputs_name", &GlobalAveragePooling_Op::getOutputsName); +void init_GlobalAveragePooling(py::module &m) { + py::class_<GlobalAveragePooling_Op, std::shared_ptr<GlobalAveragePooling_Op>, + OperatorTensor, Attributes>(m, "GlobalAveragePooling", + py::multiple_inheritance()) + .def("get_inputs_name", &GlobalAveragePooling_Op::getInputsName) + .def("get_outputs_name", &GlobalAveragePooling_Op::getOutputsName); - m.def("Hardmax", &Hardmax, py::arg("name") = ""); + m.def("Globalaveragepooling", &Globalaveragepooling, py::arg("name") = ""); } -} // namespace Aidge \ No newline at end of file +} // namespace Aidge