pytorch/torch/onnx
Justin Chu 299fbbccec [ONNX] Fix check_training_mode in symbolic_helper (#78376)
`check_training_mode` always warned that an op is set to training because it was comparing an int `op_train_mode` with an Enum `GLOBALS.training_mode`. This PR fixes the behavior.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/78376
Approved by: https://github.com/garymm
2022-05-27 00:38:16 +00:00
..
__init__.py [ONNX] Clean up module imports (#77423) 2022-05-20 01:56:24 +00:00
_constants.py [ONNX] Refactor to remove inline imports - attempt 2 (#77448) 2022-05-16 14:44:24 +00:00
_globals.py [ONNX] Clean up module imports (#77423) 2022-05-20 01:56:24 +00:00
_patch_torch.py [ONNX] Modernize python syntax (#77935) 2022-05-24 22:52:37 +00:00
onnx_supported_ops.py [ONNX] Modernize python syntax (#77935) 2022-05-24 22:52:37 +00:00
operators.py [ONNX] Clean up module imports (#77423) 2022-05-20 01:56:24 +00:00
README.md [ONNX] Minor doc update (#69501) (#69550) 2022-02-11 22:05:15 +00:00
symbolic_caffe2.py [ONNX] Clean up module imports (#77423) 2022-05-20 01:56:24 +00:00
symbolic_helper.py [ONNX] Fix check_training_mode in symbolic_helper (#78376) 2022-05-27 00:38:16 +00:00
symbolic_opset7.py [ONNX] Clean up module imports (#77423) 2022-05-20 01:56:24 +00:00
symbolic_opset8.py [ONNX] Modernize python syntax (#77935) 2022-05-24 22:52:37 +00:00
symbolic_opset9.py [ONNX] Modernize python syntax (#77935) 2022-05-24 22:52:37 +00:00
symbolic_opset10.py [ONNX] Modernize python syntax (#77935) 2022-05-24 22:52:37 +00:00
symbolic_opset11.py [ONNX] Clean up module imports (#77423) 2022-05-20 01:56:24 +00:00
symbolic_opset12.py [ONNX] Clean up module imports (#77423) 2022-05-20 01:56:24 +00:00
symbolic_opset13.py [ONNX] Clean up module imports (#77423) 2022-05-20 01:56:24 +00:00
symbolic_opset14.py [ONNX] Clean up module imports (#77423) 2022-05-20 01:56:24 +00:00
symbolic_opset15.py [ONNX] Clean up module imports (#77423) 2022-05-20 01:56:24 +00:00
symbolic_opset16.py [ONNX] Clean up module imports (#77423) 2022-05-20 01:56:24 +00:00
symbolic_registry.py [ONNX] Modernize python syntax (#77935) 2022-05-24 22:52:37 +00:00
utils.py [ONNX] Modernize python syntax (#77935) 2022-05-24 22:52:37 +00:00
verification.py [ONNX] Extract export verification as standalone api from unittest 2022-05-26 19:49:19 +00:00

torch.onnx

Torch->ONNX converter / exporter.

User-facing docs.

Developer docs.