onnxruntime/onnxruntime/python
Tianlei Wu ecbacd7d79
Add Benchmark of GPT2 CPU inference (#3351)
* Add benchmark script and notebook for GPT2
* Update Reshape fusion for GPT2 model
* Add opt_level option for bert_model_optimization to disable onnxruntime by setting --opt_level 0
* Fix keras optimization
2020-03-31 13:43:09 -07:00
..
backend Use yapf to format python (#3276) 2020-03-20 14:34:10 -07:00
datasets Validate input shapes. (#1352) 2019-07-19 13:42:34 -07:00
tools Add Benchmark of GPT2 CPU inference (#3351) 2020-03-31 13:43:09 -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 Support non-sequence tensor fed through as a python list (#2782) 2020-01-20 09:45:10 +10: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 Thread pool changes (#3153) 2020-03-30 12:18:40 -07:00
onnxruntime_validation.py Use yapf to format python (#3276) 2020-03-20 14:34:10 -07:00
pybind.def Fix a build warning in onnxruntime python extension (#416) 2019-02-01 00:19:41 -08:00
session.py Use yapf to format python (#3276) 2020-03-20 14:34:10 -07:00
version_script.lds Initial bootstrap commit. 2018-11-19 16:48:22 -08:00