Commit Graph

270 Commits (develop)

Author SHA1 Message Date
Shang Zhizhou 8084b7594b
fix batchnorm when inpu dims < 3 (#31933)
5 years ago
zlsh80826 64ee255ffd
[Paddle-TRT] yolobox (#31755)
5 years ago
zlsh80826 e3a38d790a
[Paddle-TRT] roi_align_plugin (#31732)
5 years ago
zlsh80826 bfb5cf5567
[Paddle-TRT] trt affine channel converter (#31628)
5 years ago
zlsh80826 01aa252624
[Paddle-TRT] multiclass nms (#31742)
5 years ago
Wilber f4d9212de2
trt plugin upgrade to pluginv2ext (#31670)
5 years ago
Shang Zhizhou 9d04ef7369
fix tensorrt output varible reshape (#31733)
5 years ago
zlsh80826 bfced39eb6
[Paddle-TRT] nearest_interp op (#31626)
5 years ago
zlsh80826 fe241fd02f
[Paddle-TRT] gather converter (#31640)
5 years ago
zlsh80826 4ea3427865
[Paddle-TRT] support batch axis concatenation when using dynamic shape (#31627)
5 years ago
Shang Zhizhou 50ac7dbfd0
Trt elementwise plugin serialize (#31587)
5 years ago
Shang Zhizhou f57739be35
fix ernie_varlen when cutting head (#31497)
5 years ago
Pei Yang 32211fe9c4
TRT conv2d converter support SAME padding (#31379)
5 years ago
Shang Zhizhou 77c44e2f1b
change prelu plugin to tensorRT layer (#30210)
5 years ago
Pei Yang 2e9e3fad15
add n-d input support for trt scale converter (#31316)
5 years ago
Qi Li 72d99c5dcd
[ROCM] update fluid operators for rocm (part4), test=develop (#31225)
5 years ago
Pei Yang 00b09e86ac
[Paddle-TRT] support group_norm (#31040)
5 years ago
Shang Zhizhou a5c56d83a1
update trt int8 calibrator to IEntropyCalibratorV2 (#31060)
5 years ago
Wilber 01ccfbcde9
update trt error message when input height or width is -1 (#31019)
5 years ago
Pei Yang 9b54fe4154
add trt transpose and flatten converter (#31022)
5 years ago
Shang Zhizhou e6095bc2ce
fix split trt plugin initialize (#30875)
5 years ago
wanghuancoder 35c5b23f68
use iwyu clean include second time, test=develop (#30829)
5 years ago
Shang Zhizhou b909450994
fix trt plugin clone and initialize bugs in TRT7.1+ (#30709)
5 years ago
Shang Zhizhou ae0f88a988
add DLA support:C++&&Python api (#30165)
5 years ago
Leo Chen 81217a94d8
unify calling cudaSetDevice (#30470)
5 years ago
alncat 7bbf3ac5ab
Added support for inference using quantization aware trained dygraph (#30288)
6 years ago
WeiXin 66dc4ac77b
modify error message based on comments (#30189)
6 years ago
Wilber 01a287bf0a
fix windows compile when WITH_PYTHON=ON and WITH_TENSORRT=ON (#30194)
6 years ago
Pei Yang 2480bdef6c
change hard_swish from plugin to layer (#29177)
6 years ago
Pei Yang f860de4af7
support clip op trt converter (#29411)
6 years ago
Shang Zhizhou b9e76a0103
detect tensorRT plugin fp16 in runtime (#27933)
6 years ago
Pei Yang 994673bf4f
change avg pooling and global pooling to trt layer in dynamic shape mode (#28702)
6 years ago
Shang Zhizhou 8699f38d08
裁剪transformer模型trt支持;修复tensorRT不支持DeletePass的bug (#28517)
6 years ago
Shang Zhizhou ea851796e5
TensorRT中ernie模型推理性能优化,支持变长输入 (#28367)
6 years ago
Pei Yang 602d2ce5c9
change avg pooling from trt plugin to trt layer (#28032)
6 years ago
Shang Zhizhou bbc837ee72
add info log for trt input dynamic shape check (#27796)
6 years ago
Pei Yang ae6e40a7fd
Add unittests and OP version registry for tensorrt_subgraph_pass (#27544)
6 years ago
wanghuancoder df43905f12
use iwyu clean include (#27267)
6 years ago
Chen Weihang 765064476b
Polish some lost invalid error message (#27445)
6 years ago
Pei Yang fda54c0212
errmsg refine of trt plugin (#27309)
6 years ago
Pei Yang a5ef246cac
Optimize emb_eltwise_layernorm_plugin and support fp16 (#27128)
6 years ago
Shang Zhizhou 47fdc60ecc
Optimize slice trt plugin (#26970)
6 years ago
Shang Zhizhou e6e2e53782
Optimize error report (#27254)
6 years ago
Pei Yang aae41c6fca
refine error message related to paddle-TRT (#27256)
6 years ago
Zhaolong Xing 932bbe955b
fix pool trt plugin bug (#26463)
6 years ago
zlsh80826 ad6e3dd69c
[Paddle-TRT] Stack op plugin (#25605)
6 years ago
Pei Yang 78a530c219
[Paddle-TRT] TRT dynamic shape support PaddleSlim quant models (#26536)
6 years ago
zlsh80826 ac63c7cdef
fix a skip_layernorm bug, test=develop (#26800)
6 years ago
Pei Yang e3f8e5cf5c
trt int8 support conv2d_transpose (#26636)
6 years ago
Pei Yang 379222c3f1
add output scale and trt op teller support for hard_swish and hard_sigmoid (#26499)
6 years ago