pytorch/test/onnx
2022-12-14 01:45:31 +00:00
..
assets [ONNX] add test images to repo (#63717) 2021-08-23 07:43:21 -07:00
expect [ONNX] Add repro export from GraphInfo (#89947) 2022-12-12 19:13:39 +00:00
internal [ONNX] Add setType from user into InferredType and Reliable in ConstantValueMap (#88622) 2022-11-19 17:16:59 +00:00
model_defs [ONNX] Enable script tests based on recent exporter changes (#77254) 2022-06-08 17:18:49 +00:00
autograd_helper.py [ONNX] Format ONNX python with black 2022-05-05 00:19:22 +00:00
debug_embed_params.py Apply ufmt linter to all py files under test/onnx (#81335) 2022-07-15 18:51:38 +00:00
onnx_test_common.py [ONNX] Use VerificationOptions to wrap option arguments (#89807) 2022-12-09 23:49:51 +00:00
pytorch_helper.py [ONNX] Modernize python syntax (#77935) 2022-05-24 22:52:37 +00:00
pytorch_test_common.py [ONNX] Create common ExportTestCase base class (#88145) 2022-11-10 21:51:59 +00:00
test_autograd_funs.py [ONNX] Create common ExportTestCase base class (#88145) 2022-11-10 21:51:59 +00:00
test_custom_ops.py Disable BUILD_CAFFE2 from ONNX builds (#90475) 2022-12-09 18:02:48 +00:00
test_export_modes.py [ONNX] Create common ExportTestCase base class (#88145) 2022-11-10 21:51:59 +00:00
test_models_onnxruntime.py Call _sdp_attention in nn.functional.mha (#89470) 2022-12-02 19:46:22 +00:00
test_models_quantized_onnxruntime.py [ONNX] Add quantized model tests to CI (#80398) 2022-07-28 21:25:29 +00:00
test_models.py [ONNX] Create common ExportTestCase base class (#88145) 2022-11-10 21:51:59 +00:00
test_onnx_opset.py [ONNX] Create common ExportTestCase base class (#88145) 2022-11-10 21:51:59 +00:00
test_onnxscript_no_runtime.py [ONNX] Add onnx-script test cases (#86907) 2022-11-16 23:57:25 +00:00
test_onnxscript_runtime.py [ONNX] Add onnx-script into ONNX docs (#89078) 2022-11-17 06:27:17 +00:00
test_operators.py Reland Dispatch torch.norm to linalg.vector_norm and linalg.matrix_norm (#81761) (#84624) 2022-11-22 07:53:24 +00:00
test_pytorch_jit_onnx.py [ONNX] Introduce ONNX reference evaluator for verification (#89808) 2022-12-10 01:29:12 +00:00
test_pytorch_onnx_no_runtime.py Disable BUILD_CAFFE2 from ONNX builds (#90475) 2022-12-09 18:02:48 +00:00
test_pytorch_onnx_onnxruntime_cuda.py Fix exception causes all over the codebase (#90271) 2022-12-07 04:29:00 +00:00
test_pytorch_onnx_onnxruntime.py [ONNX] Introduce ONNX reference evaluator for verification (#89808) 2022-12-10 01:29:12 +00:00
test_pytorch_onnx_shape_inference.py [ONNX] Add setType from user into InferredType and Reliable in ConstantValueMap (#88622) 2022-11-19 17:16:59 +00:00
test_symbolic_helper.py [ONNX] Fix check_training_mode in symbolic_helper (#78376) 2022-05-27 00:38:16 +00:00
test_utility_funs.py Fix missing return statement for test_it_returns_empty_list_when_model_contains_supported_inplace_ops in #89299 (#90797) 2022-12-14 01:45:31 +00:00
test_verification.py [ONNX] Add repro export from GraphInfo (#89947) 2022-12-12 19:13:39 +00:00
verify.py [ONNX] replace all _C._flatten to torch.jit._flatten (#83598) 2022-09-01 00:31:28 +00:00