• D
    Memory optimization of depthwise conv op and group norm op (#15313) · 9f8f0fc2
    Dun 提交于
    * mem opt
    
    * test=develop
    
    * test=develop
    
    * test=develop
    
    * test=develop
    
    * test=develop
    
    * test=develop
    
    * test=develop
    
    * refine code  test=develop
    
    * refine code  test=develop
    
    * refine code  test=develop
    
    * refine code  test=develop
    
    * refine with cub test=develop
    
    * fix mkldnn test && remove comments && test=develop
    
    * polish code && test=develop
    
    * add only_forward test && test=develop
    9f8f0fc2
depthwise_conv.cu 29.3 KB