F/tflite parser

.gitignore

@@ -1,3 +1,6 @@
+#system files
+__pycache__
+.DS_Store
 # virtual env
 .venv
 venv
@@ -20,4 +23,4 @@ tests/deep_mlp/data
 .vscode
 # docs
 doc/build
-doc/source/_build
+doc/source/_build

setup.py

@@ -61,6 +61,7 @@ class _Develop(_CompileFlatbuffMixin, _develop): pass
         'graphviz',
         'matplotlib',
         'toml',
+        'flatbuffers',
         'ortools',
     ],
     extras_require={

tests/test_frontend/conftest.py

@@ -0,0 +1,13 @@
+import os
+
+import pytest
+
+
+@pytest.fixture(name='tflm_mnist_path', scope='session')
+def tflm_mnist_path():
+    model_path = os.path.join(
+        os.path.dirname(__file__),
+        'model_files',
+        'quant_mnist_cnn.tflite'
+    )
+    return model_path

tests/test_frontend/test_tflite.py

@@ -0,0 +1,11 @@
+def test_parser(tflm_mnist_path):
+    from utensor_cgen.frontend.tflite import TFLiteParser
+
+    parser = TFLiteParser({})
+    ugraph = parser.parse(tflm_mnist_path)
+
+    assert ugraph.output_nodes, \
+        'output_nodes is empty: {}'.format(ugraph.output_nodes)
+    assert ugraph.topo_order, \
+        'topo_order is empty: {}'.format(ugraph.topo_order)
+    assert ugraph.lib_name == 'tflite'

utensor_cgen/backend/utensor/snippets/_types.py

@@ -55,6 +55,8 @@ def _init(cls):
                                             tensor_type_str="uint16_t"),
         np.dtype('int8'): _TYPE_MAP_VALUE(importer_type_str="int8",
                                           tensor_type_str="q7_t"),
+        np.dtype('uint8'): _TYPE_MAP_VALUE(importer_type_str="uint8",
+                                          tensor_type_str="uint8_t"),
       }
       cls._inited = True