tf.eye has a bug when version is below 1.11.

Author: zhijxu-MS

tests/test_backend.py

@@ -130,7 +130,34 @@ def test_expand_dims_more_unknown_rank(self):
         self._run_test_case([_OUTPUT], {_INPUT: x_val})
 
     @check_opset_min_version(9, "ConstantOfShape")
-    def test_eye_non_const(self):
+    def test_eye_non_const1(self):
+        # tf.eye(num_rows), num_rows is not const here
+        tf.reset_default_graph()
+        x_val = np.array(5, dtype=np.int32)
+        x = tf.placeholder(tf.int32, shape=[], name=_TFINPUT)
+        y = tf.eye(x, dtype=tf.int32)
+        _ = tf.identity(y, name=_TFOUTPUT)
+        y1 = tf.eye(x, dtype=tf.int64)
+        _ = tf.identity(y1, name=_TFOUTPUT1)
+        y2 = tf.eye(x, dtype=tf.float32)
+        _ = tf.identity(y2, name=_TFOUTPUT2)
+        self._run_test_case([_OUTPUT, _OUTPUT1, _OUTPUT2], {_INPUT: x_val}, rtol=0)
+
+        # tf.eye(num_rows, num_columns), both num_rows and num_columns are not const here
+        tf.reset_default_graph()
+        x_val = np.array([5, 10], dtype=np.int32)
+        x = tf.placeholder(tf.int32, shape=[2], name=_TFINPUT)
+        y = tf.eye(x[0], x[1], dtype=tf.int32)
+        _ = tf.identity(y, name=_TFOUTPUT)
+        y1 = tf.eye(x[0], x[1], dtype=tf.int64)
+        _ = tf.identity(y1, name=_TFOUTPUT1)
+        y2 = tf.eye(x[0], x[1], dtype=tf.float32)
+        _ = tf.identity(y2, name=_TFOUTPUT2)
+        self._run_test_case([_OUTPUT, _OUTPUT1, _OUTPUT2], {_INPUT: x_val}, rtol=0)
+
+    @check_tf_min_version("1.11", "eye has bug when version is below 1.11")
+    @check_opset_min_version(9, "ConstantOfShape")
+    def test_eye_non_const2(self):
         # tf.eye(num_rows), num_rows is not const here
         for np_dtype, tf_dtype in zip([np.int32, np.int64, np.float32, np.float64],
                                       [tf.int32, tf.int64, tf.float32, tf.float64]):