saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-11 17:48:34 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/onnxruntime/test/python
History
stevenlix 4f10024868
Fix shape inference issue in Gather op (#9147) 
* add initializer checker for Gather with 1D input

* Check if indices value exists

* Update symbolic_shape_infer.py

* add unit test

* Update symbolic_shape_infer.py

* Update symbolic_shape_infer.py
2021-09-28 22:46:12 -07:00
..
contrib_ops Adding TorchEmbedding contrib op (#7136) 2021-03-30 14:33:25 -07:00
quantization Add QDQ for output of node (#9134) 2021-09-27 16:55:23 -07:00
testdata Adding TorchEmbedding contrib op (#7136) 2021-03-30 14:33:25 -07:00
transformers Fuse attention for BERT without num_heads, hidden_size (#9108) 2021-09-20 12:14:33 -07:00
helper.py Introduce training changes. 2020-03-11 14:39:03 -07:00
onnx_backend_test_series.py Update migraphx to rocm4.2 (#7994) 2021-06-22 13:39:51 -07:00
onnxruntime_test_ort_trainer.py Remove previous temporary fixes and address TODOs (#9020) 2021-09-13 10:10:07 -07:00
onnxruntime_test_ort_trainer_with_mixed_precision.py Remove unused imports from Python tests. (#5405) 2020-10-08 17:24:10 -07:00
onnxruntime_test_python.py Execution Provider Profiler (#8406) 2021-09-28 13:59:52 -07:00
onnxruntime_test_python_backend.py Remove unused imports from Python tests. (#5405) 2020-10-08 17:24:10 -07:00
onnxruntime_test_python_backend_mlops.py Remove unused imports from Python tests. (#5405) 2020-10-08 17:24:10 -07:00
onnxruntime_test_python_iobinding.py Remove unused imports from Python tests. (#5405) 2020-10-08 17:24:10 -07:00
onnxruntime_test_python_keras.py Remove unused imports from Python tests. (#5405) 2020-10-08 17:24:10 -07:00
onnxruntime_test_python_mlops.py Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
onnxruntime_test_python_nuphar.py [Nuphar] Add support for opset 14 (#8483) 2021-07-27 06:13:47 -07:00
onnxruntime_test_python_sparse_matmul.py Implement SparseTensor and infrastructure suppport and advance ONNX commit (#8038) 2021-07-22 15:24:36 -07:00
onnxruntime_test_python_symbolic_shape_infer.py Fix shape inference issue in Gather op (#9147) 2021-09-28 22:46:12 -07:00
onnxruntime_test_training_unit_tests.py Remove unused imports from Python tests. (#5405) 2020-10-08 17:24:10 -07:00
onnxruntime_test_training_unittest_utils.py Remove unused imports from Python tests. (#5405) 2020-10-08 17:24:10 -07:00
test_pytorch_export_contrib_ops.py Include pytorch_export_contrib_ops in inference builds (#8878) 2021-09-02 14:26:58 -07:00