saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-23 19:32:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/onnxruntime/python
History
Tianlei Wu b0f8ec7a7d
Script for converting BERT model for performance optimization (#2037) 
* script for converting ONNX model for BERT performance optimization

* Remove code that not needed anymore.

* refine the script

* Support BERT model exported from PyTorch 1.3
Keep opset version
Exact match in Attention, Layer normalziation fusions.

* read batch_size from input model directly
2019-10-15 16:50:08 -07:00
..
backend python session.run() fallback to CPU/CUDA provider for EP failures. (#1960) 2019-10-02 02:38:03 -07:00
datasets Validate input shapes. (#1352) 2019-07-19 13:42:34 -07:00
tools Script for converting BERT model for performance optimization (#2037) 2019-10-15 16:50:08 -07:00
__init__.py Initial bootstrap commit. 2018-11-19 16:48:22 -08:00
_ld_preload.py Use manylinux2010 image to build linux python wheels (#1282) 2019-06-27 15:45:06 -07:00
_pybind_state.py Use manylinux2010 image to build linux python wheels (#1282) 2019-06-27 15:45:06 -07:00
exported_symbols.lst Initial bootstrap commit. 2018-11-19 16:48:22 -08:00
onnxruntime_pybind_exceptions.h Remove ml_status.h, add StatusCode to pybind exception mappings (#1889) 2019-09-24 11:13:14 -07:00
onnxruntime_pybind_mlvalue.cc Add support for output seq(tensor) in python binding and test framework. Implement SequenceConstruct, SequenceEmpty, SequenceInsert and SequenceErase ops. (#2040) 2019-10-10 15:58:49 -07:00
onnxruntime_pybind_mlvalue.h Add support for output seq(tensor) in python binding and test framework. Implement SequenceConstruct, SequenceEmpty, SequenceInsert and SequenceErase ops. (#2040) 2019-10-10 15:58:49 -07:00
onnxruntime_pybind_state.cc Introduce execution mode enum for clarity and extensibility; Change Python, C and C# APIs accordingly; Removed EnableSequentialExecution, DisableSequentialExecution in favor of the more general SetExecutionModeAPI. (#2098) 2019-10-14 09:48:19 -07:00
onnxruntime_validation.py Fixes #31, add option numpy_version, skip_keras_test to the parser of build.py, add flag PRIVATE for the python bindings (#544) 2019-03-07 13:08:02 +01:00
pybind.def Fix a build warning in onnxruntime python extension (#416) 2019-02-01 00:19:41 -08:00
session.py python session.run() fallback to CPU/CUDA provider for EP failures. (#1960) 2019-10-02 02:38:03 -07:00
version_script.lds Initial bootstrap commit. 2018-11-19 16:48:22 -08:00