saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-27 20:02:15 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix prefast:Warning C26814 in non_max_suppression.cc (#12934)

Browse source
This commit is contained in:
Yulong Wang 2022-09-13 15:22:55 -07:00 committed by GitHub
parent bb98922cc8
commit 78bc53f91d
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
onnxruntime/core/providers/cuda/object_detection/non_max_suppression.cc
View file

@ -88,7 +88,7 @@ Status NonMaxSuppression::ComputeInternal(OpKernelContext* ctx) const {
ctx->Output(0, {0, 3});
} else {
// concatenate outputs
const int last_dim = 3;
constexpr int last_dim = 3;
const int num_elements = last_dim * total_num_saved_outputs;
Tensor* output = ctx->Output(0, {static_cast<int64_t>(total_num_saved_outputs), last_dim});
ORT_ENFORCE(output != nullptr);