Skip to content

Commit 9bbd3c4

Browse files
committed
disable topk-10 test for topk's shape inference bug.
1 parent 0de86ee commit 9bbd3c4

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
@@ -1417,6 +1417,7 @@ def test_cancel_transpose(self):
14171417
_ = tf.identity(x_, name=_TFOUTPUT)
14181418
self._run_test_case([_OUTPUT], {_INPUT: x_val})
14191419

1420+
@check_onnxruntime_min_version("0.5.0", "topk-10's shape inference function has a bug")
14201421
@check_opset_min_version(6, "cast")
14211422
def test_topk1(self):
14221423
x_val = np.arange(3 * 2 * 3).astype("float32")

0 commit comments

Comments
 (0)