Skip to content
Snippets Groups Projects
Commit 25e11cc2 authored by Cyril Moineau's avatar Cyril Moineau Committed by Maxence Naud
Browse files

Make MetaOperator registrable in Python.

parent d8e391dd
No related branches found
No related tags found
3 merge requests!279v0.4.0,!253v0.4.0,!163Export refactor
Hide whitespace changes
Inline Side-by-side
python_binding/operator/pybind_MetaOperatorDefs.cpp
+ 2
− 2
View file @ 25e11cc2
...@@ -194,8 +194,8 @@ void init_MetaOperatorDefs(py::module &m) { ...@@ -194,8 +194,8 @@ void init_MetaOperatorDefs(py::module &m) {
// declare_PaddedMaxPoolingOp<3>(m); // declare_PaddedMaxPoolingOp<3>(m);
declare_LSTMOp(m); declare_LSTMOp(m);
py::class_<MetaOperator_Op, std::shared_ptr<MetaOperator_Op>, OperatorTensor>(m, "MetaOperator_Op", py::multiple_inheritance()) py::class_<MetaOperator_Op, std::shared_ptr<MetaOperator_Op>, OperatorTensor>(m, "MetaOperatorOp", py::multiple_inheritance())
.def(py::init<const char *, const std::shared_ptr<GraphView>&, const std::vector<InputCategory>&>(), .def(py::init<const char *, const std::shared_ptr<GraphView>&>(),
py::arg("type"), py::arg("type"),
py::arg("graph"), py::arg("graph"),
py::arg("forced_inputs_category") = std::vector<InputCategory>()) py::arg("forced_inputs_category") = std::vector<InputCategory>())
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment

Copyright © Eclipse Foundation, Inc. All Rights Reserved.     Privacy Policy | Terms of Use | Copyright Agent