“b1b13f8f5494b7ccd38f8b095c74d2d8172e2d9d”上不存在“paddle/fluid/framework/init_test.cc”
  • 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