update tests

uTensor · May 6, 2020 · 963cbfa · 963cbfa
1 parent d348f9c
commit 963cbfa
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 10 deletions.
diff --git a/tests/tflm/tflite_export/__init__.py → ...sformer/test_tflm/test_export/__init__.py b/tests/tflm/tflite_export/__init__.py → ...sformer/test_tflm/test_export/__init__.py
diff --git a/tests/tflm/tflite_export/conftest.py → ...sformer/test_tflm/test_export/conftest.py b/tests/tflm/tflite_export/conftest.py → ...sformer/test_tflm/test_export/conftest.py
@@ -1,17 +1,14 @@
 import numpy as np
 from pytest import fixture
 
-import tensorflow as tf
-from utensor_cgen.ir import TensorInfo, OperationInfo, uTensorGraph
-from utensor_cgen.ir.converter import (AttrValueConverter, DataTypeConverter,
-                                       GenericTensorConverterMixin)
-from utensor_cgen.utils import prune_graph, topologic_order_graph
-from utensor_cgen.backend.operators import OperatorFactory, _Operator
-from utensor_cgen.matcher import OpEqualityDelegate, _morphism
+from utensor_cgen.backend.utensor.code_generator.legacy._operators import (
+    OperatorFactory, _Operator, uTensorOpEqualityDelegate)
+from utensor_cgen.ir import OperationInfo, TensorInfo, uTensorGraph
+from utensor_cgen.ir.converter import AttrValueConverter, DataTypeConverter
 
 
 @OperatorFactory.register
-@OpEqualityDelegate.is_associative(
+@uTensorOpEqualityDelegate.is_associative(
   permutations=((0, 1), (1, 0))
 )
 class _TFLM_AddOperator(_Operator):
@@ -51,7 +48,7 @@ def build_op_info(cls, ugraph, name, tensor_x, tensor_y, **kwargs):
         )
       },
       ugraph=ugraph,
-      backend=kwargs.get('backend', 'TFLM')
+      lib_name=kwargs.get('lib_name', 'tflite')
     )
 
 
@@ -93,7 +90,7 @@ def build_op_info(cls, ugraph, name, tensor_x, tensor_w, tensor_b, **kwargs):
         )
       },
       ugraph=ugraph,
-      backend=kwargs.get('backend', 'TFLM')
+      lib_name=kwargs.get('lib_name', 'tflite')
     )
 
 @fixture(name='hybrid_quant_output')

diff --git a/tests/tflm/tflite_export/test_write.py → ...ormer/test_tflm/test_export/test_write.py b/tests/tflm/tflite_export/test_write.py → ...ormer/test_tflm/test_export/test_write.py