From dbe90d1c6298b6195092a67a6873178e7c3d68db Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=9B=BE=E9=9B=A8=E9=AD=94=E7=90=86=E6=B2=99?=
 <lolisa@marisa.moe>
Date: Fri, 28 Dec 2018 14:41:22 -0800
Subject: [PATCH] Update ir_pass.py

---
 python/tvm/relay/ir_pass.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/python/tvm/relay/ir_pass.py b/python/tvm/relay/ir_pass.py
index 356e746e352ee..38fa26645e17f 100644
--- a/python/tvm/relay/ir_pass.py
+++ b/python/tvm/relay/ir_pass.py
@@ -19,6 +19,7 @@ def post_order_visit(expr, fvisit):
     ----------
     expr : tvm.relay.Expr
         The input expression.
+
     fvisit : function
         The visitor function to be applied.
     """
@@ -35,7 +36,6 @@ def infer_type(expr, mod=None):
     mod: Optional[tvm.relay.Module]
         The global module.
 
-
     Returns
     -------
     checked_expr : tvm.relay.Expr
@@ -368,6 +368,9 @@ def to_anf(expr, mod=None):
     expr : tvm.relay.Expr
         The input expression
 
+    mod: Optional[tvm.relay.Module]
+        The global module.
+
     Returns
     -------
     expr: tvm.relay.Expr