@@ -202,7 +202,7 @@ def test_expand_dims_more_unknown_rank(self):
202
202
self ._test_expand_dims_more_unknown_rank (i )
203
203
204
204
@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 " )
206
206
def test_eye_non_const1 (self ):
207
207
# tf.eye(num_rows), num_rows is not const here
208
208
x_val = np .array (5 , dtype = np .int32 )
@@ -224,7 +224,7 @@ def func(x):
224
224
225
225
@check_tf_min_version ("1.11" , "eye has bug when version is below 1.11" )
226
226
@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 " )
228
228
def test_eye_non_const2 (self ):
229
229
# tf.eye(num_rows), num_rows is not const here
230
230
for np_dtype in [np .int32 , np .int64 , np .float32 , np .float64 ]:
@@ -1386,6 +1386,7 @@ def func():
1386
1386
self ._run_test_case (func , [_OUTPUT ], {}, check_value = False , check_shape = True )
1387
1387
1388
1388
@skip_caffe2_backend ()
1389
+ @check_opset_after_tf_version ("2.2" , 9 , "RandomUniform" )
1389
1390
def test_randomuniform_dyn_shape (self ):
1390
1391
# test for dynamic shape coming from a shape op
1391
1392
x_val = np .array ([0 , 1 , 2 , 3 , 5 ], dtype = np .int64 )
0 commit comments