Commit Graph

9 Commits (8439384e257f6a69b6afe5d1944cdf1a0c4af3c2)

Author SHA1 Message Date
Huihuang Zheng 12542320c5
Replace TemporaryAllocator by CUDADeviceContextAllocator () 6 years ago
Zhaolong Xing ff7f911b4d
add quant_dequant_moving_avg_max_abs op () 6 years ago
Zhen Wang a914d9b116
Quant output scale () 6 years ago
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 () 6 years ago
Zhen Wang 545247d7b4 add channel wise quantize op. 6 years ago
qingqing01 9bd933d3fb
Improve and fix fake_quantize_op () 7 years ago
achao2013 8e4b225fe4 Add fake_quantize_op. () 7 years ago