“8f156fd7f2c4cff08a2b93196d288bfc15514c0d”上不存在“test/ir/inference/test_trt_convert_grid_sampler.py”
  • 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
cudnn_helper.h 7.2 KB