Skip to content

Commit d77c8d7

Browse files
authored
Update 13-tvm-relax_block_builder.md
1 parent dd86b8b commit d77c8d7

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

docs/04-api-reference/01-python-api/13-tvm-relax_block_builder.md

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -278,7 +278,7 @@ class Module:
278278
为函数发出输出。
279279
* **参数:**
280280
* **output** (*Expr|*[Tuple](/docs/api-reference/python-api/tvm-relax#classtvmrelaxtuplefieldslistrelaxexprtuplerelaxexprspanspannonenone)|List[*Expr]*):当前块/函数的输出。
281-
* **params** ([tvm.relax.Var](/docs/api-reference/python-api/tvm-relax#classtvmrelaxvarname_hintstridstruct_infostructinfononenonespanspannonenone)*|*[Tuple](/docs/api-reference/python-api/tvm-relax#classtvmrelaxtuplefieldslistrelaxexprtuplerelaxexprspanspannonenone)*|List[***[tvm.relax.Var](/docs/api-reference/python-api/tvm-relax#classtvmrelaxvarname_hintstridstruct_infostructinfononenonespanspannonenone)***]*,optional)[:](https://tvm..org/docs/reference/api/python/ir.html#tvm.ir.GlobalVar)要构建的函数的参数。如果 params 为 None,则表示 params 已在函数中初始化并具有作用域。
281+
* **params** ([tvm.relax.Var](/docs/api-reference/python-api/tvm-relax#classtvmrelaxvarname_hintstridstruct_infostructinfononenonespanspannonenone)*|*[Tuple](/docs/api-reference/python-api/tvm-relax#classtvmrelaxtuplefieldslistrelaxexprtuplerelaxexprspanspannonenone)*|List[***[tvm.relax.Var](/docs/api-reference/python-api/tvm-relax#classtvmrelaxvarname_hintstridstruct_infostructinfononenonespanspannonenone)***]*,optional)[:](/docs/api-reference/python-api/tvm-ir#class-tvmirglobalvarname_hintstr-type_annottypenone-none)要构建的函数的参数。如果 params 为 None,则表示 params 已在函数中初始化并具有作用域。
282282
* **返回:gvar:** 代表函数的 GlobalVar。
283283
* **返回类型:**[tvm.ir.GlobalVar](/docs/api-reference/python-api/tvm-ir#class-tvmirglobalvarname_hintstr-type_annottypenone-none)
284284

@@ -317,22 +317,22 @@ class Module:
317317
* **返回:ret:** 生成的名称。
318318
* **返回类型:** [str](https://docs.python.org/3/library/stdtypes.html#str)
319319

320-
### add_func(*func:*[BaseFunc](https://tvm..org/docs/reference/api/python/ir.html#tvm.ir.BaseFunc), *func_name:*[str](https://docs.python.org/3/library/stdtypes.html#str)) → [GlobalVar](https://tvm..org/docs/reference/api/python/ir.html#tvm.ir.GlobalVar)
320+
### add_func(*func:*[BaseFunc](https://tvm..org/docs/reference/api/python/ir.html#tvm.ir.BaseFunc), *func_name:*[str](https://docs.python.org/3/library/stdtypes.html#str)) → [GlobalVar](/docs/api-reference/python-api/tvm-ir#class-tvmirglobalvarname_hintstr-type_annottypenone-none)
321321

322322

323323
向正在构建的 IRModule 添加 Relax 函数或 TIR PrimFunc。
324324
* **参数:**
325325
* **func** ([BaseFunc](https://tvm..org/docs/reference/api/python/ir.html#tvm.ir.BaseFunc))[:](https://tvm..org/docs/reference/api/python/ir.html#tvm.ir.BaseFunc)要添加的函数。
326326
* **func_name** ([str](https://docs.python.org/3/library/stdtypes.html#str)):要添加的函数的名称。
327327
* **返回:gvar**:与添加的函数绑定的全局变量。
328-
* **返回类型:**[GlobalVar](https://tvm..org/docs/reference/api/python/ir.html#tvm.ir.GlobalVar)
328+
* **返回类型:**[GlobalVar](/docs/api-reference/python-api/tvm-ir#class-tvmirglobalvarname_hintstr-type_annottypenone-none)
329329

330-
### update_func(*gv:*[GlobalVar](https://tvm..org/docs/reference/api/python/ir.html#tvm.ir.GlobalVar), *updated_func:*[BaseFunc](https://tvm..org/docs/reference/api/python/ir.html#tvm.ir.BaseFunc)) → [None](https://docs.python.org/3/library/constants.html#None)
330+
### update_func(*gv:*[GlobalVar](/docs/api-reference/python-api/tvm-ir#class-tvmirglobalvarname_hintstr-type_annottypenone-none), *updated_func:*[BaseFunc](https://tvm..org/docs/reference/api/python/ir.html#tvm.ir.BaseFunc)) → [None](https://docs.python.org/3/library/constants.html#None)
331331

332332

333333
向正在构建的 IRModule 添加 Relax 函数或 TIR PrimFunc。
334334
* **参数:**
335-
* **gv** ([GlobalVar](https://tvm..org/docs/reference/api/python/ir.html#tvm.ir.GlobalVar)):引用要更新的函数的全局变量。
335+
* **gv** ([GlobalVar](/docs/api-reference/python-api/tvm-ir#class-tvmirglobalvarname_hintstr-type_annottypenone-none)):引用要更新的函数的全局变量。
336336
* **updated_func** ([BaseFunc](https://tvm..org/docs/reference/api/python/ir.html#tvm.ir.BaseFunc)):更新后的函数。
337337

338338
### current_block_is_dataflow() → [bool](https://docs.python.org/3/library/functions.html#bool)

0 commit comments

Comments
 (0)