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
Pei Yang ae6e40a7fd
Add unittests and OP version registry for tensorrt_subgraph_pass (#27544)
4 years ago
..
CMakeLists.txt [Paddle-TRT] Stack op plugin (#25605) 5 years ago
elementwise_op_plugin.cu refine error message related to paddle-TRT (#27256) 4 years ago
elementwise_op_plugin.h [Paddle-TRT]: add eltwise,pool2d, prelu, scale, concat, gelu dynamic shape support (#23396) 5 years ago
emb_eltwise_layernorm_plugin.cu Optimize emb_eltwise_layernorm_plugin and support fp16 (#27128) 4 years ago
emb_eltwise_layernorm_plugin.h Optimize emb_eltwise_layernorm_plugin and support fp16 (#27128) 4 years ago
gelu_op_plugin.cu Fix registering trt plugin (#25744) 5 years ago
gelu_op_plugin.h Fix registering trt plugin (#25744) 5 years ago
hard_swish_op_plugin.cu [Paddle-TRT] Add hard_sigmoid and hard_swish support(support MobilenetV3) (#23672) 5 years ago
hard_swish_op_plugin.h [Paddle-TRT] Add hard_sigmoid and hard_swish support(support MobilenetV3) (#23672) 5 years ago
instance_norm_op_plugin.cu fix trt instance norm plugin on gcc8. test=develop (#25730) 5 years ago
instance_norm_op_plugin.h fix trt instance_norm serialize bug. test=develop (#22152) 5 years ago
layer_norm_op_plugin.cu Add unittests and OP version registry for tensorrt_subgraph_pass (#27544) 4 years ago
layer_norm_op_plugin.h Add TRT support for BERT (#21135) 5 years ago
pool_op_plugin.cu fix pool trt plugin bug (#26463) 5 years ago
pool_op_plugin.h [Paddle-TRT]: add eltwise,pool2d, prelu, scale, concat, gelu dynamic shape support (#23396) 5 years ago
prelu_op_plugin.cu fix dy shape bug in trt7.1 (#26273) 5 years ago
prelu_op_plugin.h fix dy shape bug in trt7.1 (#26273) 5 years ago
qkv_to_context_plugin.cu Fix registering trt plugin (#25744) 5 years ago
qkv_to_context_plugin.h Fix registering trt plugin (#25744) 5 years ago
skip_layernorm_op_plugin.cu Fix registering trt plugin (#25744) 5 years ago
skip_layernorm_op_plugin.h fix a skip_layernorm bug, test=develop (#26800) 5 years ago
slice_op_plugin.cu Optimize slice trt plugin (#26970) 4 years ago
slice_op_plugin.h Optimize slice trt plugin (#26970) 4 years ago
split_op_plugin.cu Optimize the error messages of paddle CUDA API (#23816) 5 years ago
split_op_plugin.h add swish split gelu plugin dynamic support (#23305) 5 years ago
stack_op_plugin.cu [Paddle-TRT] Stack op plugin (#25605) 5 years ago
stack_op_plugin.h [Paddle-TRT] Stack op plugin (#25605) 5 years ago
swish_op_plugin.cu add swish split gelu plugin dynamic support (#23305) 5 years ago
swish_op_plugin.h add swish split gelu plugin dynamic support (#23305) 5 years ago
trt_plugin.cc Implement the Tensorrt plugin for elementwise op (#14487) 6 years ago
trt_plugin.h Fix TRT plugin registry without TRT lib (#25982) 5 years ago
trt_plugin_factory.cc Polish some lost invalid error message (#27445) 4 years ago
trt_plugin_factory.h use iwyu clean include (#27267) 4 years ago
trt_plugin_utils.h Polish some lost invalid error message (#27445) 4 years ago