• Z
    [Paddle-TRT] add anchor generator op plugin (#31730) · b807e408
    zlsh80826 提交于
    * add anchor generator op plugin
    
    * add anchor generator unit_test
    
    * remove dbg info
    
    * remove redundant line
    
    * replace assertion with paddle enforce
    
    * dynamic plugin replaces assertion with paddle enforce
    
    * anchor generator support dynamic shape on spatial axis
    
    * anchor generator test with fp16, dynamic shape
    
    * add anchor generator test all
    
    * add back main
    
    * reduce test input size to not exceed the timelimit of ci
    
    * change super to InferencePassTest for python2 compatibility
    
    * reuse paddle operator anchor generator
    
    * move creator construct to header with default
    
    * add cuda ifdef
    
    * reduce line
    
    * change super to InferencePassTest for python2 compatibility
    
    * fix anchor generator fp16 serialize setting
    
    * split unittest from test_all
    
    * restrict anchor generator input format before version 7234
    
    * anchor generator only support greater than trt7.1
    
    * change min_graph_size to 2
    
    * min_graph size to 3 if dynamic shape
    
    * reduce dynamic shape size to avoid trt search tactic too long to exceed time limit
    
    * remove anchor from fetch list
    
    * anchor generator support all trt version
    
    * fix memory not allocated but if serialized
    b807e408
op_teller.cc 10.3 KB