diff options
Diffstat (limited to 'tests/validation/reference/YOLOLayer.cpp')
-rw-r--r-- | tests/validation/reference/YOLOLayer.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tests/validation/reference/YOLOLayer.cpp b/tests/validation/reference/YOLOLayer.cpp index cf5e256cf9..0011b85c1e 100644 --- a/tests/validation/reference/YOLOLayer.cpp +++ b/tests/validation/reference/YOLOLayer.cpp @@ -46,7 +46,8 @@ SimpleTensor<T> yolo_layer(const SimpleTensor<T> &src, const ActivationLayerInfo const T a(info.a()); const T b(info.b()); - for(int i = 0; i < src.num_elements(); ++i) + const uint32_t num_elements = src.num_elements(); + for(uint32_t i = 0; i < num_elements; ++i) { const size_t z = index2coord(dst.shape(), i).z() % (num_classes + 5); |