diff options
| author | Rickard Bolin <rickard.bolin@arm.com> | 2023-04-20 15:19:57 +0000 |
|---|---|---|
| committer | Rickard Bolin <rickard.bolin@arm.com> | 2023-04-27 13:40:21 +0000 |
| commit | bb46af7c97ca665041a848a3137a3b7855a4f82a (patch) | |
| tree | 9acd091ee91703ad0b6af34c46f5b403f47f1b62 /ethosu/vela/tflite_model_semantic.py | |
| parent | a68b82a055167e12761c3d2a309e8230011dc401 (diff) | |
| download | ethos-u-vela-bb46af7c97ca665041a848a3137a3b7855a4f82a.tar.gz | |
MLBEDSW-7530: Enable int16 input precision for mean operator
Signed-off-by: Rickard Bolin <rickard.bolin@arm.com>
Change-Id: Iaeb8f2cea0d3b576a6b138e64a882c701ac88ccb
Diffstat (limited to 'ethosu/vela/tflite_model_semantic.py')
| -rw-r--r-- | ethosu/vela/tflite_model_semantic.py | 1 |
1 files changed, 0 insertions, 1 deletions
diff --git a/ethosu/vela/tflite_model_semantic.py b/ethosu/vela/tflite_model_semantic.py index 66770487..7537d7da 100644 --- a/ethosu/vela/tflite_model_semantic.py +++ b/ethosu/vela/tflite_model_semantic.py @@ -186,7 +186,6 @@ class TFLiteSemantic: self.specific_constraints[Op.HardSwish].append(TFLiteSemantic.constraint_matching_in_out_types) # Mean specific checks: - self.specific_constraints[Op.Mean].append(TFLiteSemantic.constraint_input_8bit) self.specific_constraints[Op.Mean].append(TFLiteSemantic.constraint_mean_input_dims) self.specific_constraints[Op.Mean].append(TFLiteSemantic.constraint_mean_axis) |