Fix cudnn bn param desc (#3253)
* use cudnnDeriveBNTensorDescriptor
* CudnnTensorDescHelper
* fix bn min eps
Co-authored-by: Nguo ran <360112263@qq.com>
Showing
想要评论请 注册 或 登录
* use cudnnDeriveBNTensorDescriptor
* CudnnTensorDescHelper
* fix bn min eps
Co-authored-by: Nguo ran <360112263@qq.com>