[mir/Caffe importer] Introduce `importModel` method (#5986)

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-caffe-importer/caffe_importer.cpp b/compiler/mir-caffe-importer/caffe_importer.cpp
index 509c549..9d93aac 100644 (file)
--- a/compiler/mir-caffe-importer/caffe_importer.cpp
+++ b/compiler/mir-caffe-importer/caffe_importer.cpp
@@ -88,6 +88,12 @@ Graph *CaffeImporter::createIR()
   return _graph;
 }
 
+std::unique_ptr<mir::Graph> CaffeImporter::importModel()
+{
+  import();
+  return std::unique_ptr<mir::Graph>(createIR());
+}
+
 void CaffeImporter::collectUnsupportedLayers()
 {
   processDeprecatedInput();

diff --git a/compiler/mir-caffe-importer/caffe_importer.h b/compiler/mir-caffe-importer/caffe_importer.h
index cf4f663..4c7a65d 100644 (file)
--- a/compiler/mir-caffe-importer/caffe_importer.h
+++ b/compiler/mir-caffe-importer/caffe_importer.h
@@ -36,6 +36,9 @@ public:
   void import();
   mir::Graph *createIR();
 
+  /// @brief Load the model and convert it into a MIR Graph.
+  std::unique_ptr<mir::Graph> importModel();
+
   void cleanup();
 
   ~CaffeImporter();