OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
fcc1063566
pytorch/test/onnx
History
Justin Chu 160ab53dd5 Update weight tensor initialization in RMSNormalization (#166550) 
Ensure a >1d tensor as weight for ORT compatibility.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/166550
Approved by: https://github.com/titaiwangms
2025-10-31 14:29:27 +00:00
..
assets
expect [ONNX] Remove the operators test (#136335) 2024-09-25 17:44:18 +00:00
exporter Update weight tensor initialization in RMSNormalization (#166550) 2025-10-31 14:29:27 +00:00
internal [ONNX] Refactor torchscript based exporter (#161323) 2025-09-02 16:10:30 +00:00
model_defs PEP585 update - test (#145176) 2025-01-22 04:48:28 +00:00
ops [ONNX] Create fake implementations for onnx ops; fix boolean mask in attention (#165780) 2025-10-29 04:51:49 +00:00
torchlib Update weight tensor initialization in RMSNormalization (#166550) 2025-10-31 14:29:27 +00:00
autograd_helper.py [2/N][Fix] Fix typo in test folder (#166374) 2025-10-29 03:02:07 +00:00
onnx_test_common.py [1/N] Apply UP035 rule in tests (#163947) 2025-09-29 01:42:01 +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 [ONNX] Refactor torchscript based exporter (#161323) 2025-09-02 16:10:30 +00:00
test_custom_ops.py Fix unused Python variables in test/[e-z]* (#136964) 2024-12-18 23:02:30 +00:00
test_fx_type_promotion.py [ONNX] Remove type promotion rule for pow (#139527) 2024-11-02 02:19:50 +00:00
test_lazy_import.py [ONNX][DORT] Lazy-import onnxruntime (#134662) 2024-08-31 00:06:28 +00:00
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 Enable all PIE rules on ruff (#165814) 2025-10-18 07:36:18 +00:00
test_onnxscript_no_runtime.py [ONNX] Default to dynamo export (#159646) 2025-09-02 22:45:55 +00:00
test_onnxscript_runtime.py [ONNX] Refactor torchscript based exporter (#161323) 2025-09-02 16:10:30 +00:00
test_op_consistency.py PEP585 update - test (#145176) 2025-01-22 04:48:28 +00:00
test_pytorch_jit_onnx.py [ONNX] Refactor torchscript based exporter (#161323) 2025-09-02 16:10:30 +00:00
test_pytorch_onnx_onnxruntime_cuda.py
test_pytorch_onnx_onnxruntime.py [4/N] Remove unused loop variables in tests (#166690) 2025-10-31 10:20:48 +00:00
test_pytorch_onnx_shape_inference.py [ONNX] Default to dynamo export (#159646) 2025-09-02 22:45:55 +00:00
test_symbolic_helper.py [ONNX] Refactor torchscript based exporter (#161323) 2025-09-02 16:10:30 +00:00
test_utility_funs.py [ONNX] Default to dynamo export (#159646) 2025-09-02 22:45:55 +00:00
verify.py Update ruff linter for PEP585 (#147540) 2025-02-22 04:45:17 +00:00