| 6 | using dnn::OnnxConverter; |
| 7 | |
| 8 | void convert(const std::string &model_str, const std::string &filepath, |
| 9 | const std::string &table_file = "") { |
| 10 | OnnxConverter converter; |
| 11 | converter.Convert(model_str, filepath, table_file); |
| 12 | } |
| 13 | |
| 14 | PYBIND11_MODULE(_onnx2daq, m) { |
| 15 | m.def("convert", &convert, ""); |