commit 8ae292931aa9b4677092f51a46c0fb07a8807f89
author Tim Hall <tim.hall@arm.com> Wed Jul 28 16:52:03 2021 +0100
committer Tim Hall <tim.hall@arm.com> Wed Jul 28 16:52:03 2021 +0100
tree e9db07830d0e3343319aff9517e4f5a7d0e992f9
parent 45e653dbd81633b8d78215b16a9b2205e39dd8e2

MLBEDSW-4974: create_const_tensor invalid argument

 - Fix bug with MEAN ops calling create_const_tensor using the
quant_value_dtype keyword argument.

Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: I8cff542ae840fb110ea97c0cc86bb761d5a884d3

ethosu/vela/tflite_graph_optimiser.py

diff --git a/ethosu/vela/tflite_graph_optimiser.py b/ethosu/vela/tflite_graph_optimiser.py
index 68b4e8e..2959803 100644
--- a/ethosu/vela/tflite_graph_optimiser.py
+++ b/ethosu/vela/tflite_graph_optimiser.py
@@ -1515,13 +1515,7 @@
             bias_shape = [shape[-1]]
             op.set_input_tensor(
                 create_const_tensor(
-                    "bias",
-                    bias_shape,
-                    inp.dtype,
-                    np.ones(bias_shape) * bias,
-                    value_dtype=np.int32,
-                    quant_value_dtype=np.int32,
-                    quantization=None,
+                    "bias", bias_shape, inp.dtype, np.ones(bias_shape) * bias, value_dtype=np.int32, quantization=None,
                 ),
                 2,
             )