[tf2tflite] Initial commit (#3779)

This commit introduces tf2tflite with minimal implementation.

Signed-off-by: Jonghyun Park <jh1302.park@samsung.com>

diff --git a/contrib/tf2tflite/.FORMATCHECKED b/contrib/tf2tflite/.FORMATCHECKED
new file mode 100644 (file)
index 0000000..e69de29

diff --git a/contrib/tf2tflite/CMakeLists.txt b/contrib/tf2tflite/CMakeLists.txt
new file mode 100644 (file)
index 0000000..f1562cc
--- /dev/null
+++ b/contrib/tf2tflite/CMakeLists.txt
@@ -0,0 +1,19 @@
+# TODO Allow users to force tf2tflite build
+if(NOT TARGET moco_tf_frontend)
+  return()
+endif(NOT TARGET moco_tf_frontend)
+
+if(NOT TARGET nnkit_support_tftestinfo)
+  return()
+endif(NOT TARGET nnkit_support_tftestinfo)
+
+if(NOT TARGET loco_exporter)
+  return()
+endif(NOT TARGET loco_exporter)
+
+file(GLOB_RECURSE SOURCES "src/*.cpp")
+
+add_executable(tf2tflite ${SOURCES})
+target_link_libraries(tf2tflite PRIVATE moco_tf_frontend)
+target_link_libraries(tf2tflite PRIVATE nnkit_support_tftestinfo)
+target_link_libraries(tf2tflite PRIVATE loco_exporter)

diff --git a/contrib/tf2tflite/README.md b/contrib/tf2tflite/README.md
new file mode 100644 (file)
index 0000000..64f4fb7
--- /dev/null
+++ b/contrib/tf2tflite/README.md
@@ -0,0 +1,3 @@
+# tf2tflite
+
+_tf2tflite_ is a TensorFlow-to-TensorFlow Lite model converter.

diff --git a/contrib/tf2tflite/requires.cmake b/contrib/tf2tflite/requires.cmake
new file mode 100644 (file)
index 0000000..1037a6c
--- /dev/null
+++ b/contrib/tf2tflite/requires.cmake
@@ -0,0 +1,3 @@
+require("moco")
+require("nnkit")
+require("loco-exporter")

diff --git a/contrib/tf2tflite/src/Driver.cpp b/contrib/tf2tflite/src/Driver.cpp
new file mode 100644 (file)
index 0000000..bca0e35
--- /dev/null
+++ b/contrib/tf2tflite/src/Driver.cpp
@@ -0,0 +1,68 @@
+#include <moco/tf/Frontend.h>
+#include <LocoExporter.h>
+
+#include <nnkit/support/tftestinfo/TensorInfoParser.h>
+
+#include <cassert>
+
+#include <iostream>
+#include <stdexcept>
+#include <string>
+
+namespace
+{
+
+std::unique_ptr<loco::Graph> import(const moco::tf::ModelSignature &sig, const std::string &path)
+{
+  moco::tf::Frontend frontend;
+  return frontend.load(sig, path.c_str(), moco::tf::Frontend::FileType::Binary);
+}
+
+} // namespace
+
+int main(int argc, char **argv)
+{
+  if (argc != 4)
+  {
+    std::cerr << "ERROR: tf2tflite <path/to/info> <path/to/pb> <path/to/tflite/model>" << std::endl;
+    return 255;
+  }
+
+  std::string info_path{argv[1]};
+  std::string tf_path{argv[2]}; // .pb file
+  std::string tflite_path{argv[3]};
+
+  std::cout << "Read '" << info_path << "'" << std::endl;
+
+  moco::tf::ModelSignature sig;
+  {
+    for (const auto &info : nnkit::support::tftestinfo::parse(info_path.c_str()))
+    {
+      switch (info->kind())
+      {
+      case nnkit::support::tftestinfo::ParsedTensor::Kind::Input:
+        sig.add_input(info->nodeName());
+        break;
+
+      case nnkit::support::tftestinfo::ParsedTensor::Kind::Output:
+        sig.add_output(info->nodeName());
+        break;
+
+      default:
+        throw std::runtime_error{"Unknown kind"};
+      }
+    }
+  }
+
+  std::cout << "Read '" << info_path << "' - Done" << std::endl;
+
+  std::cout << "Import from '" << tf_path << "'" << std::endl;
+  auto g = import(sig, tf_path);
+  std::cout << "Import from '" << tf_path << "' - Done" << std::endl;
+
+  std::cout << "Export into '" << tflite_path << "'" << std::endl;
+  loco_exporter::Exporter(g.get()).dumpToFile(tflite_path.c_str());
+  std::cout << "Export into '" << tflite_path << "' - Done" << std::endl;
+
+  return 0;
+}