Avoid division imposed by modulo

1 files changed, 4 insertions, 2 deletions

diff --git a/eval/src/vespa/eval/tensor/dense/dense_tensor_cells_iterator.cpp b/eval/src/vespa/eval/tensor/dense/dense_tensor_cells_iterator.cpp
index eabd58efd74..eaf80f95c83 100644
--- a/eval/src/vespa/eval/tensor/dense/dense_tensor_cells_iterator.cpp
+++ b/eval/src/vespa/eval/tensor/dense/dense_tensor_cells_iterator.cpp
@@ -10,10 +10,12 @@ DenseTensorCellsIterator::next()
     ++_cellIdx;
     if (valid()) {
         for (int64_t i = (_address.size() - 1); i >= 0; --i) {
-            _address[i] = (_address[i] + 1) % _type.dimensions()[i].size;
-            if (_address[i] != 0) {
+            _address[i]++;
+            if (_address[i] != _type.dimensions()[i].size) {
                 // Outer dimension labels can only be increased when this label wraps around.
                 break;
+            } else {
+                _address[i] = 0;
             }
         }
     }