Skip to content
Snippets Groups Projects
Commit ce278c8b authored by Grégoire Kubler's avatar Grégoire Kubler
Browse files

fix : removed Hardmax and put Globalaveragepooling instead

chore : format
parent e2b4f0d8
No related branches found
No related tags found
2 merge requests!105version 0.2.0,!91Feat/operator global average pooling
......@@ -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
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