aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorRickard Bolin <rickard.bolin@arm.com>2023-04-20 15:19:57 +0000
committerRickard Bolin <rickard.bolin@arm.com>2023-04-27 13:40:21 +0000
commitbb46af7c97ca665041a848a3137a3b7855a4f82a (patch)
tree9acd091ee91703ad0b6af34c46f5b403f47f1b62
parenta68b82a055167e12761c3d2a309e8230011dc401 (diff)
downloadethos-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
-rw-r--r--SUPPORTED_OPS.md3
-rw-r--r--ethosu/vela/test/test_tflite_model_semantic.py2
-rw-r--r--ethosu/vela/tflite_model_semantic.py1
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)