onnxruntime/onnxruntime/python/tools
Chi Lo 945fae8f56
Lochi/quantization tool for trt (#6103)
* Initial implementation of generating calibration dynamic range table

* Initialize validation support for Quantization

* Initialize validation support for Quantization (cont.)

* Improve validation support for Quantization

* Improve validation support for Quantization

* Rewrite/Refine for calibration and validation

* Rewrite/Refine for calibration and validation (cont.)

* Refine code

* Refine code

* Add data reader for BERT

* Add flatbuffers to serialize calibration table

* Refine code and add BERT evaluation

* Refine the code

* minor modification

* Add preprocess/postprocess of vision team yolov3 and refine the code

* Update annotation

* Make bbox cooridates more accurate

* Fix bug

* Add support of batch processing

* Batch processing for model zoo yolov3

* Add batch inference for evaluation

* Refine the code

* Add README

* Add comments

* Refine the code for PR

* Remove batch support checking in data_reader and refine the code

* Refine the code for PR

* Refine the code for PR review

Co-authored-by: Olivia Jain <oljain@microsoft.com>
2020-12-21 20:59:08 -08:00
..
featurizer_ops Update data_frame_tool to latest (#3919) 2020-05-18 21:13:56 -07:00
quantization Lochi/quantization tool for trt (#6103) 2020-12-21 20:59:08 -08:00
tensorrt/perf Add Status.csv to EP Perf Tool (#6167) 2020-12-21 20:23:19 -08:00
transformers Model Fusion For Bart (#6105) 2020-12-15 14:30:15 -08:00
__init__.py Enable running PEP8 on python scripts using flake8 (#3928) 2020-05-15 07:15:06 +10:00
onnxruntime_test.py Improve handling of symbolic dimensions in the onnxruntime_test.py script. (#3959) 2020-05-18 16:51:09 +10:00
symbolic_shape_infer.py Symbolic shape inference: fix rank for ConstantOfShape (#5912) 2020-11-24 14:50:41 -08:00