From ce278c8b15c084ba4dfae80852a9b93199c6b784 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gr=C3=A9goire=20KUBLER?= <gregoire.kubler@proton.me>
Date: Tue, 5 Mar 2024 11:29:51 +0100
Subject: [PATCH] fix : removed Hardmax and put Globalaveragepooling instead
chore : format
---
.../operator/pybind_GlobalAveragePooling.cpp | 14 ++++++++------
1 file changed, 8 insertions(+), 6 deletions(-)
diff --git a/python_binding/operator/pybind_GlobalAveragePooling.cpp b/python_binding/operator/pybind_GlobalAveragePooling.cpp
index 3f5cd4d05..26b36fa80 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
--
GitLab