|
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