From b5fcfc0cfeffeb3fe9f05c32dc678ab09b2cee31 Mon Sep 17 00:00:00 2001 From: Suraj Sudhir Date: Tue, 16 Apr 2024 16:14:36 -0700 Subject: [reference_model] Remove output_shape from transpose_conv2d Signed-off-by: Suraj Sudhir Change-Id: Ib2b95e73b226d64c4db5ad1ed22c123e04d7e6f9 --- verif/generator/tosa_error_if.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) (limited to 'verif/generator/tosa_error_if.py') 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 -- cgit v1.2.1