Skip to content
Snippets Groups Projects
Commit 352bdfb6 authored by Cyril Moineau's avatar Cyril Moineau
Browse files

Add binding for MetaOperator

parent 32d3a670
No related branches found
No related tags found
1 merge request!50Onnx tuto
......@@ -121,6 +121,15 @@ void init_MetaOperatorDefs(py::module &m) {
declare_PaddedMaxPoolingOp<2>(m);
declare_PaddedMaxPoolingOp<3>(m);
py::class_<MetaOperator_Op, std::shared_ptr<MetaOperator_Op>, Operator>(m, "MetaOperator_Op", py::multiple_inheritance());
m.def("meta_operator", &MetaOperator,
py::arg("type"),
py::arg("graph"),
py::arg("name") = "",
py::arg("input_nodes") = std::vector<NodePtr>(),
py::arg("output_nodes") = std::vector<NodePtr>()
);
}
} // 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