Justin Chu e2a87fb1e9 [ONNX] Update exporter logic (#134304) ...

Sync the exporter logic with torch-onnx at https://github.com/justinchuby/torch-onnx/compare/v0.1.12...v0.1.15.

https://github.com/pytorch/pytorch/issues/129277

- Create a `testing` module to facilitate testing model accuracy. The model is internal
- Improve decomp table
- Improve model verification logic
- Add tests

The next PRs will enable OpInfo tests and clean up existing code.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/134304
Approved by: https://github.com/titaiwangms

2024-08-24 00:49:54 +00:00

..

assets

[ONNX] add test images to repo (#63717)

2021-08-23 07:43:21 -07:00

dynamo

[ONNX] Opt into ruff fmt (#134120)

2024-08-22 22:44:03 +00:00

expect

[ONNX] Adds Support for Some Bitwise Ops in Onnx Exporter (#126229)

2024-05-22 07:47:43 +00:00

exporter

[ONNX] Update exporter logic (#134304)

2024-08-24 00:49:54 +00:00

internal

[ONNX] Opt into ruff fmt (#134120)

2024-08-22 22:44:03 +00:00

model_defs

[ONNX] Opt into ruff fmt (#134120)

2024-08-22 22:44:03 +00:00

torch_export

[ONNX] Opt into ruff fmt (#134120)

2024-08-22 22:44:03 +00:00

autograd_helper.py

[ONNX] Format ONNX python with black

2022-05-05 00:19:22 +00:00

error_reproduction.py

[BE][Easy][8/19] enforce style for empty lines in import segments in test/[k-p]*/ (#129759)

2024-07-31 02:09:20 +00:00

onnx_test_common.py

[ONNX] Disable op_level_debug tests (#133485)

2024-08-14 22:02:12 +00:00

pytorch_test_common.py

[ONNX] Opt into ruff fmt (#134120)

2024-08-22 22:44:03 +00:00

test_autograd_funs.py

[5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126)

2024-05-27 14:49:57 +00:00

test_custom_ops.py

Add None return type to init -- tests (#132352)

2024-08-01 15:44:51 +00:00

test_fx_op_consistency.py

[ONNX] Disable op_level_debug tests (#133485)

2024-08-14 22:02:12 +00:00

test_fx_passes.py

Add None return type to init -- tests (#132352)

2024-08-01 15:44:51 +00:00

test_fx_to_onnx_decomp_skip.py

Add None return type to init -- tests (#132352)

2024-08-01 15:44:51 +00:00

test_fx_to_onnx_with_onnxruntime.py

[ONNX] Disable op_level_debug tests (#133485)

2024-08-14 22:02:12 +00:00

test_fx_to_onnx.py

[ONNX] Disable op_level_debug tests (#133485)

2024-08-14 22:02:12 +00:00

test_fx_type_promotion.py

[5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126)

2024-05-27 14:49:57 +00:00

test_models_onnxruntime.py

[ONNX] Opt into ruff fmt (#134120)

2024-08-22 22:44:03 +00:00

test_models_quantized_onnxruntime.py

[5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126)

2024-05-27 14:49:57 +00:00

test_models.py

[BE][Easy][8/19] enforce style for empty lines in import segments in test/[k-p]*/ (#129759)

2024-07-31 02:09:20 +00:00

test_onnx_opset.py

Add None return type to init -- tests (#132352)

2024-08-01 15:44:51 +00:00

test_onnxscript_no_runtime.py

[ONNX] Opt into ruff fmt (#134120)

2024-08-22 22:44:03 +00:00

test_onnxscript_runtime.py

[ONNX] Opt into ruff fmt (#134120)

2024-08-22 22:44:03 +00:00

test_op_consistency.py

[BE][Easy][8/19] enforce style for empty lines in import segments in test/[k-p]*/ (#129759)

2024-07-31 02:09:20 +00:00

test_operators.py

[ONNX] Opt into ruff fmt (#134120)

2024-08-22 22:44:03 +00:00

test_pytorch_jit_onnx.py

[5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126)

2024-05-27 14:49:57 +00:00

test_pytorch_onnx_no_runtime.py

Add None return type to init -- tests (#132352)

2024-08-01 15:44:51 +00:00

test_pytorch_onnx_onnxruntime_cuda.py

Add None return type to init -- tests (#132352)

2024-08-01 15:44:51 +00:00

test_pytorch_onnx_onnxruntime.py

[ONNX] Opt into ruff fmt (#134120)

2024-08-22 22:44:03 +00:00

test_pytorch_onnx_shape_inference.py

[BE][Easy][8/19] enforce style for empty lines in import segments in test/[k-p]*/ (#129759)

2024-07-31 02:09:20 +00:00

test_symbolic_helper.py

[ONNX] Fix check_training_mode in symbolic_helper (#78376)

2022-05-27 00:38:16 +00:00

test_utility_funs.py

[ONNX] Opt into ruff fmt (#134120)

2024-08-22 22:44:03 +00:00

test_verification.py

[ONNX] Opt into ruff fmt (#134120)

2024-08-22 22:44:03 +00:00

verify.py

[ONNX] Remove torch.onnx._export (#133824)

2024-08-20 20:54:48 +00:00