1. 29 3月, 2021 1 次提交
    • 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 1 次提交
    • Z
      [Paddle-TRT] multiclass nms (#31742) · 01aa2526
      zlsh80826 提交于
      * add multiclass_nms
      
      * add multiclass_nms unittest
      
      * add default enable_tensorrt_oss option
      
      * refine multiclas nms unittest and add serialization/dynamic test
      
      * change super to InferencePassTest for python2 compatibility
      
      * refine multiclass nms unittest
      
      * move out dynamic shape test due to ci timelimit
      01aa2526
  3. 22 3月, 2021 1 次提交
    • 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
  4. 18 3月, 2021 1 次提交
  5. 24 2月, 2021 1 次提交
    • P
      [Paddle-TRT] support group_norm (#31040) · 00b09e86
      Pei Yang 提交于
      * add group norm plugin
      
      * fix compile problems
      
      * move concat axis check to trt op teller
      
      * add nbDims for scale and bias nv dims
      
      * add group norm unit test
      
      * fix unittest
      
      * add trt version restriction for group norm op teller
      
      * fix unittest
      00b09e86
  6. 18 2月, 2021 1 次提交
  7. 07 12月, 2020 1 次提交
  8. 03 11月, 2020 1 次提交
    • S
      TensorRT中ernie模型推理性能优化,支持变长输入 (#28367) · ea851796
      Shang Zhizhou 提交于
      * fp16 result ok
      
      * change -DWITH_NVINFER_PLUGIN toconfig.EnableTensorRtOSS
      
      * auto detect special slice op converter for ernie with trt oss
      
      * ernie oss only support fp16
      
      * fix special_slice_plugin serialize bug
      
      * matmul in tensorrt ok
      
      * ernie unittest ok
      
      * add matmul tensorrt unittest
      
      * remove demo code
      ea851796
  9. 01 9月, 2020 1 次提交
    • 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
  10. 19 4月, 2020 1 次提交
  11. 14 4月, 2020 1 次提交
  12. 12 4月, 2020 1 次提交
  13. 08 4月, 2020 2 次提交
  14. 26 3月, 2020 1 次提交
    • Z
      [Paddle-TRT]: Ernie Dynamic shape support. (#23138) · 430b0099
      Zhaolong Xing 提交于
      * add dynamic plugin support.
      test=develop
      
      * change emb eltwise layernorm to math function
      test=develop
      
      * add emb eltwise layernorm
      test=develop
      
      * can run dynamic shape ernie
      test=develop
      
      * fix ci
      test=develop
      
      * add ut for trt ernie dynamic
      
      test=develop
      
      * refine dynamic shape c++ interface.
      test=develop
      
      * fix comments
      test=develop
      
      * fix comments
      test=develop
      430b0099
  15. 14 1月, 2020 1 次提交
  16. 07 1月, 2020 1 次提交
  17. 06 1月, 2020 1 次提交
  18. 24 7月, 2019 1 次提交
    • Z
      Update trt5 for paddle-trt (#18645) · 26ae6d49
      Zhaolong Xing 提交于
      * update paddle-trt for:
          1. fix bug: when batch > 2, core in split plugin.
          2. add leaky_relu trt5.0 support (yolov3 from 65ms to 42ms.)
          3. add new attr to dropout.
          4. shuffle channel, swish, relu6 support
          test=develop
      
      * 1. fix ci
      test=develop
      26ae6d49
  19. 23 5月, 2019 1 次提交
  20. 21 5月, 2019 1 次提交
  21. 21 11月, 2018 1 次提交
  22. 20 11月, 2018 1 次提交
  23. 19 11月, 2018 1 次提交
  24. 16 11月, 2018 3 次提交
  25. 13 11月, 2018 2 次提交
  26. 12 10月, 2018 1 次提交
  27. 17 9月, 2018 1 次提交
  28. 24 8月, 2018 1 次提交
  29. 17 8月, 2018 1 次提交
  30. 09 8月, 2018 1 次提交
  31. 08 8月, 2018 1 次提交
  32. 27 7月, 2018 1 次提交
  33. 25 7月, 2018 1 次提交
  34. 24 7月, 2018 2 次提交
  35. 23 7月, 2018 1 次提交