From 36f1fe32533690fa7e364c1b411e9bd32b06083e Mon Sep 17 00:00:00 2001 From: Neil Tan Date: Thu, 28 Nov 2019 23:02:47 -0800 Subject: [PATCH] clean up flatbuffers import --- tests/tflm/tflite_export/test_write.py | 3 +-- utensor_cgen/transformer/tflite_exporter.py | 1 - 2 files changed, 1 insertion(+), 3 deletions(-) diff --git a/tests/tflm/tflite_export/test_write.py b/tests/tflm/tflite_export/test_write.py index c5930e71..7fdaf3e2 100644 --- a/tests/tflm/tflite_export/test_write.py +++ b/tests/tflm/tflite_export/test_write.py @@ -5,9 +5,8 @@ from utensor_cgen.matcher import uTensorGraphMatcher from utensor_cgen.utils import prune_graph, topologic_order_graph from utensor_cgen.transformer import TFLiteExporter -import flatbuffers +import utensor_cgen.third_party.flatbuffers as flatbuffers import utensor_cgen.third_party.tflite as tflite -from utensor_cgen.third_party.tflite import * from utensor_cgen.third_party.tflite.BuiltinOperator import BuiltinOperator from utensor_cgen.third_party.tflite.Model import Model from utensor_cgen.third_party.tflite.BuiltinOptions import BuiltinOptions diff --git a/utensor_cgen/transformer/tflite_exporter.py b/utensor_cgen/transformer/tflite_exporter.py index 664cb8b2..2d21b303 100644 --- a/utensor_cgen/transformer/tflite_exporter.py +++ b/utensor_cgen/transformer/tflite_exporter.py @@ -9,7 +9,6 @@ from copy import deepcopy from utensor_cgen.ir import OperationInfo, uTensorGraph -#import flatbuffers import utensor_cgen.third_party.flatbuffers as flatbuffers import utensor_cgen.third_party.tflite as tflite from utensor_cgen.third_party.tflite import *