• Y
    Add cuda implementation for `prelu` backward pass (#18633) · ce1ec332
    Yang Zhang 提交于
    * Add GPU implementation for `prelu` backward pass
    
    test=develop
    
    * Fix logic error in `prelu` GPU backward and simplify a bit
    
    test=develop
    
    * Fix `prelu` backward CUDA implementation
    
    test=develop
    
    CPU version was not used actually, so test passed
    ce1ec332
prelu_op.cu 7.7 KB