pytorch/test/onnx
Bludator f4187050fe [ONNX] Remove special handling of torchvision.ops imports in onnx export (#141569)
Fixes #141568

Pull Request resolved: https://github.com/pytorch/pytorch/pull/141569
Approved by: https://github.com/titaiwangms

Co-authored-by: Justin Chu <justinchuby@users.noreply.github.com>
Co-authored-by: Ti-Tai Wang <titaiwang@microsoft.com>
2024-11-28 18:05:40 +00:00
..
assets
dynamo [ONNX] Refactor exporter errors (#135180) 2024-09-07 00:50:15 +00:00
expect [ONNX] Remove the operators test (#136335) 2024-09-25 17:44:18 +00:00
exporter [ONNX] Remove special handling of torchvision.ops imports in onnx export (#141569) 2024-11-28 18:05:40 +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] Enable experimental exporter logic to dynamo_export and support refine dynamic_shapes (#134976) 2024-09-06 01:29:56 +00:00
autograd_helper.py
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] Unify ONNXProgram and remove the old one (#136281) 2024-09-24 20:52:19 +00:00
pytorch_test_common.py [ONNX] Support FakeTensor in ONNXProgram (#135399) 2024-09-07 04:48:18 +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_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 [ONNX] Enable experimental exporter logic to dynamo_export and support refine dynamic_shapes (#134976) 2024-09-06 01:29:56 +00:00
test_fx_to_onnx.py [ONNX] Remove test_save_with_without_initializer test (#141263) 2024-11-21 22:06:15 +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 [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_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 Replace clone.detach with detach.clone (#140264) 2024-11-13 07:01:02 +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] Fix scaled_dot_product_attention with float scale (#135594) 2024-09-10 23:04:02 +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
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