fix rknn output index error in SDK (#1682)

pull/1690/head
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

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();
}