aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/test/test_tflite_supported_operators.py
diff options
context:
space:
mode:
authorJohan Alfven <johan.alfven@arm.com>2023-04-13 18:54:47 +0200
committerJohan Alfven <johan.alfven@arm.com>2023-04-19 12:26:19 +0200
commit7b3008a905d2a5122e21f945db7d2a2132473c53 (patch)
tree824b7a1b6b4ed4e5e382901e60331ff8e2159d5f /ethosu/vela/test/test_tflite_supported_operators.py
parent0ac0804e76e098695ee2b8a9e24e2f0a1efc324f (diff)
downloadethos-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.py7
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)