diff options
author | Dwight Lidman <dwight.lidman@arm.com> | 2021-03-15 19:06:10 +0100 |
---|---|---|
committer | patrik.gustavsson <patrik.gustavsson@arm.com> | 2021-03-16 16:29:01 +0000 |
commit | 9b3791817c32529dfeddae57c29c2abe19311fc4 (patch) | |
tree | 866c769bb8627b686f7d25ddce3413380c34cbdb /ethosu/vela/test | |
parent | c822d62ba27b874a130e9d8d434c12b419d10d62 (diff) | |
download | ethos-u-vela-9b3791817c32529dfeddae57c29c2abe19311fc4.tar.gz |
MLBEDSW-4215: Add support for MEAN to match QuantizedMeanOrSum implementation
This commit adds support for emulating the behavior
of the QuantizedMeanOrSum implementation of MEAN in
TensorFlow Lite.
Signed-off-by: Dwight Lidman <dwight.lidman@arm.com>
Change-Id: Ifd24e0e678e2f85cd66ab82deeaaf010d5351b1e
Diffstat (limited to 'ethosu/vela/test')
-rw-r--r-- | ethosu/vela/test/test_supported_operators.py | 7 |
1 files changed, 0 insertions, 7 deletions
diff --git a/ethosu/vela/test/test_supported_operators.py b/ethosu/vela/test/test_supported_operators.py index 34ddb900..aad2849a 100644 --- a/ethosu/vela/test/test_supported_operators.py +++ b/ethosu/vela/test/test_supported_operators.py @@ -858,13 +858,6 @@ def test_mean_dtype(): assert not support.is_operator_supported(op) -def test_mean_properties(): - op = create_mean([1, 6, 6, 256], [1, 1, 256], [1, 2], DataType.uint8, {}) - assert support.is_operator_supported(op) - op.ifm.quantization.zero_point = 55 - assert not support.is_operator_supported(op) - - def test_mean_axis(): op = create_mean([1, 6, 6, 16], [1, 1, 1, 16], [1], DataType.int8, {"keep_dims": True}) assert not support.is_operator_supported(op) |