onnxruntime/onnxruntime/python
Tianlei Wu 268d2283c0
Export GPT-2 ONNX model without postion_ids and attention_mask inputs (#4852)
* Export GPT-2 ONNX model without postion_ids and attention_mask inputs
* allow benchmark_gpt2 on user's model
* refactor:  get_dummy_inputs returns a data class.
2020-08-24 13:05:25 -07:00
..
backend Remove ONNX from requirements.txt (#4073) 2020-05-29 21:44:20 +05:30
datasets replace invalid sample (#4567) 2020-07-21 23:51:17 -07:00
tools Export GPT-2 ONNX model without postion_ids and attention_mask inputs (#4852) 2020-08-24 13:05:25 -07:00
training Enable running PEP8 on python scripts using flake8 (#3928) 2020-05-15 07:15:06 +10:00
__init__.py Enable running PEP8 on python scripts using flake8 (#3928) 2020-05-15 07:15:06 +10:00
_ld_preload.py Enable running PEP8 on python scripts using flake8 (#3928) 2020-05-15 07:15:06 +10:00
_pybind_state.py Enable running PEP8 on python scripts using flake8 (#3928) 2020-05-15 07:15:06 +10:00
exported_symbols.lst
onnxruntime_pybind_exceptions.h refactor frontend (#3235) 2020-03-19 20:59:41 -07:00
onnxruntime_pybind_mlvalue.cc Move memory info into IAllocator (#2850) 2020-06-22 11:18:52 +10:00
onnxruntime_pybind_mlvalue.h refactor prototypes into headers (#4337) 2020-06-26 12:02:14 -07:00
onnxruntime_pybind_state.cc Reduce ONNX dependency in minimal build (#4890) 2020-08-23 07:02:13 +10:00
onnxruntime_pybind_state_common.cc fix pybind issue introduced by merge 2020-03-23 23:23:34 +00:00
onnxruntime_pybind_state_common.h refactor prototypes into headers (#4337) 2020-06-26 12:02:14 -07:00
onnxruntime_validation.py Enable running PEP8 on python scripts using flake8 (#3928) 2020-05-15 07:15:06 +10:00
pybind.def
session.py Add Python API for specifying device options. (#4205) 2020-07-21 07:28:13 -07:00
version_script.lds