diff options
author | Suraj Sudhir <suraj.sudhir@arm.com> | 2024-04-16 16:14:36 -0700 |
---|---|---|
committer | Suraj Sudhir <suraj.sudhir@arm.com> | 2024-04-18 21:46:54 -0700 |
commit | b5fcfc0cfeffeb3fe9f05c32dc678ab09b2cee31 (patch) | |
tree | a4cf45d48fcc17db3d41d8663ffe8c05ef41624c /verif/generator/tosa_error_if.py | |
parent | 7296e16351554a2c80daab18b2d69b44c67b807a (diff) | |
download | reference_model-b5fcfc0cfeffeb3fe9f05c32dc678ab09b2cee31.tar.gz |
[reference_model] Remove output_shape from transpose_conv2d
Signed-off-by: Suraj Sudhir <suraj.sudhir@arm.com>
Change-Id: Ib2b95e73b226d64c4db5ad1ed22c123e04d7e6f9
Diffstat (limited to 'verif/generator/tosa_error_if.py')
-rw-r--r-- | verif/generator/tosa_error_if.py | 6 |
1 files changed, 1 insertions, 5 deletions
diff --git a/verif/generator/tosa_error_if.py b/verif/generator/tosa_error_if.py index 53a3199..9fd13d2 100644 --- a/verif/generator/tosa_error_if.py +++ b/verif/generator/tosa_error_if.py @@ -2788,7 +2788,6 @@ class TosaInvalidValidator: if opName.startswith("transpose_conv2d"): # transpose_conv2d - output_shape = args_dict["out_shape"] filter_shape = inputShapes[1] kernel_shape = filter_shape[1:-1] @@ -2810,10 +2809,7 @@ class TosaInvalidValidator: padding[2], padding[3], ) - if output_shape[1] == h and output_shape[2] == w: - return False - # output shape does not match the expected shape - return True + return h < 1 or w < 1 if "conv2d" in opName or "conv3d" in opName: # conv2d, conv3d, depthwise_conv2d |