Skip to content

T
tic
Commit af1fcf97 by 雾雨魔理沙 Committed by Tianqi Chen
Options

[Relay] fix format in ty.py (#1948)

parent c3b569d2
Showing with 4 additions and 4 deletions
python/tvm/relay/ty.py
...@@ -163,17 +163,17 @@ class TypeRelation(TypeConstraint): ...@@ -163,17 +163,17 @@ class TypeRelation(TypeConstraint):
Parameters Parameters
---------- ----------
func : EnvFunc func: EnvFunc
User defined relation function. User defined relation function.
args : list of types args: list of types
List of types to the func. List of types to the func.
num_inputs: int num_inputs: int
Number of input arguments in args, Number of input arguments in args,
this act as a hint for type inference. this act as a hint for type inference.
attrs : Attrs attrs: Attrs
The attribute attached to the relation information The attribute attached to the relation information
""" """
def __init__(self, func, args, num_inputs, attrs): def __init__(self, func, args, num_inputs, attrs):
...@@ -188,7 +188,7 @@ def scalar_type(dtype): ...@@ -188,7 +188,7 @@ def scalar_type(dtype):
Parameters Parameters
---------- ----------
dtype : str dtype: str
The content data type. The content data type.
Returns Returns
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment