diff options
author | Jacob Bohlin <jacob.bohlin@arm.com> | 2020-08-24 11:42:14 +0200 |
---|---|---|
committer | Jacob Bohlin <jacob.bohlin@arm.com> | 2020-08-26 08:57:59 +0200 |
commit | d47cc2777f3a3f371958c30a3c1880f692b0b0a2 (patch) | |
tree | 47dd9c57f13719dbbc6bd886042f9c8795353c98 /ethosu/vela/tensor.py | |
parent | 814cfbb8124ba0b3828db2bb12d9342ae9c39f19 (diff) | |
download | ethos-u-vela-d47cc2777f3a3f371958c30a3c1880f692b0b0a2.tar.gz |
MLBEDSW-2847: Fix for TransposeConv crash and u8 output diff
Signed-off-by: Jacob Bohlin <jacob.bohlin@arm.com>
Change-Id: I2cb3f6639e4bb8a984fa3647ee7b4678ed6f5890
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 f0e7ea44..d4f6a409 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 = np.frombuffer(const_tensor.values.tobytes(), dtype=np.uint8) + const_tensor.quant_values = const_tensor.values # Operator const_op = Operation("Const", name) const_op.set_output_tensor(const_tensor) |