diff options
| author | Jacob Bohlin <jacob.bohlin@arm.com> | 2020-08-26 18:21:28 +0200 |
|---|---|---|
| committer | tim.hall <tim.hall@arm.com> | 2020-08-27 14:23:03 +0000 |
| commit | a41cd4de2af1e43b76a2a33d78eeb2d90a88b757 (patch) | |
| tree | e7f81ab5fbddca95928e2111fea2f6cff9b75679 /ethosu/vela/tensor.py | |
| parent | 2abd3dd75bd3d20e1a3aeaf12362f9872b40fa0a (diff) | |
| download | ethos-u-vela-a41cd4de2af1e43b76a2a33d78eeb2d90a88b757.tar.gz | |
Small fix for Softmax regression
Signed-off-by: Jacob Bohlin <jacob.bohlin@arm.com>
Change-Id: I287c24725126c169afec779b921e43c3ab26f739
Diffstat (limited to 'ethosu/vela/tensor.py')
| -rw-r--r-- | ethosu/vela/tensor.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py index 3ad9b253..83dc61a3 100644 --- a/ethosu/vela/tensor.py +++ b/ethosu/vela/tensor.py @@ -232,7 +232,7 @@ def create_const_tensor(name, shape, dtype, values, value_dtype=None, purpose=Te 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) |