Skip to content

Commit 608817a

Browse files
authored
Add backend and kwargs for to_static API (#5793)
1 parent 59635ad commit 608817a

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

docs/api/paddle/jit/to_static_cn.rst

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,10 @@ to_static
1414
- **function** (callable) - 待转换的动态图函数。若以装饰器形式使用,则被装饰函数默认会被解析为此参数值,无需显式指定。
1515
- **input_spec** (list[InputSpec]|tuple[InputSpec]) - 用于指定被装饰函数中输入 Tensor 的 shape、dtype 和 name 信息,为包含 InputSpec 的 list/tuple 类型。
1616
- **build_strategy** (BuildStrategy|None):通过配置 build_strategy,对转换后的计算图进行优化,例如:计算图中算子融合、计算图执行过程中开启内存/显存优化等。关于 build_strategy 更多信息,请参阅 ``paddle.static.BuildStrategy``。默认为 None。
17-
- **property** (bool, Optional): 被装饰的函数是否以 class property 属性的方式进行导出,默认为 False。
17+
- **backend** (str, Optional): 指定后端编译器,可以指定为 `CINN` 或者 None。当该参数指定为 `CINN` 时,将会使用 CINN 编译器来加速训练和推理。
18+
- **kwargs**: 支持的 key 包括 `property`
19+
20+
- **property**: 表示被装饰的函数是否以 class property 属性的方式进行导出
1821

1922

2023
代码示例

0 commit comments

Comments
 (0)