Update ir_pass.py

apache · Dec 28, 2018 · dbe90d1 · dbe90d1
1 parent 10185c7
commit dbe90d1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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