aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/test
diff options
context:
space:
mode:
authorJohan Alfvén <johan.alfven@arm.com>2022-09-06 20:33:22 +0200
committerFredrik Svedberg <fredrik.svedberg@arm.com>2022-09-12 08:02:11 +0000
commit059166304f9ef47f0b916c1325700ed826f25581 (patch)
tree49200e8bc2014e2cdbae52ba49a9e9b5b2ed776f /ethosu/vela/test
parentd6efcd3cdcd8295fbbe7dd47a7074be39eaf03e4 (diff)
downloadethos-u-vela-059166304f9ef47f0b916c1325700ed826f25581.tar.gz
MLBEDSW-6909: Use int32 acc for the Mean op
Changed acc type from int16 to int32. This will solve saturation problems and the constraint added in commit "MLBEDSW-5029: Output diff for Mean op" can be removed. Signed-off-by: Johan Alfven <johan.alfven@arm.com> Change-Id: I05ec8835b43313b1a264d61a2b147fa62da123fe
Diffstat (limited to 'ethosu/vela/test')
-rw-r--r--ethosu/vela/test/test_tflite_supported_operators.py16
1 files changed, 0 insertions, 16 deletions
diff --git a/ethosu/vela/test/test_tflite_supported_operators.py b/ethosu/vela/test/test_tflite_supported_operators.py
index cc8b3d2c..89c27997 100644
--- a/ethosu/vela/test/test_tflite_supported_operators.py
+++ b/ethosu/vela/test/test_tflite_supported_operators.py
@@ -623,22 +623,6 @@ def test_mean_hw_product_int8():
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)
- # Create OP that will not saturate the accumulator
- op = create_mean([1, 5, 14, 16], [1, 1, 1, 16], [1, 2], DataType.int8, {"keep_dims": True})
- op.ifm.quantization.scale_f32 = 2.0
- op.ifm.quantization.zero_point = 95
- op.ofm.quantization.scale_f32 = 1.0
- op.ofm.quantization.zero_point = 95
- assert support.is_operator_supported(op)
-
- # Create OP that can saturate the accumulator
- op = create_mean([1, 6, 14, 16], [1, 1, 1, 16], [1, 2], DataType.int8, {"keep_dims": True})
- op.ifm.quantization.scale_f32 = 2.0
- op.ifm.quantization.zero_point = 95
- op.ofm.quantization.scale_f32 = 1.0
- op.ofm.quantization.zero_point = 95
- 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})