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

fix : removed attributes since globalavgpool doesn't have any

parent ce278c8b
No related branches found
No related tags found
2 merge requests!105version 0.2.0,!91Feat/operator global average pooling
......@@ -20,11 +20,11 @@ namespace Aidge {
void init_GlobalAveragePooling(py::module &m) {
py::class_<GlobalAveragePooling_Op, std::shared_ptr<GlobalAveragePooling_Op>,
OperatorTensor, Attributes>(m, "GlobalAveragePooling",
py::multiple_inheritance())
OperatorTensor>(m, "GlobalAveragePooling",
py::multiple_inheritance())
.def("get_inputs_name", &GlobalAveragePooling_Op::getInputsName)
.def("get_outputs_name", &GlobalAveragePooling_Op::getOutputsName);
m.def("Globalaveragepooling", &Globalaveragepooling, py::arg("name") = "");
m.def("globalaveragepooling", &GlobalAveragePooling, py::arg("name") = "");
}
} // 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