• T
    Add multiclass_nms3 GPU kernel (#52401) · f71c805e
    Tian Zheng 提交于
    * Add GPU kernel for multiclass_nms3 op
    
    * Make multiclass_nms3 gpu kernel output consistent with cpu kernel
    
    * Fix API incompatibility
    
    * Fix unittests on builds without CUDA
    
    * Fix ROCM build
    
    * Remove fluid headers; Use default atol for unittest
    
    * Change function and variable naming
    
    * Add comments; Reduce redundant code
    
    * Use paddle test framework
    f71c805e
test_multiclass_nms_op.py 24.9 KB