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.

Pei Yang 0a51098a71 Add TRT support for BERT (#21135) ... * add gelu plugin * align trt bert with gpu * add support for fused fc with relu, * add unittest for bert trt		6 years ago
..
CMakeLists.txt	Add TRT support for BERT (#21135)	6 years ago
elementwise_op_plugin.cu	5. add static trt load model	6 years ago
elementwise_op_plugin.h	5. add static trt load model	6 years ago
gelu_op_plugin.cu	Add TRT support for BERT (#21135)	6 years ago
gelu_op_plugin.h	Add TRT support for BERT (#21135)	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	Bug Fix: Paddle-TRT cannot handle adaptive pooling in pool2d op converter and "num" attribute in split op converter (#20733)	6 years ago
pool_op_plugin.h	Bug Fix: Paddle-TRT cannot handle adaptive pooling in pool2d op converter and "num" attribute in split op converter (#20733)	6 years ago
prelu_op_plugin.cu	Update trt5 for paddle-trt (#18645)	6 years ago
prelu_op_plugin.h	5. add static trt load model	6 years ago
split_op_plugin.cu	unify PADDLE_ASSERT_MSG into PADDLE_ENFORCE(error_message) (#19631)	6 years ago
split_op_plugin.h	Update trt5 for paddle-trt (#18645)	6 years ago
swish_op_plugin.cu	Update trt5 for paddle-trt (#18645)	6 years ago
swish_op_plugin.h	Update trt5 for paddle-trt (#18645)	6 years ago
trt_plugin.cc	Implement the Tensorrt plugin for elementwise op (#14487)	7 years ago
trt_plugin.h	fix comments and fix cpplint	6 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