[mir/ONNX importer] Introduce `importModel` method (#5987)

Add `importModel` method as a future replacement for `import` and `createIR` methods.

Signed-off-by: Sergei Barannikov <s.barannikov@samsung.com>

diff --git a/compiler/mir-onnx-importer/ONNXImporterImpl.cpp b/compiler/mir-onnx-importer/ONNXImporterImpl.cpp
index 91a1a29..eaec152 100644 (file)
--- a/compiler/mir-onnx-importer/ONNXImporterImpl.cpp
+++ b/compiler/mir-onnx-importer/ONNXImporterImpl.cpp
@@ -347,4 +347,11 @@ mir::Graph *ONNXImporterImpl::createIR()
 
   return _graph;
 }
+
+std::unique_ptr<mir::Graph> ONNXImporterImpl::importModel()
+{
+  import();
+  return std::unique_ptr<mir::Graph>(createIR());
+}
+
 } // namespace nnc

diff --git a/compiler/mir-onnx-importer/ONNXImporterImpl.h b/compiler/mir-onnx-importer/ONNXImporterImpl.h
index 8985c32..5ae3184 100644 (file)
--- a/compiler/mir-onnx-importer/ONNXImporterImpl.h
+++ b/compiler/mir-onnx-importer/ONNXImporterImpl.h
@@ -41,6 +41,9 @@ public:
   void import();
   mir::Graph *createIR();
 
+  /// @brief Load the model and convert it into a MIR Graph.
+  std::unique_ptr<mir::Graph> importModel();
+
   static mir::TensorVariant createTensor(const onnx::TensorProto *tensor);
 
 private: