Zhen Wang
|
8965819fbb
|
rewrite the cuda kernels of channel_wise_quant_op and channe_wise_dequant_op. test=develop
|
6 years ago |
|
Zhen Wang
|
ec88b6cc5a
|
add channel wise quantization in ir pass.
|
6 years ago |
achao2013
|
81b4fad8b9
|
add moving average absmax op and fix bug (#15155)
* Add moving average absmax op in quantilize-aware training.
|
6 years ago |
|
Zhen Wang
|
4ab0607634
|
fix NaN bugs.
|
6 years ago |
|
Zhen Wang
|
3cf0ee414d
|
update some details. test=develop
|
6 years ago |
|
Zhen Wang
|
0cb50bb983
|
avoid ce fails on windows.
|
6 years ago |
|
Zhen Wang
|
33f99d6197
|
add IrNode&IrVarNode&IrOpNode. test=develop
|
6 years ago |
|
WangZhen
|
28dfad5e27
|
fix some bugs about python3. test=develop
|
6 years ago |
|
WangZhen
|
a7efab7ec1
|
add comments for public API. test=develop
|
6 years ago |
|
WangZhen
|
0db41a9c44
|
add op_role attr when creating op node.
|
6 years ago |
|
WangZhen
|
c67b29c178
|
fix some bugs of graph.to_program and get_pass.
|
6 years ago |
|
WangZhen
|
c64f22048a
|
add convert_to_int8 pass and transform_for_mobile pass and their UTs.
|
6 years ago |
|
WangZhen
|
c8095eeb82
|
add freeze pass, and UT is passed.
|
6 years ago |
|
WangZhen
|
dde19a0ff8
|
add quantization freeze pass.
|
6 years ago |
