From bb46af7c97ca665041a848a3137a3b7855a4f82a Mon Sep 17 00:00:00 2001 From: Rickard Bolin Date: Thu, 20 Apr 2023 15:19:57 +0000 Subject: MLBEDSW-7530: Enable int16 input precision for mean operator Signed-off-by: Rickard Bolin Change-Id: Iaeb8f2cea0d3b576a6b138e64a882c701ac88ccb --- ethosu/vela/test/test_tflite_model_semantic.py | 2 +- ethosu/vela/tflite_model_semantic.py | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) (limited to 'ethosu') diff --git a/ethosu/vela/test/test_tflite_model_semantic.py b/ethosu/vela/test/test_tflite_model_semantic.py index d4c92553..ebfdbf3f 100644 --- a/ethosu/vela/test/test_tflite_model_semantic.py +++ b/ethosu/vela/test/test_tflite_model_semantic.py @@ -502,7 +502,7 @@ def test_mean_dtype(): assert semantic_checker.is_operator_semantic_valid(op) op.ifm.dtype = DataType.int16 op.ofm.dtype = DataType.int16 - assert not semantic_checker.is_operator_semantic_valid(op) + assert semantic_checker.is_operator_semantic_valid(op) def test_mean_axis(): 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) -- cgit v1.2.1