Skip to content

Commit cffe8c5

Browse files
committed
Add support for Keras resnet50 pretrained model testing.
1 parent 3a16b3d commit cffe8c5

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/test_backend.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2860,6 +2860,7 @@ def func(a, b, c):
28602860
graph_validator=lambda g: check_op_count(g, "Gemm", 1))
28612861

28622862
# test for gemm pattern0: alpha*A*B + beta*C
2863+
@check_opset_min_version(12, "Optimizer bug in ORT 1.2")
28632864
def test_gemm_pattern0_fail_broadcast(self):
28642865
# shapes (3, 3) * (3, 1) + (1, 4) => (3, 1) + (1, 4)
28652866
# c not uni-broadcastable to a * b, so should not use GEMM

0 commit comments

Comments
 (0)