commit 9b3791817c32529dfeddae57c29c2abe19311fc4
author Dwight Lidman <dwight.lidman@arm.com> Mon Mar 15 19:06:10 2021 +0100
committer patrik.gustavsson <patrik.gustavsson@arm.com> Tue Mar 16 16:29:01 2021 +0000
tree 866c769bb8627b686f7d25ddce3413380c34cbdb
parent c822d62ba27b874a130e9d8d434c12b419d10d62

MLBEDSW-4215: Add support for MEAN to match QuantizedMeanOrSum implementation

This commit adds support for emulating the behavior
of the QuantizedMeanOrSum implementation of MEAN in
TensorFlow Lite.

Signed-off-by: Dwight Lidman <dwight.lidman@arm.com>
Change-Id: Ifd24e0e678e2f85cd66ab82deeaaf010d5351b1e

ethosu/vela/tensor.py

diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py
index 97885d0..e915363 100644
--- a/ethosu/vela/tensor.py
+++ b/ethosu/vela/tensor.py
@@ -300,13 +300,16 @@
     value_dtype: np.dtype = None,
     purpose: TensorPurpose = TensorPurpose.Unknown,
     quantization: QuantizationParameters = None,
+    quant_value_dtype: np.dtype = None,
 ):
     # Tensor
     const_tensor = Tensor(shape, dtype, name + "_0")
     const_tensor.purpose = purpose
     const_tensor.quantization = quantization
     const_tensor.values = np.array(values, dtype=value_dtype)
-    const_tensor.quant_values = np.frombuffer(const_tensor.values.tobytes(), dtype=np.uint8)
+    const_tensor.quant_values = np.frombuffer(
+        const_tensor.values.tobytes(), dtype=np.uint8 if not quant_value_dtype else quant_value_dtype
+    )
     # Operator
     const_op = Operation(Op.Const, name)
     const_op.set_output_tensor(const_tensor)