pytorch/torch/onnx/_internal/exporter
Justin Chu fdacf3c920 [ONNX] Update types in VerificationInfo (#149377)
torch.types.Number was rendered as is in the documentation and can be confusing. We write the original types instead to reduce confusion for users.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/149377
Approved by: https://github.com/titaiwangms
2025-03-18 15:37:39 +00:00
..
_torchlib Update ruff linter for PEP585 (#147540) 2025-02-22 04:45:17 +00:00
__init__.py
_analysis.py
_building.py [BE][CI] bump ruff to 0.9.2: multiline assert statements (#144546) 2025-02-27 20:46:16 +00:00
_capture_strategies.py [ONNX] Use torch export to get dynamic shapes for JIT convert strategy (#148627) 2025-03-07 23:41:50 +00:00
_compat.py [ONNX] Use onnxscript apis for 2.7 (#148453) 2025-03-05 20:10:00 +00:00
_constants.py [ONNX] Consolidate constants to a single location (#147166) 2025-02-14 19:08:19 +00:00
_core.py [ONNX] Update saved exported program in debugging report if the exporting passes run_decomposition() (#148617) 2025-03-06 07:03:45 +00:00
_decomp.py [ONNX] Avoid overwriting overlapped decomposed functions (#142831) 2024-12-11 18:47:40 +00:00
_dispatching.py [ONNX] Refactor dispatcher and registry (#147396) 2025-02-19 19:38:28 +00:00
_dynamic_shapes.py [ONNX] Fix missed None type support in dyamic shapes string cases (#148025) 2025-02-27 07:57:47 +00:00
_errors.py
_fx_passes.py remove allow-untyped-defs onnx/_internal/exporter/_fx_passes.py (#144134) 2025-01-03 20:18:40 +00:00
_ir_passes.py [ONNX] Refactor dispatcher and registry (#147396) 2025-02-19 19:38:28 +00:00
_isolated.py
_onnx_program.py [ONNX] Assert capture strategy in tests (#148348) 2025-03-05 22:31:54 +00:00
_registration.py [ONNX] Use onnxscript apis for 2.7 (#148453) 2025-03-05 20:10:00 +00:00
_reporting.py [ONNX] Separate decomp into single step and add to the report (#140767) 2024-11-15 04:26:16 +00:00
_schemas.py [BE][CI] bump ruff to 0.9.2: multiline assert statements (#144546) 2025-02-27 20:46:16 +00:00
_tensors.py
_testing.py [ONNX] Assert capture strategy in tests (#148348) 2025-03-05 22:31:54 +00:00
_verification.py [ONNX] Update types in VerificationInfo (#149377) 2025-03-18 15:37:39 +00:00