Fix issue mutating if expressions (apache#2601)

(cherry picked from commit f6be4d6)
Duffyxx · Feb 18, 2019 · f08015e · f08015e
1 parent 3d53ef3
commit f08015e
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
@@ -117,9 +117,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])