Skip to content

Commit 9e72839

Browse files
author
yunjing.lh
committed
improve accuracy
1 parent e1a57bc commit 9e72839

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/relay/qnn/op/add.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -74,12 +74,12 @@ Expr QnnAddCanonicalize(const Attrs& attrs, const Array<Expr>& new_args,
7474
auto requantized_lhs = Requantize(args.lhs, input_type.shape,
7575
real_lhs_scale, args.lhs_zero_point,
7676
itmd_out_scale, itmd_zero_point,
77-
DataType::Int(32));
77+
DataType::Int(32), rounding);
7878

7979
auto requantized_rhs = Requantize(args.rhs, input_type.shape,
8080
real_rhs_scale, args.rhs_zero_point,
8181
itmd_out_scale, itmd_zero_point,
82-
DataType::Int(32));
82+
DataType::Int(32), rounding);
8383

8484
output = Add(requantized_lhs, requantized_rhs);
8585
output = Requantize(output, input_type.shape,

0 commit comments

Comments
 (0)