File tree Expand file tree Collapse file tree 1 file changed +1
-1
lines changed Expand file tree Collapse file tree 1 file changed +1
-1
lines changed Original file line number Diff line number Diff line change @@ -13,7 +13,7 @@ to_static
13
13
- **function ** (callable) - 待转换的动态图函数。若以装饰器形式使用,则被装饰函数默认会被解析为此参数值,无需显式指定。
14
14
- **input_spec ** (list[InputSpec]|tuple[InputSpec]) - 用于指定被装饰函数中输入 Tensor 的 shape、dtype 和 name 信息,为包含 InputSpec 的 list/tuple 类型。
15
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 `` 编译器来加速训练和推理。
16
+ - **backend ** (str,可选): 指定后端编译器,可以指定为 ``"CINN" `` 或者 ``None ``。当该参数指定为 ``"CINN" `` 时,将会使用 ``CINN `` 编译器来加速训练和推理。默认值为 `` "CINN" ``。
17
17
- **kwargs **: 支持的 key 包括 :attr: `property ` 和 :attr: `full_graph `。
18
18
19
19
- **property ** (bool): 表示被装饰的函数是否以 class property 属性的方式进行导出,默认为 ``False ``。
You can’t perform that action at this time.
0 commit comments