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

add 'GetOutputInfos' and 'GetInputInfos' interface (#232)

Browse Source 
add GetOutputInfos GetInputInfos
This commit is contained in:
heliqi
2022-09-15 00:09:31 -05:00
committed by GitHub
parent 6963296a89
commit 0805ead0ed
13 changed files with 125 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -589,6 +589,14 @@ TensorInfo TrtBackend::GetInputInfo(int index) {
return info;
}
std::vector<TensorInfo> TrtBackend::GetInputInfos() {
std::vector<TensorInfo> infos;
for (auto i = 0; i < inputs_desc_.size(); i++) {
infos.emplace_back(GetInputInfo(i));
}
return infos;
}
TensorInfo TrtBackend::GetOutputInfo(int index) {
FDASSERT(index < NumOutputs(),
"The index: %d should less than the number of outputs: %d.", index,
@@ -600,4 +608,13 @@ TensorInfo TrtBackend::GetOutputInfo(int index) {
info.dtype = GetFDDataType(outputs_desc_[index].dtype);
return info;
}
std::vector<TensorInfo> TrtBackend::GetOutputInfos() {
std::vector<TensorInfo> infos;
for (auto i = 0; i < outputs_desc_.size(); i++) {
infos.emplace_back(GetOutputInfo(i));
}
return infos;
}
} // namespace fastdeploy

2
fastdeploy/backends/tensorrt/trt_backend.h
View File

@@ -72,6 +72,8 @@ class TrtBackend : public BaseBackend {
int NumOutputs() const { return outputs_desc_.size(); }
TensorInfo GetInputInfo(int index);
TensorInfo GetOutputInfo(int index);
std::vector<TensorInfo> GetInputInfos() override;
std::vector<TensorInfo> GetOutputInfos() override;
~TrtBackend() {
if (parser_) {