pytorch/test/onnx
xadupre 0105cd89ab [ONNX] Fix conversion of attention - 4D (#157130)
Fixes a wrong conversion to onnx while investigation #149662.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/157130
Approved by: https://github.com/gramalingam, https://github.com/justinchuby, https://github.com/titaiwangms

Co-authored-by: Justin Chu <justinchuby@users.noreply.github.com>
2025-07-02 18:05:10 +00:00
..
assets
dynamo [ONNX] Clean up legacy dynamo export code (#149745) 2025-03-23 19:35:16 +00:00
expect
exporter [ONNX] Fix conversion of attention - 4D (#157130) 2025-07-02 18:05:10 +00:00
internal Typo fixes for "overridden" in comments and function names (#155944) 2025-06-14 03:37:38 +00:00
model_defs PEP585 update - test (#145176) 2025-01-22 04:48:28 +00:00
ops [ONNX] Implement Attention-23 (#156431) 2025-06-20 23:54:57 +00:00
torchlib [BE]: Update ruff to 0.11.8 (#153249) 2025-05-12 18:30:52 +00:00
autograd_helper.py
onnx_test_common.py [ONNX] Update default opset to 18 (#156023) 2025-06-16 08:40:49 +00:00
pytorch_test_common.py [ONNX] Clean up the diagnostics module (#149864) 2025-03-26 05:58:32 +00:00
test_autograd_funs.py
test_custom_ops.py
test_fx_passes.py [BE][CI] bump ruff to 0.9.2: multiline assert statements (#144546) 2025-02-27 20:46:16 +00:00
test_fx_type_promotion.py
test_lazy_import.py
test_models_onnxruntime.py PEP585 update - test (#145176) 2025-01-22 04:48:28 +00:00
test_models_quantized_onnxruntime.py Add __main__ guards to tests (#154716) 2025-06-04 14:38:13 +00:00
test_models.py
test_onnx_opset.py
test_onnxscript_no_runtime.py Add __main__ guards to tests (#154716) 2025-06-04 14:38:13 +00:00
test_onnxscript_runtime.py PEP585: More UP006 fixes (#146392) 2025-02-20 06:18:13 +00:00
test_op_consistency.py PEP585 update - test (#145176) 2025-01-22 04:48:28 +00:00
test_pytorch_jit_onnx.py
test_pytorch_onnx_no_runtime.py [ONNX] Clean up the diagnostics module (#149864) 2025-03-26 05:58:32 +00:00
test_pytorch_onnx_onnxruntime_cuda.py
test_pytorch_onnx_onnxruntime.py [BE][CI] bump ruff to 0.9.2: multiline assert statements (#144546) 2025-02-27 20:46:16 +00:00
test_pytorch_onnx_shape_inference.py
test_symbolic_helper.py
test_utility_funs.py
test_verification.py [BE][Ez]: ISC001 Auto concatenate implicit one line strings (#146408) 2025-02-04 19:07:04 +00:00
verify.py Update ruff linter for PEP585 (#147540) 2025-02-22 04:45:17 +00:00