diff options
| author | William Isaksson <william.isaksson@arm.com> | 2023-08-10 12:22:44 +0000 |
|---|---|---|
| committer | William Isaksson <william.isaksson@arm.com> | 2023-08-10 13:23:38 +0100 |
| commit | 75d3402204145731c2ebe0131ee47d966fd95562 (patch) | |
| tree | 7c9388454b7e257251fff8a1c1fef9af49de2d89 | |
| parent | a71efe00bcbac0e601a0e3812bba89da452f2aff (diff) | |
| download | ethos-u-vela-75d3402204145731c2ebe0131ee47d966fd95562.tar.gz | |
MLBEDSW-7832: test_tflite_model_semantic converting array to scalar
- now only converts array directly if ndim==0
Signed-off-by: William Isaksson <william.isaksson@arm.com>
Change-Id: Id23e419bc7dd717f9694013180d4609819fd2f56
| -rw-r--r-- | ethosu/vela/tflite_model_semantic.py | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/ethosu/vela/tflite_model_semantic.py b/ethosu/vela/tflite_model_semantic.py index ea7ef4a..d2e0ba5 100644 --- a/ethosu/vela/tflite_model_semantic.py +++ b/ethosu/vela/tflite_model_semantic.py @@ -472,7 +472,10 @@ class TFLiteSemantic: input_tens = op.inputs[1] dims = len(input_tens.shape) # handle axis being a scalar or 1-D array - axis = int(axis_tens.values) if len(axis_tens.values.shape) == 0 else int(axis_tens.values[0]) + if axis_tens.values.ndim == 0: + axis = int(axis_tens.values) + else: + axis = int(axis_tens.values[0]) axis += dims if axis < 0 else 0 valid = 0 <= axis < dims return valid, f"Op has ifm_dimensions={dims} and axis value is: {axis}" @@ -485,7 +488,10 @@ class TFLiteSemantic: input_tens = op.inputs[1] dims = len(input_tens.shape) # handle axis being a scalar or 1-D array - axis = int(axis_tens.values) if len(axis_tens.values.shape) == 0 else int(axis_tens.values[0]) + if axis_tens.values.ndim == 0: + axis = int(axis_tens.values) + else: + axis = int(axis_tens.values[0]) axis += dims if axis < 0 else 0 valid = input_tens.shape[axis] % num_splits == 0 return valid, f"Op has ifm shape={input_tens.shape} axis={axis} num_splits={num_splits}" |