• Q
    Cudnn batch norm op (#5067) · 56b723c4
    Qiao Longfei 提交于
    * init cudnn batch norm op
    
    * rename batch_norm_cudnn_op.cc batch_norm_op.cu
    
    * correct name style
    
    * add ExtractNCWHD, simplify code
    
    * fix ExtractNCWHD
    
    * use CUDNN_ENFORCE instead of PADDLE_ENFORCE
    56b723c4
cudnn.h 5.8 KB