OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

fix: do_constant_folding arg when exporting ONNX (#71348)

Browse Source 
Summary:
Fixes #ISSUE_NUMBER

Pull Request resolved: https://github.com/pytorch/pytorch/pull/71348

Reviewed By: H-Huang

Differential Revision: D33662228

Pulled By: msaroufim

fbshipit-source-id: a69c72838b7ff41a2305453ef00666c060ade593
This commit is contained in:
Tran N.M. Hoang 2022-01-19 21:38:04 -08:00 committed by Facebook GitHub Bot
parent 4cb02e62a6
commit 75dd62b406
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
torch/onnx/__init__.py
View File

@ -186,7 +186,7 @@ def export(model, args, f, export_params=True, verbose=False, training=TrainingM
opset_version (int, default 9): The version of the
`default (ai.onnx) opset <https://github.com/onnx/onnx/blob/master/docs/Operators.md>`_
to target. Must be >= 7 and <= 15.
do_constant_folding (bool, default False): Apply the constant-folding optimization.
do_constant_folding (bool, default True): Apply the constant-folding optimization.
Constant-folding will replace some of the ops that have all constant inputs
with pre-computed constant nodes.
dynamic_axes (dict<string, dict<int, string>> or dict<string, list(int)>, default empty dict):