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 b717895f64
Fix registering trt plugin (#25744)
5 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 Fix registering trt plugin (#25744) 5 years ago
emb_eltwise_layernorm_plugin.h Fix registering trt plugin (#25744) 5 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) 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 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) 6 years ago
layer_norm_op_plugin.cu Add TRT support for BERT (#21135) 6 years ago
layer_norm_op_plugin.h Add TRT support for BERT (#21135) 6 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 fix bug of prelu when rank not equal 4, test=develop (#25067) 5 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 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 registering trt plugin (#25744) 5 years ago
slice_op_plugin.cu [Paddle-TRT] slice kernel optimization (#24783) 5 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) 7 years ago
trt_plugin.h Fix registering trt plugin (#25744) 5 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) 6 years ago
trt_plugin_utils.h Fix registering trt plugin (#25744) 5 years ago