Skip to content

Commit 81cdbd9

Browse files
committed
test tf.eye on opset 12
1 parent bec7f3b commit 81cdbd9

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/test_backend.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -202,6 +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")
205206
def test_eye_non_const1(self):
206207
# tf.eye(num_rows), num_rows is not const here
207208
x_val = np.array(5, dtype=np.int32)
@@ -223,6 +224,7 @@ def func(x):
223224

224225
@check_tf_min_version("1.11", "eye has bug when version is below 1.11")
225226
@check_opset_min_version(9, "ConstantOfShape")
227+
@check_opset_after_tf_version("2.2", 12, "ConstantOfShape")
226228
def test_eye_non_const2(self):
227229
# tf.eye(num_rows), num_rows is not const here
228230
for np_dtype in [np.int32, np.int64, np.float32, np.float64]:

0 commit comments

Comments
 (0)