Skip to content

Commit 9ea1837

Browse files
committed
Fix incorrect merge of maxpool2d
Signed-off-by: Erik Lundell <[email protected]> Change-Id: If415a5ee3185c3401069652906a0633491bca1b0
1 parent 8455a71 commit 9ea1837

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

backends/arm/operators/op_max_pool2d.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ def define_node(
5757
output_zp = 0
5858

5959
if is_quant_node:
60-
input_zp = get_quant_arg_upstream(torch.fx.Node, node.all_input_nodes[0]).zp
60+
input_zp = get_quant_arg_upstream(node.all_input_nodes[0]).zp
6161
output_zp = get_quant_arg_downstream(list(node.users)[0]).zp
6262

6363
attr = ts.TosaSerializerAttribute()

0 commit comments

Comments
 (0)