diff options
| author | Tim Hall <tim.hall@arm.com> | 2021-11-18 22:05:17 +0000 |
|---|---|---|
| committer | tim.hall <tim.hall@arm.com> | 2021-11-25 14:36:26 +0000 |
| commit | 3584a9cfdf0bcf0e75d38b78ec39e5b083947e19 (patch) | |
| tree | 00a08343ce1c3a95b9ec8eb3d4414d0d6e58daf7 /ethosu/vela/tflite_model_semantic.py | |
| parent | f88771bcdca9bbd7582ba3183d698f5d8f83b7ac (diff) | |
| download | ethos-u-vela-3584a9cfdf0bcf0e75d38b78ec39e5b083947e19.tar.gz | |
MLBEDSW-3602: Output mismatch on some mobilenet_v1 int8 and int16
- The failing tests contain operations with dynamic tensors which
are not supported and therefore they should be placed on the CPU.
However, a bug in the removal of RESHAPEs which contain a dynamic
shape prevented this happening.
- This change adds a check to make sure that RESHAPE ops with a
dynamic shape tensor are not removed and instead are placed on the
CPU.
Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: I2d7481f7f80f99a0f01df100d956933777e6875a
Diffstat (limited to 'ethosu/vela/tflite_model_semantic.py')
| -rw-r--r-- | ethosu/vela/tflite_model_semantic.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/tflite_model_semantic.py b/ethosu/vela/tflite_model_semantic.py index 51d1f072..b783bb74 100644 --- a/ethosu/vela/tflite_model_semantic.py +++ b/ethosu/vela/tflite_model_semantic.py @@ -160,7 +160,7 @@ class TFLiteSemantic: valid, extra = constraint(op) if not valid: print( - f"Warning: unsupported TensorFlow Lite semantics for {ext_type} '{op.name}'. Placing on CPU instead" + f"Warning: Unsupported TensorFlow Lite semantics for {ext_type} '{op.name}'. Placing on CPU instead" ) print(f" - {constraint.__doc__}") if extra: |