Skip to content

Commit 7127c43

Browse files
committed
set min opset for UniformRandom
1 parent 79d2a6b commit 7127c43

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/test_backend.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -202,7 +202,7 @@ def test_expand_dims_more_unknown_rank(self):
202202
self._test_expand_dims_more_unknown_rank(i)
203203

204204
@check_opset_min_version(9, "ConstantOfShape")
205-
@check_opset_after_tf_version("2.2", 12, "ConstantOfShape")
205+
@check_opset_after_tf_version("2.2", 12, "MatrixDiag")
206206
def test_eye_non_const1(self):
207207
# tf.eye(num_rows), num_rows is not const here
208208
x_val = np.array(5, dtype=np.int32)
@@ -224,7 +224,7 @@ def func(x):
224224

225225
@check_tf_min_version("1.11", "eye has bug when version is below 1.11")
226226
@check_opset_min_version(9, "ConstantOfShape")
227-
@check_opset_after_tf_version("2.2", 12, "ConstantOfShape")
227+
@check_opset_after_tf_version("2.2", 12, "MatrixDiag")
228228
def test_eye_non_const2(self):
229229
# tf.eye(num_rows), num_rows is not const here
230230
for np_dtype in [np.int32, np.int64, np.float32, np.float64]:
@@ -1386,6 +1386,7 @@ def func():
13861386
self._run_test_case(func, [_OUTPUT], {}, check_value=False, check_shape=True)
13871387

13881388
@skip_caffe2_backend()
1389+
@check_opset_after_tf_version("2.2", 9, "RandomUniform")
13891390
def test_randomuniform_dyn_shape(self):
13901391
# test for dynamic shape coming from a shape op
13911392
x_val = np.array([0, 1, 2, 3, 5], dtype=np.int64)

0 commit comments

Comments
 (0)