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

Add paddle quantize model support for ORT, TRT and MKLDNN deploy backend (#257)

Browse Source 
* add quantize model support for trt and paddle

* fix bugs

* fix

* update paddle2onnx version

* update version

* add quantize test

Co-authored-by: Jason <jiangjiajun@baidu.com>
This commit is contained in:
yeliang2258
2022-10-09 20:00:05 +08:00
committed by GitHub
parent ff5e798b7f
commit 2a68a23baf
10 changed files with 187 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
fastdeploy/backends/paddle/paddle_backend.h Normal file → Executable file
View File

@@ -20,6 +20,7 @@
#include <vector>
#include "fastdeploy/backends/backend.h"
#include "paddle2onnx/converter.h"
#include "paddle_inference_api.h" // NOLINT
namespace fastdeploy {
@@ -61,7 +62,8 @@ class PaddleBackend : public BaseBackend {
public:
PaddleBackend() {}
virtual ~PaddleBackend() = default;
void BuildOption(const PaddleBackendOption& option);
void BuildOption(const PaddleBackendOption& option,
const std::string& model_file);
bool InitFromPaddle(
const std::string& model_file, const std::string& params_file,