• Cudnn conv op (#4195) · a3ccbdb3
    武毅 提交于
    * add cudnn_conv_op
    
    * WIP
    
    * update
    
    * update
    
    * fix grad check
    
    * use platform::memory
    
    * add support group for cudnn
    
    * update
    
    * follow comments
    
    * fix onlycpu build
    
    * update cuda define
    
    * follow comments
    
    * follow comments
    
    * merge with updates
    
    * fix compile error
    
    * follow comments
    
    * follow comments
    a3ccbdb3
test_conv2d_op.py 4.2 KB