diff --git a/tensorflow/lite/toco/graph_transformations/ensure_uint8_weights_safe_for_fast_int8_kernels.cc b/tensorflow/lite/toco/graph_transformations/ensure_uint8_weights_safe_for_fast_int8_kernels.cc
index ce4574cdfbf..2a85140488f 100644
--- a/tensorflow/lite/toco/graph_transformations/ensure_uint8_weights_safe_for_fast_int8_kernels.cc
+++ b/tensorflow/lite/toco/graph_transformations/ensure_uint8_weights_safe_for_fast_int8_kernels.cc
@@ -166,7 +166,7 @@ namespace toco {
   int index_of_previous_bad_value = 0;
   bool changed = false;
 
-  for (size_t i = 0; i < buffer_data.size(); i++) {
+  for (int i = 0, iter_limit = buffer_data.size(); i < iter_limit; i++) {
     if (buffer_data[i] == 0) {
       count_bad++;
       if (count_bad > 1) {