From 1b0d741fc1e18e18cc790f4908ac3fe7a80c1b99 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Dmitry=20Mozolev/AI=20Tools=20Lab=20/SRR/Engineer/=EC=82=BC?= =?utf8?q?=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Fri, 29 Jun 2018 14:25:22 +0300 Subject: [PATCH] Finalize initial version of TFLite v3 importer (#406) This commit uses TFLite format visitor to finalize TFLite importer interface. Signed-off-by: Dmitry Mozolev --- .../nnc/libs/frontend/tflite/src/tflite_importer.inline.cpp | 10 ++++++++-- contrib/nnc/libs/frontend/tflite/src/tflite_v3_importer.cpp | 1 + 2 files changed, 9 insertions(+), 2 deletions(-) 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 --- 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{&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 --- 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 #include "tflite_v3_importer.h" +#include "tflite_ir_visitor.h" #include "tflite_dump_visitor.h" #include "tflite_walker.h" -- 2.7.4