You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Paddle/paddle/fluid/inference/tensorrt/plugin
zlsh80826 64ee255ffd
[Paddle-TRT] yolobox (#31755)
5 years ago
..
CMakeLists.txt [Paddle-TRT] yolobox (#31755) 5 years ago
elementwise_op_plugin.cu Trt elementwise plugin serialize (#31587) 5 years ago
elementwise_op_plugin.h Trt elementwise plugin serialize (#31587) 5 years ago
emb_eltwise_layernorm_plugin.cu fix tensorrt output varible reshape (#31733) 5 years ago
emb_eltwise_layernorm_plugin.h fix trt plugin clone and initialize bugs in TRT7.1+ (#30709) 5 years ago
gelu_op_plugin.cu detect tensorRT plugin fp16 in runtime (#27933) 5 years ago
gelu_op_plugin.h detect tensorRT plugin fp16 in runtime (#27933) 5 years ago
hard_swish_op_plugin.cu [Paddle-TRT] Add hard_sigmoid and hard_swish support(support MobilenetV3) (#23672) 6 years ago
hard_swish_op_plugin.h [Paddle-TRT] Add hard_sigmoid and hard_swish support(support MobilenetV3) (#23672) 6 years ago
instance_norm_op_plugin.cu fix trt plugin clone and initialize bugs in TRT7.1+ (#30709) 5 years ago
instance_norm_op_plugin.h fix trt plugin clone and initialize bugs in TRT7.1+ (#30709) 5 years ago
layer_norm_op_plugin.cu Add unittests and OP version registry for tensorrt_subgraph_pass (#27544) 6 years ago
layer_norm_op_plugin.h Add TRT support for BERT (#21135) 6 years ago
pool_op_plugin.cu [ROCM] update fluid operators for rocm (part4), test=develop (#31225) 5 years ago
pool_op_plugin.h [Paddle-TRT]: add eltwise,pool2d, prelu, scale, concat, gelu dynamic shape support (#23396) 6 years ago
prelu_op_plugin.cu fix trt plugin clone and initialize bugs in TRT7.1+ (#30709) 5 years ago
prelu_op_plugin.h fix trt plugin clone and initialize bugs in TRT7.1+ (#30709) 5 years ago
qkv_to_context_plugin.cu fix tensorrt output varible reshape (#31733) 5 years ago
qkv_to_context_plugin.h detect tensorRT plugin fp16 in runtime (#27933) 5 years ago
roi_align_op_plugin.cu [Paddle-TRT] roi_align_plugin (#31732) 5 years ago
roi_align_op_plugin.h [Paddle-TRT] roi_align_plugin (#31732) 5 years ago
skip_layernorm_op_plugin.cu fix tensorrt output varible reshape (#31733) 5 years ago
skip_layernorm_op_plugin.h fix trt plugin clone and initialize bugs in TRT7.1+ (#30709) 5 years ago
slice_op_plugin.cu detect tensorRT plugin fp16 in runtime (#27933) 5 years ago
slice_op_plugin.h detect tensorRT plugin fp16 in runtime (#27933) 5 years ago
special_slice_plugin.cu fix tensorrt output varible reshape (#31733) 5 years ago
special_slice_plugin.h TensorRT中ernie模型推理性能优化,支持变长输入 (#28367) 5 years ago
split_op_plugin.cu trt plugin upgrade to pluginv2ext (#31670) 5 years ago
split_op_plugin.h trt plugin upgrade to pluginv2ext (#31670) 5 years ago
stack_op_plugin.cu detect tensorRT plugin fp16 in runtime (#27933) 5 years ago
stack_op_plugin.h detect tensorRT plugin fp16 in runtime (#27933) 5 years ago
swish_op_plugin.cu detect tensorRT plugin fp16 in runtime (#27933) 5 years ago
swish_op_plugin.h detect tensorRT plugin fp16 in runtime (#27933) 5 years ago
test_split_plugin.cc trt plugin upgrade to pluginv2ext (#31670) 5 years ago
trt_plugin.cc trt plugin upgrade to pluginv2ext (#31670) 5 years ago
trt_plugin.h trt plugin upgrade to pluginv2ext (#31670) 5 years ago
trt_plugin_factory.cc use iwyu clean include second time, test=develop (#30829) 5 years ago
trt_plugin_factory.h use iwyu clean include (#27267) 6 years ago
trt_plugin_utils.h Polish some lost invalid error message (#27445) 6 years ago
yolo_box_op_plugin.cu [Paddle-TRT] yolobox (#31755) 5 years ago
yolo_box_op_plugin.h [Paddle-TRT] yolobox (#31755) 5 years ago