Skip to content

Commit 31b9f50

Browse files
committed
update tests
1 parent cae7806 commit 31b9f50

File tree

3 files changed

+7
-10
lines changed

3 files changed

+7
-10
lines changed

tests/tflm/tflite_export/conftest.py renamed to tests/test_transformer/test_tflm/test_export/conftest.py

Lines changed: 7 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,14 @@
11
import numpy as np
22
from pytest import fixture
33

4-
import tensorflow as tf
5-
from utensor_cgen.ir import TensorInfo, OperationInfo, uTensorGraph
6-
from utensor_cgen.ir.converter import (AttrValueConverter, DataTypeConverter,
7-
GenericTensorConverterMixin)
8-
from utensor_cgen.utils import prune_graph, topologic_order_graph
9-
from utensor_cgen.backend.operators import OperatorFactory, _Operator
10-
from utensor_cgen.matcher import OpEqualityDelegate, _morphism
4+
from utensor_cgen.backend.utensor.code_generator.legacy._operators import (
5+
OperatorFactory, _Operator, uTensorOpEqualityDelegate)
6+
from utensor_cgen.ir import OperationInfo, TensorInfo, uTensorGraph
7+
from utensor_cgen.ir.converter import AttrValueConverter, DataTypeConverter
118

129

1310
@OperatorFactory.register
14-
@OpEqualityDelegate.is_associative(
11+
@uTensorOpEqualityDelegate.is_associative(
1512
permutations=((0, 1), (1, 0))
1613
)
1714
class _TFLM_AddOperator(_Operator):
@@ -51,7 +48,7 @@ def build_op_info(cls, ugraph, name, tensor_x, tensor_y, **kwargs):
5148
)
5249
},
5350
ugraph=ugraph,
54-
backend=kwargs.get('backend', 'TFLM')
51+
lib_name=kwargs.get('lib_name', 'tflite')
5552
)
5653

5754

@@ -93,7 +90,7 @@ def build_op_info(cls, ugraph, name, tensor_x, tensor_w, tensor_b, **kwargs):
9390
)
9491
},
9592
ugraph=ugraph,
96-
backend=kwargs.get('backend', 'TFLM')
93+
lib_name=kwargs.get('lib_name', 'tflite')
9794
)
9895

9996
@fixture(name='hybrid_quant_output')

0 commit comments

Comments
 (0)