pytorch/test/onnx
Justin Chu a9e6356271 [ONNX] Update torch.onnx.export API (#131501)
- Add a `kwargs` option; add the `dynamic_shapes` option so users can supply it directly to `torch.export`.
- Make the options keyword-only arguments (bc-breaking)
- Deprecate the `training` and `operator_export_type` options and include a warning message. The exact time for removal is TBD but the message should discourage users from using the options.
- Deprecate two functions `exportable_ops` and pretty print export

Pull Request resolved: https://github.com/pytorch/pytorch/pull/131501
Approved by: https://github.com/titaiwangms
2024-07-24 20:03:17 +00:00
..
assets
dynamo [ONNX] Update torch.onnx.export API (#131501) 2024-07-24 20:03:17 +00:00
expect
internal [ONNX] Remove beartype usage (#130484) 2024-07-18 22:07:40 +00:00
model_defs [Ez][BE]: Enable new stable ruff rules (#129825) 2024-07-02 14:47:10 +00:00
torch_export
autograd_helper.py
error_reproduction.py
onnx_test_common.py [ONNX] Remove beartype usage (#130484) 2024-07-18 22:07:40 +00:00
pytorch_test_common.py [Caffe2] [2/N] Remove Caffe2 from tests (#128911) 2024-06-19 00:05:50 +00:00
test_autograd_funs.py
test_custom_ops.py
test_export_modes.py [1/N] [Caffe2] Remove caffe2_aten_fallback code (#128675) 2024-06-17 21:25:59 +00:00
test_fx_op_consistency.py Fix out_wrapper, _make_copy_from_view to handle all signatures (#130937) 2024-07-21 20:39:24 +00:00
test_fx_passes.py
test_fx_to_onnx.py [ONNX] Add upsample trilinear to skip decomp (#128259) 2024-06-14 21:20:44 +00:00
test_fx_to_onnx_decomp_skip.py [ONNX] Add upsample trilinear to skip decomp (#128259) 2024-06-14 21:20:44 +00:00
test_fx_to_onnx_with_onnxruntime.py [ONNX] Remove beartype usage (#130484) 2024-07-18 22:07:40 +00:00
test_fx_type_promotion.py
test_models.py
test_models_onnxruntime.py
test_models_quantized_onnxruntime.py
test_onnx_opset.py
test_onnxscript_no_runtime.py
test_onnxscript_runtime.py
test_op_consistency.py
test_operators.py [Caffe2] [2/N] Remove Caffe2 from tests (#128911) 2024-06-19 00:05:50 +00:00
test_pytorch_jit_onnx.py
test_pytorch_onnx_no_runtime.py [1/N] [Caffe2] Remove caffe2_aten_fallback code (#128675) 2024-06-17 21:25:59 +00:00
test_pytorch_onnx_onnxruntime.py [BE][Easy] apply autofix for ruff rules unnecessary-collection-call (C408): list() / tuple() / dict() (#130199) 2024-07-11 17:30:28 +00:00
test_pytorch_onnx_onnxruntime_cuda.py
test_pytorch_onnx_shape_inference.py
test_symbolic_helper.py
test_utility_funs.py [ONNX] Add onnx::Gelu support for version 20 (#128773) 2024-06-19 15:39:02 +00:00
test_verification.py
verify.py