diff --git a/python/tvm/relay/dataflow_pattern/__init__.py b/python/tvm/relay/dataflow_pattern/__init__.py index b368f4e5175e..320a599d5d91 100644 --- a/python/tvm/relay/dataflow_pattern/__init__.py +++ b/python/tvm/relay/dataflow_pattern/__init__.py @@ -547,11 +547,11 @@ class CallPattern(DFPattern): Parameters ---------- - op: realy.dataflow_pattern.DFPattern + op: relay.dataflow_pattern.DFPattern The operation to be called. - args: List[realy.dataflow_pattern.DFPattern] - The arguments to the call. + args: List[relay.dataflow_pattern.DFPattern] + The arguments to the call or None to match any arguments. """ @@ -569,10 +569,10 @@ class FunctionPattern(DFPattern): Parameters ---------- - params: List[realy.dataflow_pattern.DFPattern] - The parameters to the Function. + params: List[relay.dataflow_pattern.DFPattern] + The parameters to the Function or None to match any parameters. - body: realy.dataflow_pattern.DFPattern + body: relay.dataflow_pattern.DFPattern The body fo the Function """ @@ -886,7 +886,7 @@ def partition( Parameters ---------- - partion: tvm.relay.dataflow_pattern.DFPattern + pattern: tvm.relay.dataflow_pattern.DFPattern The pattern to match expr : tvm.relay.Expr The expression to split into functions