saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-19 21:32:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/onnxruntime/test/python
History
Zhang Lei fba09faf5b
Implement reuse past and present tensor in Attention Ops. (#13791) 
Implement reuse kv_cache past and present tensor in Attention Ops. 
Unit test for abover feature.
Utilize the reuse kv_cache for past and present tensor in Greedy Search.
Correctness test for it.

Co-authored-by: Zhang Lei <phill.zhang@gmail.com>
2022-12-18 10:03:53 -08:00
..
contrib_ops ATen Fallback for Inference (#11597) 2022-06-09 16:07:30 +08:00
quantization Cjian/where python operator (#12795) 2022-12-12 13:27:47 -08:00
testdata
transformers Implement reuse past and present tensor in Attention Ops. (#13791) 2022-12-18 10:03:53 -08:00
helper.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
onnx_backend_test_series.py prepare test folder from GitHub (#12220) 2022-07-20 22:01:08 -07:00
onnxruntime_test_ort_trainer.py Set black's target version (#11370) 2022-04-27 14:52:19 -07:00
onnxruntime_test_ort_trainer_with_mixed_precision.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
onnxruntime_test_python.py Allow CUDA EP enable or disable TunableOp via session options and environment variable (#13601) 2022-11-15 14:43:54 +08:00
onnxruntime_test_python_backend.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
onnxruntime_test_python_backend_mlops.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
onnxruntime_test_python_cudagraph.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
onnxruntime_test_python_iobinding.py Move OrtValueVector from onnxruntime-training to onnxruntime (#11176) 2022-06-15 09:36:28 +02:00
onnxruntime_test_python_keras.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
onnxruntime_test_python_mlops.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
onnxruntime_test_python_sparse_matmul.py Move OrtValueVector from onnxruntime-training to onnxruntime (#11176) 2022-06-15 09:36:28 +02:00
onnxruntime_test_python_symbolic_shape_infer.py skip some models failed in dynamic shape infer (#13400) 2022-10-25 01:48:46 +08:00
onnxruntime_test_python_tvm.py [TVM EP][CI] Added TVMso EP testing into CI (#12188) 2022-07-19 16:05:28 +02:00
onnxruntime_test_training_unit_tests.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
onnxruntime_test_training_unittest_utils.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
test_pytorch_export_contrib_ops.py fix(onnx_export): Extract arg value from torch Value (#11471) 2022-05-11 11:36:43 -07:00