1. 29 3月, 2021 7 次提交
    • W
      modify API nn.Bilinear's doc (#31889) · b48841ba
      wanghuancoder 提交于
      * modify API nn.Bilinear's doc, test=develop
      
      * modify API nn.Bilinear's doc, test=develop
      b48841ba
    • L
      525c32e3
    • R
      123949eb
    • S
      fix cmake model path (#31866) · 61805d8f
      Shang Zhizhou 提交于
      * fix cmake model path
      
      * update cmake
      
      * fix unittest
      
      * fix unittest
      61805d8f
    • J
      [CustomOP] Add shape related constructor for Tensor (#31681) · 51eb29de
      Jiabin Yang 提交于
      * give shape related contructor and reshape warning
      
      * change line num to fit ut
      
      * change ut to fit
      
      * remove useless code
      
      * call resize directly in constructor
      51eb29de
    • 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