From cc8569fcd243d7c96a20e0f531f6f97a90df83f7 Mon Sep 17 00:00:00 2001 From: Fredrik Svedberg Date: Mon, 1 Nov 2021 14:25:29 +0100 Subject: MLBEDSW-5209 Vela: output diff depthwise with non-zero zero points Fixed by adjusting zero points for ops with int8 IFM and asymmetric weights since the reference does not support asymmetric weights for int8 IFM and ignores the zero points. Signed-off-by: Fredrik Svedberg Change-Id: I2a206a01a471a53aa864a6a3616aa23d2a5a23c8 --- ethosu/vela/tflite_writer.py | 2 ++ 1 file changed, 2 insertions(+) (limited to 'ethosu/vela/tflite_writer.py') diff --git a/ethosu/vela/tflite_writer.py b/ethosu/vela/tflite_writer.py index e6dd85b5..d134c07c 100644 --- a/ethosu/vela/tflite_writer.py +++ b/ethosu/vela/tflite_writer.py @@ -236,6 +236,8 @@ class TFLiteSerialiser: QuantizationParameters.QuantizationParametersAddScale(builder, scale) if zero_point is not None: QuantizationParameters.QuantizationParametersAddZeroPoint(builder, zero_point) + if quant.quant_dim is not None: + QuantizationParameters.QuantizationParametersAddQuantizedDimension(builder, quant.quant_dim) qp = QuantizationParameters.QuantizationParametersEnd(builder) return qp -- cgit v1.2.1