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
Zhaolong Xing 35148d17f7
[BUG]: Head number can only be > 1 on multihead op (#23974)
6 years ago
..
CMakeLists.txt [Eernie TRT]: add slice op and add emb eltwise layernorm fp16 support (#23723) 6 years ago
elementwise_op_plugin.cu [Paddle-TRT]: add eltwise,pool2d, prelu, scale, concat, gelu dynamic shape support (#23396) 6 years ago
elementwise_op_plugin.h [Paddle-TRT]: add eltwise,pool2d, prelu, scale, concat, gelu dynamic shape support (#23396) 6 years ago
emb_eltwise_layernorm_plugin.cu [Eernie TRT]: add slice op and add emb eltwise layernorm fp16 support (#23723) 6 years ago
emb_eltwise_layernorm_plugin.h [Eernie TRT]: add slice op and add emb eltwise layernorm fp16 support (#23723) 6 years ago
gelu_op_plugin.cu [BUG]: Head number can only be > 1 on multihead op (#23974) 6 years ago
gelu_op_plugin.h add swish split gelu plugin dynamic support (#23305) 6 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 add TRT support for instance_norm op (#21928) 7 years ago
instance_norm_op_plugin.h fix trt instance_norm serialize bug. test=develop (#22152) 7 years ago
layer_norm_op_plugin.cu Add TRT support for BERT (#21135) 7 years ago
layer_norm_op_plugin.h Add TRT support for BERT (#21135) 7 years ago
pool_op_plugin.cu [Paddle-TRT]: add eltwise,pool2d, prelu, scale, concat, gelu dynamic shape support (#23396) 6 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 [Paddle-TRT]: add eltwise,pool2d, prelu, scale, concat, gelu dynamic shape support (#23396) 6 years ago
prelu_op_plugin.h [Paddle-TRT]: add eltwise,pool2d, prelu, scale, concat, gelu dynamic shape support (#23396) 6 years ago
qkv_to_context_plugin.cu [Paddle-TRT]: Ernie Dynamic shape support. (#23138) 6 years ago
qkv_to_context_plugin.h [Paddle-TRT]: Ernie Dynamic shape support. (#23138) 6 years ago
skip_layernorm_op_plugin.cu [Paddle-TRT]: Ernie Dynamic shape support. (#23138) 6 years ago
skip_layernorm_op_plugin.h [Paddle-TRT]: Ernie Dynamic shape support. (#23138) 6 years ago
slice_op_plugin.cu [BUG]: Head number can only be > 1 on multihead op (#23974) 6 years ago
slice_op_plugin.h [Eernie TRT]: add slice op and add emb eltwise layernorm fp16 support (#23723) 6 years ago
split_op_plugin.cu Optimize the error messages of paddle CUDA API (#23816) 6 years ago
split_op_plugin.h add swish split gelu plugin dynamic support (#23305) 6 years ago
swish_op_plugin.cu add swish split gelu plugin dynamic support (#23305) 6 years ago
swish_op_plugin.h add swish split gelu plugin dynamic support (#23305) 6 years ago
trt_plugin.cc Implement the Tensorrt plugin for elementwise op (#14487) 8 years ago
trt_plugin.h [Paddle-TRT]: Ernie Dynamic shape support. (#23138) 6 years ago
trt_plugin_factory.cc 5. add static trt load model 7 years ago
trt_plugin_factory.h add tensorrt support for windows (#19084) 7 years ago
trt_plugin_utils.h 6. delete useless predictor id 7 years ago