diff options
author | Johan Alfven <johan.alfven@arm.com> | 2023-04-13 18:54:47 +0200 |
---|---|---|
committer | Johan Alfven <johan.alfven@arm.com> | 2023-04-19 12:26:19 +0200 |
commit | 7b3008a905d2a5122e21f945db7d2a2132473c53 (patch) | |
tree | 824b7a1b6b4ed4e5e382901e60331ff8e2159d5f /ethosu/vela/test/test_tflite_supported_operators.py | |
parent | 0ac0804e76e098695ee2b8a9e24e2f0a1efc324f (diff) | |
download | ethos-u-vela-7b3008a905d2a5122e21f945db7d2a2132473c53.tar.gz |
MLBEDSW-7487: Updated implementation for the Mean op
- Latest reference has changed implementation for the Mean op
and now only contain one variant.
- Updated Vela implementation to match reference. The full sum
is first calculated and then divided by the numbers of elements.
- Removed the avg pool variant and test case.
- Updated SUPPORTED_OPS.md
Change-Id: I4275e36e3697fa837f119f2cefd7c0ff94231605
Signed-off-by: Johan Alfven <johan.alfven@arm.com>
Diffstat (limited to 'ethosu/vela/test/test_tflite_supported_operators.py')
-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 04f10e9a..74dd3bf2 100644 --- a/ethosu/vela/test/test_tflite_supported_operators.py +++ b/ethosu/vela/test/test_tflite_supported_operators.py @@ -618,13 +618,6 @@ def test_mean_hw_product(): 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) - op = create_mean([1, 200, 200, 16], [1, 1, 1, 16], [1, 2], DataType.int8, {"keep_dims": True}) - assert not support.is_operator_supported(op) - - def test_lstm_support(): # Test valid configuration op = testutil.create_lstm_op(3, 12, 24, 20, DataType.int8) |