diff options
author | Rickard Bolin <rickard.bolin@arm.com> | 2023-04-20 15:19:57 +0000 |
---|---|---|
committer | Rickard Bolin <rickard.bolin@arm.com> | 2023-04-27 13:40:21 +0000 |
commit | bb46af7c97ca665041a848a3137a3b7855a4f82a (patch) | |
tree | 9acd091ee91703ad0b6af34c46f5b403f47f1b62 /ethosu/vela/test/test_tflite_model_semantic.py | |
parent | a68b82a055167e12761c3d2a309e8230011dc401 (diff) | |
download | ethos-u-vela-bb46af7c97ca665041a848a3137a3b7855a4f82a.tar.gz |
MLBEDSW-7530: Enable int16 input precision for mean operator
Signed-off-by: Rickard Bolin <rickard.bolin@arm.com>
Change-Id: Iaeb8f2cea0d3b576a6b138e64a882c701ac88ccb
Diffstat (limited to 'ethosu/vela/test/test_tflite_model_semantic.py')
-rw-r--r-- | ethosu/vela/test/test_tflite_model_semantic.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/test/test_tflite_model_semantic.py b/ethosu/vela/test/test_tflite_model_semantic.py index d4c92553..ebfdbf3f 100644 --- a/ethosu/vela/test/test_tflite_model_semantic.py +++ b/ethosu/vela/test/test_tflite_model_semantic.py @@ -502,7 +502,7 @@ def test_mean_dtype(): assert semantic_checker.is_operator_semantic_valid(op) op.ifm.dtype = DataType.int16 op.ofm.dtype = DataType.int16 - assert not semantic_checker.is_operator_semantic_valid(op) + assert semantic_checker.is_operator_semantic_valid(op) def test_mean_axis(): |