Merge pull request #343 from lucienwang1009/cond_test_bug

fix cond_test bug

Author: nbcsm

tests/test_cond.py

@@ -131,22 +131,20 @@ def cond_graph2():
         output_names_with_port = ["output:0"]
         self.run_test_case(feed_dict, input_names_with_port, output_names_with_port)
 
-    def test_while_loop_between_conds(self):
+    def test_while_loop_in_cond(self):
         x_val = np.array([1, 2, 3], dtype=np.float32)
         y_val = np.array([4, 5, 6], dtype=np.float32)
         x = tf.placeholder(tf.float32, x_val.shape, name="input_1")
         y = tf.placeholder(tf.float32, y_val.shape, name="input_2")
 
         def cond_graph():
             b = tf.constant(np.array([0], dtype=np.int32), dtype=tf.int32)
-            z = tf.gather_nd(x, b)
             # while_loop
             c = lambda y: tf.reduce_any(tf.less(y, 10))
             b = lambda i: tf.add(y, 1)
-            r = tf.while_loop(c, b, [y])
-            return tf.cond(x[0] > y[0], lambda: z, lambda: r)
+            return tf.while_loop(c, b, [y])
 
-        res = x[2] * tf.cond(x[0] < y[0], lambda: x, cond_graph, name="test_cond")
+        res = tf.cond(x[0] < y[0], lambda: x, cond_graph, name="test_cond")
         _ = tf.identity(res, name="output")
 
         feed_dict = {"input_1:0": x_val, "input_2:0": y_val}