1. 20 6月, 2022 1 次提交
  2. 17 6月, 2022 1 次提交
  3. 16 6月, 2022 1 次提交
    • Z
      [Inference] add squeeze2/unsqueeze2 trt layer (#42782) · 8bc1c82d
      zhoutianzi666 提交于
      * add squeeze2
      
      * add squeeze
      
      * add squeeze2,unsqueeze2
      
      * merge develop
      
      * fix format
      
      * add conditions for squeeze2 and unsqueeze in op_teller
      
      * merge develop
      
      * add squeeze unsqueeze
      
      * add squeeze unsqueeze
      
      * add squeeze unsqueeze
      
      * remove unsqueeze2_eltwise_fuse_pass
      
      * add squeeze/unsqueeze
      8bc1c82d
  4. 13 6月, 2022 1 次提交
  5. 09 6月, 2022 1 次提交
  6. 04 6月, 2022 1 次提交
  7. 02 6月, 2022 1 次提交
  8. 20 5月, 2022 1 次提交
  9. 17 5月, 2022 1 次提交
  10. 12 5月, 2022 1 次提交
  11. 13 4月, 2022 1 次提交
    • F
      init roll convert (#41689) · 14c3c450
      feng_shuai 提交于
      * init roll convert
      
      * add ut for roll convert
      
      * roll convert don't support trt6.0
      
      * fix: change ut for trt 7.0.0.1
      14c3c450
  12. 12 4月, 2022 1 次提交
    • F
      strided_slice (#41573) · b861022a
      feng_shuai 提交于
      * strided_slice
      
      * fix: compiler error because of size()
      
      * fix: warning
      
      * fix : warning
      
      * init input_shape
      
      * fix:forget punctuation
      b861022a
  13. 31 3月, 2022 1 次提交
  14. 16 2月, 2022 1 次提交
  15. 14 1月, 2022 1 次提交
    • H
      add flatten_contiguous_range OpConvert for Paddle-TRT (#38922) · 050aa6fe
      heliqi 提交于
      * add trt_convert_flatten_contiguous_rang op
      
      * trt version >7,support trt_convert_flatten_contiguous_rang
      
      * trt version >7,support trt_convert_flatten_contiguous_rang
      
      * trt version >7,support trt_convert_flatten_contiguous_rang
      
      * test cast add trt version >=7 skip
      050aa6fe
  16. 27 10月, 2021 1 次提交
  17. 26 10月, 2021 1 次提交
  18. 19 10月, 2021 1 次提交
  19. 11 10月, 2021 1 次提交
    • W
      add mish trt plugin (#34123) · 2b7b752a
      wangxinxin08 提交于
      * add mish trt plugin, compile & install success, run error. test=develop
      * modify code according to review
      * add TRT_NOEXCEPT for mish trt plugin
      * add unittest for mish trt plugin
      * remove unnecessary check of mish in op_teller.cc
      * fix some problem of trt8
      * add check and modify unittest while converting mish to trt plugin
      Co-authored-by: Ndengkaipeng <dengkaipeng@baidu.com>
      2b7b752a
  20. 10 9月, 2021 1 次提交
    • W
      conv3d (#35507) · 42847d2e
      wenbin 提交于
      * conv3d
      
      * remove const_cast
      
      * modify ut
      
      * disable dynamic shape for trt6.0
      
      * remove trt5
      42847d2e
  21. 29 7月, 2021 1 次提交
    • W
      Tile supported (#34388) · cffa15c5
      wenbin 提交于
      * tile op
      
      * more uts
      
      * disable tile if trt6.0
      
      * typo
      
      * fix timeout issue
      
      * opteller
      
      * opteller remove duplicate code
      
      * comments.	test=document_fix
      
      * modify PADDLE_ENFORCE.
      
      * fix reduce_mean issue
      cffa15c5
  22. 05 6月, 2021 1 次提交
  23. 01 6月, 2021 1 次提交
  24. 01 4月, 2021 1 次提交
    • 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
  25. 30 3月, 2021 1 次提交
  26. 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
  27. 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
  28. 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
  29. 18 3月, 2021 1 次提交
  30. 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
  31. 18 2月, 2021 1 次提交
  32. 07 12月, 2020 1 次提交
  33. 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
  34. 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
  35. 19 4月, 2020 1 次提交
  36. 14 4月, 2020 1 次提交
  37. 12 4月, 2020 1 次提交
  38. 08 4月, 2020 2 次提交