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

Fix directly visit member of FDTensor (#193)

Browse Source 
* optimize tensorrt usage

* format code

* fix input shape error for onnx model

* Remove some code directly visit FDTensor member (#192)

remove some code directly visit FDTensor member

* fix directly visit member of FDTensor

Co-authored-by: root <root@bjyz-sys-gpu-kongming3.bjyz.baidu.com>
This commit is contained in:
Jason
2022-09-06 11:12:02 +08:00
committed by GitHub
parent 969531dcc8
commit e09ac18a7d
8 changed files with 36 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
csrc/fastdeploy/backends/ort/ort_backend.h
View File

@@ -88,6 +88,6 @@ class OrtBackend : public BaseBackend {
Ort::CustomOpDomain custom_op_domain_ = Ort::CustomOpDomain("Paddle");
#endif
OrtBackendOption option_;
void CopyToCpu(const Ort::Value& value, FDTensor* tensor);
void CopyToCpu(const Ort::Value& value, FDTensor* tensor, const std::string& name);
};
} // namespace fastdeploy