|
1 | 1 | import numpy as np
|
2 | 2 | from pytest import fixture
|
3 | 3 |
|
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 |
11 | 8 |
|
12 | 9 |
|
13 | 10 | @OperatorFactory.register
|
14 |
| -@OpEqualityDelegate.is_associative( |
| 11 | +@uTensorOpEqualityDelegate.is_associative( |
15 | 12 | permutations=((0, 1), (1, 0))
|
16 | 13 | )
|
17 | 14 | class _TFLM_AddOperator(_Operator):
|
@@ -51,7 +48,7 @@ def build_op_info(cls, ugraph, name, tensor_x, tensor_y, **kwargs):
|
51 | 48 | )
|
52 | 49 | },
|
53 | 50 | ugraph=ugraph,
|
54 |
| - backend=kwargs.get('backend', 'TFLM') |
| 51 | + lib_name=kwargs.get('lib_name', 'tflite') |
55 | 52 | )
|
56 | 53 |
|
57 | 54 |
|
@@ -93,7 +90,7 @@ def build_op_info(cls, ugraph, name, tensor_x, tensor_w, tensor_b, **kwargs):
|
93 | 90 | )
|
94 | 91 | },
|
95 | 92 | ugraph=ugraph,
|
96 |
| - backend=kwargs.get('backend', 'TFLM') |
| 93 | + lib_name=kwargs.get('lib_name', 'tflite') |
97 | 94 | )
|
98 | 95 |
|
99 | 96 | @fixture(name='hybrid_quant_output')
|
|
0 commit comments