Skip to content

Commit 652b25f

Browse files
authored
[Dy2St][CINN] Use CINN as default dy2st backend (#7168)
1 parent 36bbd04 commit 652b25f

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

docs/api/paddle/jit/to_static_cn.rst

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -7,17 +7,17 @@ to_static
77
88
本装饰器将函数内的动态图 API 转化为静态图 API。此装饰器自动处理静态图模式下的 Program 和 Executor,并将结果作为动态图 Tensor 返回。输出的动态图 Tensor 可以继续进行动态图训练、预测或其他运算。如果被装饰的函数里面调用其他动态图函数,被调用的函数也会被转化为静态图函数。
99

10-
1110
参数
1211
::::::::::::
1312

1413
- **function** (callable) - 待转换的动态图函数。若以装饰器形式使用,则被装饰函数默认会被解析为此参数值,无需显式指定。
1514
- **input_spec** (list[InputSpec]|tuple[InputSpec]) - 用于指定被装饰函数中输入 Tensor 的 shape、dtype 和 name 信息,为包含 InputSpec 的 list/tuple 类型。
16-
- **build_strategy** (BuildStrategy|None):通过配置 build_strategy,对转换后的计算图进行优化,例如:计算图中算子融合、计算图执行过程中开启内存/显存优化等。关于 build_strategy 更多信息,请参阅 ``paddle.static.BuildStrategy``。默认为 None。
17-
- **backend** (str,可选): 指定后端编译器,可以指定为 `CINN` 或者 None。当该参数指定为 `CINN` 时,将会使用 CINN 编译器来加速训练和推理。
18-
- **kwargs**: 支持的 key 包括 `property`
15+
- **build_strategy** (BuildStrategy|None):通过配置 :attr:`build_strategy`,对转换后的计算图进行优化,例如:计算图中算子融合、计算图执行过程中开启内存/显存优化等。关于 :attr:`build_strategy` 更多信息,请参阅 :ref:`paddle.static.BuildStrategy <cn_api_paddle_static_BuildStrategy>`。默认为 ``None``
16+
- **backend** (str,可选): 指定后端编译器,可以指定为 ``"CINN"`` 或者 ``None``。当该参数指定为 ``"CINN"`` 时,将会使用 ``CINN`` 编译器来加速训练和推理。
17+
- **kwargs**: 支持的 key 包括 :attr:`property`:attr:`full_graph`。
1918

20-
- **property**: 表示被装饰的函数是否以 class property 属性的方式进行导出
19+
- **property** (bool): 表示被装饰的函数是否以 class property 属性的方式进行导出,默认为 ``False``。
20+
- **full_graph** (bool): 表示被装饰的函数是否以整图静态图的方式进行导出,默认为 ``False``。
2121

2222

2323
代码示例

0 commit comments

Comments
 (0)