fix the bug of checking the sparse tensor indices order.

PiperOrigin-RevId: 676209041
Intel-tensorflow · Sep 19, 2024 · 6ef126e · 6ef126e
1 parent a0040f0
commit 6ef126e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tensorflow/core/tpu/kernels/sparse_core_preprocess_ops.cc b/tensorflow/core/tpu/kernels/sparse_core_preprocess_ops.cc
@@ -143,6 +143,7 @@ Status ComputeRowIdsBeforePadding(const Tensor& indices_or_row_splits,
             "to be sorted in ascending order.");
       }
       *(row_ids_before_padding + i) = current_row_id;
+      previous_row_id = current_row_id;
     }
   } else if (indices_or_row_splits.dims() == 1 &&
              indices_or_row_splits.NumElements() > 0) {