• Z
    [Paddle-TRT] trt affine channel converter (#31628) · bfb5cf55
    zlsh80826 提交于
    * trt affine channel converter
    
    * add trt affine channel base test
    
    * add trt affine channel NHWC
    
    * remove asterisk for python2 compatibility
    
    * trt affine channel converter
    
    * add trt affine channel base test
    
    * add trt affine channel NHWC
    
    * remove asterisk for python2 compatibility
    
    * fix rebase
    
    * move LodTensor to Tensor
    
    * add dbg info
    
    * affine channel converter only support NCHW
    
    * scale,bias are parameters, use create_parameters api
    
    * reduce test input size to not exceed the timelimit of ci
    
    * refine affine channel unittest and add serialization/dynamic test
    
    * change super to InferencePassTest for python2 compatibility
    
    * change super to InferencePassTest for python2 compatibility
    
    * fix affine channel fp16 serialize setting
    bfb5cf55
affine_channel_op.cc 3.6 KB