File tree Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change 17
17
18
18
from executorch .backends .arm .tosa_quant_utils import (
19
19
dequantize_value ,
20
- get_quant_node_args ,
21
20
QuantArgs ,
22
21
quantize_value ,
22
+ search_quant_arg_downstream ,
23
+ search_quant_arg_upstream ,
23
24
)
24
25
from serializer .tosa_serializer import TosaOp
25
26
from torch .fx import Node
@@ -49,9 +50,9 @@ def define_node(
49
50
50
51
# Create attribute for 8 bit table lookup.
51
52
input_node = node .all_input_nodes [0 ]
52
- in_quantargs = get_quant_node_args (input_node )
53
+ in_quantargs = search_quant_arg_upstream (input_node )
53
54
output_node = list (node .users )[0 ]
54
- out_quantargs = get_quant_node_args (output_node )
55
+ out_quantargs = search_quant_arg_downstream (output_node )
55
56
56
57
table = tanh_table_8bit (in_quantargs , out_quantargs )
57
58
table_attr = ts .TosaSerializerAttribute ()
You can’t perform that action at this time.
0 commit comments