mirrors/mmdeploy
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmdeploy.git synced 2025-01-14 08:09:43 +08:00
Code Issues Projects Releases Wiki Activity

fix rknn output index error in SDK (#1682)

Browse Source
This commit is contained in:
AllentDan 2023-01-30 20:51:06 +08:00 committed by GitHub
parent cbddf5a458
commit 878a364470
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
csrc/mmdeploy/net/rknn/rknn_net.cpp
View File

@ -250,7 +250,7 @@ Result<void> RKNNNet::Forward() {
for (uint32_t i = 0; i < output_tensors_.size(); ++i) {
outputs[i].want_float = 1;
outputs[i].is_prealloc = 1; // use pre-allocated buffer in `output_tensors_`
outputs[i].index = 1;
outputs[i].index = i;
outputs[i].buf = output_tensors_[i].data();
outputs[i].size = output_tensors_[i].byte_size();
}