Skip to content
Snippets Groups Projects

Fix LSTM ONNX compatibility

Merged Olivier BICHLER requested to merge fix_lstm into dev
1 file
+ 2
1
Compare changes
  • Side-by-side
  • Inline
@@ -198,7 +198,8 @@ void init_MetaOperatorDefs(py::module &m) {
.def(py::init<const char *, const std::shared_ptr<GraphView>&>(),
py::arg("type"),
py::arg("graph"))
.def("get_micro_graph", &MetaOperator_Op::getMicroGraph);
.def("get_micro_graph", &MetaOperator_Op::getMicroGraph)
.def("set_upper_node", &MetaOperator_Op::setUpperNode);
m.def("meta_operator", &MetaOperator,
py::arg("type"),
Loading