diff options
-rw-r--r-- | SUPPORTED_OPS.md | 3 | ||||
-rw-r--r-- | ethosu/vela/test/test_tflite_model_semantic.py | 2 | ||||
-rw-r--r-- | ethosu/vela/tflite_model_semantic.py | 1 |
3 files changed, 2 insertions, 4 deletions
diff --git a/SUPPORTED_OPS.md b/SUPPORTED_OPS.md index a870c5aa..ab9b0096 100644 --- a/SUPPORTED_OPS.md +++ b/SUPPORTED_OPS.md @@ -1,7 +1,7 @@ # Supported Ops This file was automatically generated by Vela using the `--supported-ops-report` parameter. -Vela version: `3.7.1.dev16+g1f9a4df.d20230417` +Vela version: `3.7.1.dev17+g7b3008a.d20230420` This file complies with [**Gitiles Markdown syntax**](https://github.com/google/gitiles/blob/master/Documentation/markdown.md) @@ -218,7 +218,6 @@ This is a list of constraints that the MAX_POOL_2D operator must satisfy in orde This is a list of constraints that the MEAN operator must satisfy in order to be scheduled on the NPU. -- IFM must be int8 or uint8 - Input tensor must be at least 2D - Axis indices must correspond to height and width axes - Product of height and width must be no greater than 4096 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) |