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 fdbe114b12
[Paddle-TRT] use float constant instead of double test=develop (#24544)
5 years ago
..
CMakeLists.txt [Eernie TRT]: add slice op and add emb eltwise layernorm fp16 support (#23723) 5 years ago
elementwise_op_plugin.cu [Paddle-TRT]: add eltwise,pool2d, prelu, scale, concat, gelu dynamic shape support (#23396) 5 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 [Eernie TRT]: add slice op and add emb eltwise layernorm fp16 support (#23723) 5 years ago
emb_eltwise_layernorm_plugin.h [Eernie TRT]: add slice op and add emb eltwise layernorm fp16 support (#23723) 5 years ago
gelu_op_plugin.cu [Paddle-TRT] use float constant instead of double test=develop (#24544) 5 years ago
gelu_op_plugin.h add swish split gelu plugin dynamic support (#23305) 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 add TRT support for instance_norm op (#21928) 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 TRT support for BERT (#21135) 5 years ago
layer_norm_op_plugin.h Add TRT support for BERT (#21135) 5 years ago
pool_op_plugin.cu [Paddle-TRT]: add eltwise,pool2d, prelu, scale, concat, gelu dynamic shape support (#23396) 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 [Paddle-TRT]: add eltwise,pool2d, prelu, scale, concat, gelu dynamic shape support (#23396) 5 years ago
prelu_op_plugin.h [Paddle-TRT]: add eltwise,pool2d, prelu, scale, concat, gelu dynamic shape support (#23396) 5 years ago
qkv_to_context_plugin.cu [Paddle-TRT]: Ernie Dynamic shape support. (#23138) 5 years ago
qkv_to_context_plugin.h [Paddle-TRT]: Ernie Dynamic shape support. (#23138) 5 years ago
skip_layernorm_op_plugin.cu [Paddle-TRT]: Ernie Dynamic shape support. (#23138) 5 years ago
skip_layernorm_op_plugin.h [Paddle-TRT]: Ernie Dynamic shape support. (#23138) 5 years ago
slice_op_plugin.cu [BUG]: Head number can only be > 1 on multihead op (#23974) 5 years ago
slice_op_plugin.h [Eernie TRT]: add slice op and add emb eltwise layernorm fp16 support (#23723) 5 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
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) 7 years ago
trt_plugin.h [Paddle-TRT]: Ernie Dynamic shape support. (#23138) 5 years ago
trt_plugin_factory.cc 5. add static trt load model 6 years ago
trt_plugin_factory.h add tensorrt support for windows (#19084) 6 years ago
trt_plugin_utils.h 6. delete useless predictor id 6 years ago