Skip to content

Commit 31f275f

Browse files
authored
[Dy2St] Use "CINN" as default backend in jit.to_static (#7242)
1 parent 50f5ce8 commit 31f275f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

docs/api/paddle/jit/to_static_cn.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ to_static
1313
- **function** (callable) - 待转换的动态图函数。若以装饰器形式使用,则被装饰函数默认会被解析为此参数值,无需显式指定。
1414
- **input_spec** (list[InputSpec]|tuple[InputSpec]) - 用于指定被装饰函数中输入 Tensor 的 shape、dtype 和 name 信息,为包含 InputSpec 的 list/tuple 类型。
1515
- **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`` 编译器来加速训练和推理。
16+
- **backend** (str,可选): 指定后端编译器,可以指定为 ``"CINN"`` 或者 ``None``。当该参数指定为 ``"CINN"`` 时,将会使用 ``CINN`` 编译器来加速训练和推理。默认值为 ``"CINN"``。
1717
- **kwargs**: 支持的 key 包括 :attr:`property` 和 :attr:`full_graph`。
1818

1919
- **property** (bool): 表示被装饰的函数是否以 class property 属性的方式进行导出,默认为 ``False``。

0 commit comments

Comments
 (0)