• W
    Add support for new QAT models (#18970) · 4286a627
    Wojciech Uss 提交于
    * Add support for new QAT models
    
    test=develop
    Co-Authored-By: NMichał Gallus <michal.gallus@intel.com>
    Co-Authored-By: NWojciech Uss <wojciech.uss@intel.com>
    
    * fixed fps results
    
    test=develop
    
    * fix top5 accuracy drop problem
    
    * updated for new QAT models
    
    * skip quantizing average pooling - dirty but working
    
    * add missing pass
    
    * added missing conv+brelu fuse pass
    
    * removed a call to non-existent pass
    
    test=develop
    
    * renamed pass
    
    test=develop
    
    * Adjust finding pooling scale to newest QAT models
    
    * Remove unnecessary code from quantization_mkldnn_pass
    
    * Copy Pooling input scale to output scale in QAT
    
    * Refactor & remove unused code in QAT
    
    * Incorporate fp32 FC into QAT
    
    test=develop
    
    * Enable graph drawing with debug flag
    
    test=develop
    
    * Add tests for QATv2
    
    * Fix paths for QATv2 models
    
    test=develop
    
    * Add option to save transformed int8 qat model
    
    test=develop
    
    * Remove redundant lines from qat mkldnn pass
    
    test=develop
    
    * Delegate disablement of avg pooling to qat
    
    test=develop
    
    * fix CI bug, test=develop
    
    * Follow Wangzhen's Review, test=develop
    
    * Update API.spec
    
    test=develop
    
    * Name False in (is_unsigned, TensorScale) tuple
    
    test=develop
    4286a627
pybind.cc 70.7 KB