• Z
    [Paddle-TRT] Stack op plugin (#25605) · ad6e3dd6
    zlsh80826 提交于
    * add stack_op to CMakeLists
    
    * add dim=3 support for scale op
    
    * add trt stack op, test=develop
    
    * remove debug message
    
    * add stack plugin serialize
    
    * remove slice, scale op, will add later
    
    * enhence error message
    
    * revise trt ernie test to conver the stack op CI testi, test=develop
    
    * add stack op serialization
    
    * fix test shape after adding stack op
    
    * remove slice op, will add after implementing serialization
    
    * roll back to min_graph=5 to avoid using slice op
    
    * fix scale op output layer
    
    * implement stack op createPlugin
    
    * use workspace and move the defination to .cu
    
    * move stack plugin creator definition to .cu, test=develop
    ad6e3dd6
stack_op_plugin.h 3.7 KB