Finalize initial version of TFLite v3 importer (#406)

This commit uses TFLite format visitor to finalize TFLite
importer interface.

Signed-off-by: Dmitry Mozolev <d.mozolev@samsung.com>

diff --git a/contrib/nnc/libs/frontend/tflite/src/tflite_importer.inline.cpp b/contrib/nnc/libs/frontend/tflite/src/tflite_importer.inline.cpp
index 3352a4b..5286c0c 100644 (file)
--- a/contrib/nnc/libs/frontend/tflite/src/tflite_importer.inline.cpp
+++ b/contrib/nnc/libs/frontend/tflite/src/tflite_importer.inline.cpp
@@ -38,8 +38,14 @@ bool TfliteImporter::import()
 
 void *TfliteImporter::createIR()
 {
-  // TODO: implement
-  return nullptr;
+  IrVisitor irCreator{};
+  ModelWalker walker{std::vector<Visitor *>{&irCreator}};
+
+  walker.walk(modelPacked);
+  irCreator.setGraphOutputs();
+  irCreator.setIrNodeNames();
+
+  return irCreator.getGraph();
 }
 
 void TfliteImporter::dump()

diff --git a/contrib/nnc/libs/frontend/tflite/src/tflite_v3_importer.cpp b/contrib/nnc/libs/frontend/tflite/src/tflite_v3_importer.cpp
index 420722e..bc65b80 100644 (file)
--- a/contrib/nnc/libs/frontend/tflite/src/tflite_v3_importer.cpp
+++ b/contrib/nnc/libs/frontend/tflite/src/tflite_v3_importer.cpp
@@ -1,6 +1,7 @@
 #include <iostream>
 
 #include "tflite_v3_importer.h"
+#include "tflite_ir_visitor.h"
 #include "tflite_dump_visitor.h"
 #include "tflite_walker.h"