1. 29 3月, 2021 2 次提交
    • 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
    • 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
  2. 26 3月, 2021 5 次提交
  3. 25 3月, 2021 5 次提交
  4. 24 3月, 2021 8 次提交
  5. 23 3月, 2021 12 次提交
  6. 22 3月, 2021 3 次提交
    • Z
      [Paddle-TRT] nearest_interp op (#31626) · bfced39e
      zlsh80826 提交于
      * nearest_interp op converter w/ dynamic/static
      
      * fix data_layout include
      
      * add trt nearest unit_test
      
      * add nearest_interp NHWC test
      
      * update trt nearest interp nhwc testcase
      
      * remove asterisk for python2 compatibility
      
      * add empty line to prevent conflict
      
      * nearest_interp op converter w/ dynamic/static
      
      * fix data_layout include
      
      * add trt nearest unit_test
      
      * add nearest_interp NHWC test
      
      * update trt nearest interp nhwc testcase
      
      * remove asterisk for python2 compatibility
      
      * add empty line to prevent conflict
      
      * change the priority of out_h, out_w
      bfced39e
    • A
      [oneDNN] Initial bf16 amp integration (#31093) · 7ccf6b60
      arlesniak 提交于
      7ccf6b60
    • L
      [3D-parallel] add 1f1b scheduler for pipeline (#31566) · a501a7b0
      lilong12 提交于
      * add 1f1b scheduler for pp, test=develop
      a501a7b0
  7. 21 3月, 2021 3 次提交
  8. 19 3月, 2021 2 次提交