commit a41cd4de2af1e43b76a2a33d78eeb2d90a88b757
author Jacob Bohlin <jacob.bohlin@arm.com> Wed Aug 26 18:21:28 2020 +0200
committer tim.hall <tim.hall@arm.com> Thu Aug 27 14:23:03 2020 +0000
tree e7f81ab5fbddca95928e2111fea2f6cff9b75679
parent 2abd3dd75bd3d20e1a3aeaf12362f9872b40fa0a

Small fix for Softmax regression

Signed-off-by: Jacob Bohlin <jacob.bohlin@arm.com>
Change-Id: I287c24725126c169afec779b921e43c3ab26f739

ethosu/vela/tensor.py

diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py
index 3ad9b25..83dc61a 100644
--- a/ethosu/vela/tensor.py
+++ b/ethosu/vela/tensor.py
@@ -232,7 +232,7 @@
     const_tensor.purpose = purpose
     const_tensor.quantization = quantization
     const_tensor.values = np.array(values, dtype=value_dtype)
-    const_tensor.quant_values = const_tensor.values
+    const_tensor.quant_values = np.frombuffer(const_tensor.values.tobytes(), dtype=np.uint8)
     # Operator
     const_op = Operation("Const", name)
     const_op.set_output_tensor(const_tensor)