@@ -391,8 +391,8 @@ def ConvertToDataflow(min_size: int = 2) -> tvm.ir.transform.Pass:
391391
392392 Note: ConvertToDataflow may need to be called first.
393393
394- Params
395- ------
394+ Parameters
395+ ----------
396396 min_size: int
397397 The minimum number of consecutive dataflow bindings
398398 the pass needs to extract a new block.
@@ -647,13 +647,8 @@ def BindParams(
647647 func_name: str
648648 The function name to be bound
649649
650- params : Dict[
651- Union[str,relax.Var],
652- Union[tvm.runtime.NDArray, np.ndarray],
653- ]
654-
655- The map from parameter or parameter name to constant
656- tensors.
650+ params: Dict[Union[str,relax.Var], Union[tvm.runtime.NDArray, np.ndarray]]
651+ The map from parameter or parameter name to constant tensors.
657652
658653 Returns
659654 -------
@@ -994,16 +989,16 @@ def LiftTransformParams(shared_transform: Union[bool, List[str]] = False) -> tvm
994989 Indicates how the parameter transformation function will be produced
995990
996991 - `False` (default): A separate parameter transformation function will be
997- produced for each function with the `"num_input"` attribute.
992+ produced for each function with the `"num_input"` attribute.
998993
999994 - `True`: A single parameter transformation function will be produced,
1000- containing the preprocessing steps common across all functions with
1001- the `"num_input"` attribute.
995+ containing the preprocessing steps common across all functions with
996+ the `"num_input"` attribute.
1002997
1003998 - List[str]: A single parameter transformation function will be produced,
1004- containing the preprocessing steps common across each function whose
1005- name is in the list. Passing a list of all functions with the `"num_input"`
1006- attribute or an empty list is equivalent to passing `True`.
999+ containing the preprocessing steps common across each function whose
1000+ name is in the list. Passing a list of all functions with the `"num_input"`
1001+ attribute or an empty list is equivalent to passing `True`.
10071002
10081003 Returns
10091004 -------
@@ -1219,7 +1214,7 @@ def MetaScheduleTuneIRMod(
12191214 maximum number of trials per task
12201215 op_names: Optional[List[str]]
12211216 A list of operator names to specify which op to tune. When it is None, all operators
1222- are tuned.
1217+ are tuned.
12231218
12241219 Returns
12251220 -------
0 commit comments