From 557ac911defa85e639b7379dbddb3fa47c48b808 Mon Sep 17 00:00:00 2001
From: peterjc123 <peter_jiachen@163.com>
Date: Wed, 25 Sep 2024 07:50:57 +0000
Subject: [PATCH] [converter] same q_params for neg in mul_scalar
---
tinynn/converter/operators/torch/quantized.py | 9 +++++++++
1 file changed, 9 insertions(+)
diff --git a/tinynn/converter/operators/torch/quantized.py b/tinynn/converter/operators/torch/quantized.py
index b2459b94..0033f761 100644
--- a/tinynn/converter/operators/torch/quantized.py
+++ b/tinynn/converter/operators/torch/quantized.py
@@ -18,7 +18,16 @@ class QuantizedMulScalarOperator(QuantizedMulScalarSchema):
def parse(self, node, attrs, args, graph_converter):
super().parse(node, attrs, args, graph_converter)
+ is_neg = False
+ if self.input_tensors[1] in (-1, -1.0):
+ self.input_tensors[1] *= -1
+ is_neg = True
+
self.run(node)
+
+ if is_neg:
+ self.input_tensors[1] *= -1
+
QuantizedMulOperator.parse_common(self, node, attrs, args, graph_converter)