Commit Graph

26 Commits (952508527a94de94a78c9250545dcd2d37d21ff4)

Author SHA1 Message Date
juncaipeng 00b11a4a1e
Support more ops in post training quantization, test=develop (#21073)
6 years ago
juncaipeng fa522dffa0
Fix bug in add_quant_dequant_pass, test=develop (#21018)
6 years ago
juncaipeng 175ba39c03
Add post_training_quantization (#20800)
6 years ago
juncaipeng f201b465ec
Move pool2d to add_quant_dequant_pass, test=develop (#20586)
7 years ago
juncaipeng b0ceed6fb4
add fake_quant_dequant_op for average pool2d, test=develop (#19880)
7 years ago
Zhen Wang 0fe72469ea
Add the max-pool2d quantization support and the partial quantization support. (#19310)
7 years ago
Zhen Wang 3398f99608
Adding AddQuantDequantPass for TensorRT int8 (#17529)
7 years ago
Zhen Wang 65541d83b0
add scale pass for calculating the output scales.test=develop (#17259)
7 years ago
Zhen Wang a40121e4c8
fix the initialization process error. test=develop (#17213)
7 years ago
Zhen Wang 183bacebe3 clean codes and fix some bugs. test=develop
7 years ago
Zhen Wang 1c11f817e9 Use the resolve hazard method.
7 years ago
Zhen Wang 2ccbfd5e10 Fix some bugs for quantization passes.
7 years ago
Zhen Wang 8965819fbb rewrite the cuda kernels of channel_wise_quant_op and channe_wise_dequant_op. test=develop
7 years ago
Zhen Wang ec88b6cc5a add channel wise quantization in ir pass.
7 years ago
achao2013 81b4fad8b9 add moving average absmax op and fix bug (#15155)
7 years ago
Zhen Wang 7c8f7df2fe add some op_des funs to IrOpNode and add some var_des funs to IrVarNode. test=develop
7 years ago
Zhen Wang 33f99d6197 add IrNode&IrVarNode&IrOpNode. test=develop
7 years ago
WangZhen 28dfad5e27 fix some bugs about python3. test=develop
7 years ago
WangZhen a7efab7ec1 add comments for public API. test=develop
7 years ago
WangZhen 0db41a9c44 add op_role attr when creating op node.
7 years ago
WangZhen c64f22048a add convert_to_int8 pass and transform_for_mobile pass and their UTs.
7 years ago
WangZhen c8095eeb82 add freeze pass, and UT is passed.
7 years ago
WangZhen dde19a0ff8 add quantization freeze pass.
7 years ago
WangZhen 3b668c1574 Update some comments in the quantization transform pass. test=develop
7 years ago
WangZhen b913463e83 Update according to the reviewers' suggestion. test=develop
7 years ago
WangZhen 59e5cc51d6 Add quantization transform pass and UT.
7 years ago