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

[Other] Optimize sophgo backend (#1273)

Browse Source 
optimize sophgo backend
This commit is contained in:
Jason
2023-02-08 19:50:37 +08:00
committed by GitHub
parent 79d0ee07c9
commit 0fb2f26352
3 changed files with 27 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
fastdeploy/runtime/backends/sophgo/sophgo_backend.h
View File

@@ -30,12 +30,7 @@ class SophgoBackend : public BaseBackend {
public:
SophgoBackend() = default;
virtual ~SophgoBackend();
bool LoadModel(void* model);
bool GetSDKAndDeviceVersion();
bool GetModelInputOutputInfos();
void BuildOption(const SophgoBackendOption& option);
bool InitFromSophgo(const std::string& model_file,
const SophgoBackendOption& option = SophgoBackendOption());
bool Init(const RuntimeOption& option);
int NumInputs() const override {
return static_cast<int>(inputs_desc_.size());
@@ -54,6 +49,10 @@ class SophgoBackend : public BaseBackend {
bool copy_to_fd = true) override;
private:
bool LoadModel(void* model);
bool GetSDKAndDeviceVersion();
bool GetModelInputOutputInfos();
std::vector<TensorInfo> inputs_desc_;
std::vector<TensorInfo> outputs_desc_;
std::string net_name_;