commit f08015e7fde92c835907d4c9b7ad6d3f634e94a5
author Jared Roesch <roeschinc@gmail.com> Mon Feb 18 14:18:57 2019 -0800
committer ZihengJiang <ziheng@apache.org> Mon Feb 18 14:42:45 2019 -0800
tree ce7427e59b44b8da70e548ef2e097906e75667f3
parent 3d53ef3cfb93fe595465564d161d85fd13e7ee45

Fix issue mutating if expressions (#2601)


(cherry picked from commit f6be4d69703afac399072f36b8e000ebbe034f6b)

python/tvm/relay/expr_functor.py

diff --git a/python/tvm/relay/expr_functor.py b/python/tvm/relay/expr_functor.py
index eafe5f0..57c5732 100644
--- a/python/tvm/relay/expr_functor.py
+++ b/python/tvm/relay/expr_functor.py
@@ -117,9 +117,9 @@
 
     def visit_if(self, ite):
         return If(
-            self.visit(ite.guard),
-            self.visit(ite.true_b),
-            self.visit(ite.false_b))
+            self.visit(ite.cond),
+            self.visit(ite.true_branch),
+            self.visit(ite.false_branch))
 
     def visit_tuple(self, tup):
         return Tuple([self.visit(field) for field in tup.fields])