diff options
| author | Johan Alfvén <johan.alfven@arm.com> | 2022-10-27 16:30:01 +0200 |
|---|---|---|
| committer | Tim Hall <tim.hall@arm.com> | 2022-11-03 16:35:22 +0000 |
| commit | 9d51ec41855a8be21bd0708c882d121e5bb5afcc (patch) | |
| tree | 69d5fa0d2da78c9df2ccc0999756b4a9e8368901 /ethosu/vela/test | |
| parent | 92cd33b7adbb799b6593d49f1d29c13a85933e55 (diff) | |
| download | ethos-u-vela-9d51ec41855a8be21bd0708c882d121e5bb5afcc.tar.gz | |
MLBEDSW-7074: Updated reference kernel for the MEAN op
The reference kernel for the MEAN operator has changed.
As a result, the mean implementation can be simplified
and the constraint for mean int8 can be removed.
Signed-off-by: Johan Alfven <johan.alfven@arm.com>
Change-Id: I318e9b495eefea99e7ac4aea4b8c436c83753405
Diffstat (limited to 'ethosu/vela/test')
| -rw-r--r-- | ethosu/vela/test/test_tflite_supported_operators.py | 7 |
1 files changed, 0 insertions, 7 deletions
diff --git a/ethosu/vela/test/test_tflite_supported_operators.py b/ethosu/vela/test/test_tflite_supported_operators.py index 790f0018..4410938e 100644 --- a/ethosu/vela/test/test_tflite_supported_operators.py +++ b/ethosu/vela/test/test_tflite_supported_operators.py @@ -608,13 +608,6 @@ def test_mean_hw_product(): assert not support.is_operator_supported(op) -def test_mean_hw_product_int8(): - op = create_mean([1, 16, 16, 16], [1, 1, 1, 16], [1, 2], DataType.int8, {"keep_dims": True}) - assert support.is_operator_supported(op) - op = create_mean([1, 16, 17, 16], [1, 1, 1, 16], [1, 2], DataType.int8, {"keep_dims": True}) - assert not support.is_operator_supported(op) - - def test_mean_hw_product_avgpool(): op = create_mean([1, 200, 200, 16], [1, 16], [1, 2], DataType.uint8, {"keep_dims": False}) assert support.is_operator_supported(op) |