Fix TF1 test
@@ -987,10 +987,7 @@ def eye(size, dtype=None, name=None):
else:
n, m = size, size
with tf_ops.init_scope():
- v = tf.eye(n, m, dtype=dtype, name=name)
- if py_all(v.shape.as_list()):
- return variable(v, dtype=dtype, name=name)
- return v
+ return tf.eye(n, m, dtype=dtype, name=name)
@symbolic
