diff options
author | Johan Gunnarsson <johan.gunnarsson@arm.com> | 2023-08-04 17:16:29 +0200 |
---|---|---|
committer | Rickard Bolin <rickard.bolin@arm.com> | 2023-08-09 11:21:24 +0000 |
commit | 81b765df02d7c7cae5f1084eec998824b68c00ab (patch) | |
tree | b7dc181f531fccc4106787c68fefe07e7b82b4c7 /ethosu/vela/test/test_tflite_model_semantic.py | |
parent | cd03504cfc29767d33d37b5c587116ab90752d74 (diff) | |
download | ethos-u-vela-81b765df02d7c7cae5f1084eec998824b68c00ab.tar.gz |
MLBEDSW-7626: Add constraint for PAD op paddings
PAD input tensor shape plus paddings must equal output tensor shape.
Change-Id: Icc5dea9bf6a8f6e1c8402f4d9af4d9796e8ef1aa
Signed-off-by: Johan Gunnarsson <johan.gunnarsson@arm.com>
Diffstat (limited to 'ethosu/vela/test/test_tflite_model_semantic.py')
-rw-r--r-- | ethosu/vela/test/test_tflite_model_semantic.py | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/ethosu/vela/test/test_tflite_model_semantic.py b/ethosu/vela/test/test_tflite_model_semantic.py index e7fd3073..7ca1bbda 100644 --- a/ethosu/vela/test/test_tflite_model_semantic.py +++ b/ethosu/vela/test/test_tflite_model_semantic.py @@ -356,6 +356,18 @@ def test_constraint_pad_input_count(): assert not semantic_checker.is_operator_semantic_valid(op) +def test_constraint_pad_output_shape(): + # Incorrect output tensor shape + op = create_pad_op( + in_shape=[1, 1, 1, 1], + out_shape=[1, 3, 3, 1], + padding=[[0, 0], [1, 1], [1, 1], [0, 0]], + ) + assert semantic_checker.is_operator_semantic_valid(op) + op.outputs[0].shape = [1, 1, 1, 1] + assert not semantic_checker.is_operator_semantic_valid(op) + + def create_strided_slice(): # Creates a valid strided slice operator with some valid inputs/outputs op = create_strided_slice_op([1, 10, 10, 10], [1, 5, 5, 10], [127, 2, 2, 0], [0, 7, -3, 0]) |