apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-08 10:00:29 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[Backend] Add AdaptivePool2d for TensorRT plugin (#668)

Browse Source 
* add adaptivepool2d for tensorrt plugin

* update code

* update code

* update code to fix bug
This commit is contained in:
yeliang2258
2022-11-25 17:36:59 +08:00
committed by GitHub
parent cc74fb800d
commit d14828cb18
9 changed files with 461 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
fastdeploy/backends/tensorrt/trt_backend.cc
View File

@@ -124,14 +124,18 @@ bool TrtBackend::InitFromPaddle(const std::string& model_file,
option_ = option;
#ifdef ENABLE_PADDLE_FRONTEND
std::vector<paddle2onnx::CustomOp> ops;
ops.resize(1);
strcpy(ops[0].op_name, "pool2d");
strcpy(ops[0].export_op_name, "AdaptivePool2d");
char* model_content_ptr;
int model_content_size = 0;
char* calibration_cache_ptr;
int calibration_cache_size = 0;
if (!paddle2onnx::Export(model_file.c_str(), params_file.c_str(),
&model_content_ptr, &model_content_size, 11, true,
verbose, true, true, true, nullptr,
0, "tensorrt",
verbose, true, true, true, ops.data(),
1, "tensorrt",
&calibration_cache_ptr, &calibration_cache_size, "", &save_external_)) {
FDERROR << "Error occured while export PaddlePaddle to ONNX format."
<< std::endl;