diff options
Diffstat (limited to 'ethosu')
-rw-r--r-- | ethosu/vela/test/test_tflite_model_semantic.py | 2 | ||||
-rw-r--r-- | ethosu/vela/tflite_model_semantic.py | 1 |
2 files changed, 1 insertions, 2 deletions
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) |