Fix issue mutating if expressions (apache#2601)

neo-ai · Feb 20, 2019 · 7e4c738 · 7e4c738
1 parent 1bc5f46
commit 7e4c738
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/tvm/relay/expr_functor.py b/python/tvm/relay/expr_functor.py
@@ -143,9 +143,9 @@ def visit_global_id(self, global_var):
 
     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])