• Z
    [Paddle-TRT] roi_align_plugin (#31732) · e3a38d79
    zlsh80826 提交于
    * add roi_align_plugin
    
    * add roi align unit_test
    
    * add roi align serialization
    
    * remove roi align static plugin because of batch dim issue
    
    * refine roi align unittest and add fp16/serialization
    
    * add trt roi align condition to op_teller
    
    * refine error message
    
    * remove unnecessary reshape layer
    e3a38d79
op_teller.cc 9.9 KB