• H
    Quant op dev (#25932) · 02606d45
    huangxu96 提交于
    * Finished ChannelWiseQuantDequantAbsMaxOp and Passed unittests.
    
    * Finished channel-wise quantize strategy in imperative quantization.
    
    * Added Cuda code of ChannelWiseQuantDequantMaxAbsOP
    Add Cuda code of ChannelWiseQuantDequantMaxAbsOp
    
    * Add quant_axis for channel_wise quant.
    
    * fixed a bug in unnitests, which will not trigger axis = 1 case and cannot meet the coverage rate requirement.
    
    * Added some assert infomation and fixed some coding style mistakes.
    02606d45
test_imperative_qat.py 16.4 KB