Enable some ConvTranspose backend tests

webonnx · Aug 12, 2023 · 4b3b6e8 · 4b3b6e8
1 parent faf3479
commit 4b3b6e8
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/wonnx-py/tests/test_onnx_backend.py b/wonnx-py/tests/test_onnx_backend.py
@@ -197,6 +197,11 @@ def do_enforce_test_coverage_safelist(model):  # type: (ModelProto) -> bool
 backend_test.include(f"test_softmax_negative_axis_cpu$")
 backend_test.include(f"test_softmax_default_axis_cpu$")
 
+# ConvTranspose
+# We only have partial attribute support right now, so we hand select a few test cases limited to the supported ones
+backend_test.include(f"test_convtranspose$")
+backend_test.include(f"test_convtranspose_pads$")
+
 globals().update(backend_test.enable_report().test_cases)
 
 if __name__ == "__main__":