Skip to content

Add backend and kwargs for to_static API #5793

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Apr 11, 2023
Merged

Add backend and kwargs for to_static API #5793

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 4 additions & 1 deletion docs/api/paddle/jit/to_static_cn.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,10 @@ to_static
- **function** (callable) - 待转换的动态图函数。若以装饰器形式使用，则被装饰函数默认会被解析为此参数值，无需显式指定。
- **input_spec** (list[InputSpec]|tuple[InputSpec]) - 用于指定被装饰函数中输入 Tensor 的 shape、dtype 和 name 信息，为包含 InputSpec 的 list/tuple 类型。
- **build_strategy** (BuildStrategy|None)：通过配置 build_strategy，对转换后的计算图进行优化，例如：计算图中算子融合、计算图执行过程中开启内存/显存优化等。关于 build_strategy 更多信息，请参阅 ``paddle.static.BuildStrategy``。默认为 None。
- **property** (bool, Optional): 被装饰的函数是否以 class property 属性的方式进行导出，默认为 False。
- **backend** (str, Optional): 指定后端编译器，可以指定为 `CINN` 或者 None。当该参数指定为 `CINN` 时，将会使用 CINN 编译器来加速训练和推理。
- **kwargs**: 支持的 key 包括 `property`

- **property**: 表示被装饰的函数是否以 class property 属性的方式进行导出


代码示例
Expand Down