onnxruntime/onnxruntime/python
kile0 f367fd921c
Use a custom allocator for temporary buffers in reduction_ops.cc (#2775)
* port the mimalloc allocator

* hook mimalloc opt into common.h and reduction ops

* repurpose USE_MIMALLOC to only denote subbing in of default allocator with mimalloc and some refactoring

* fix unintended cherry pick diffs

* polish alloctor_mimalloc

* explicitly disable mimalloc where it already had been disabled

* update mimalloc to pull in stl allocator

* switch mimalloc stl allocator to use mimalloc library version

* turn mimalloc on by default (only the stl changes are enabled, the python interacting ones are off already and shall remain so)

* move FastAllocVector into cpu specific code

* separate out defines into arena and stl changes

* the rest of the define renames

* bfc arena allocator

* some typos and rename the bfc arena allocator to fit existing class naming conventions

* adjustments in response to comments

* different template instantiations are friends
2020-02-23 16:04:30 +10: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 Bert optimization for onnx model exported by keras2onnx (#3014) 2020-02-15 23:59:49 -08: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 Use a custom allocator for temporary buffers in reduction_ops.cc (#2775) 2020-02-23 16:04:30 +10:00
onnxruntime_validation.py remove 64bit warning message from python validation. (#2727) 2019-12-23 17:06:55 -08:00
pybind.def Fix a build warning in onnxruntime python extension (#416) 2019-02-01 00:19:41 -08:00
session.py Allow providers to be set for InferenceSession at construction (#2606) 2019-12-11 10:24:43 -08:00
version_script.lds Initial bootstrap commit. 2018-11-19 16:48:22 -08:00