Improve the performance of fake quantize OP (#40491)
* Move the computation of moving average scale to device * Use register to save local maximum in a thread
Showing
想要评论请 注册 或 登录
* Move the computation of moving average scale to device * Use register to save local maximum in a thread