Fix issue mutating if expressions (#2601)

(cherry picked from commit f6be4d6970)

python/tvm/relay/expr_functor.py

@@ -117,9 +117,9 @@ class ExprMutator(ExprFunctor):
 
     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])