diff --git a/tensorflow/lite/tools/optimize/quantization_utils.cc b/tensorflow/lite/tools/optimize/quantization_utils.cc index a5b9b00b8a9..04e26dbe752 100644 --- a/tensorflow/lite/tools/optimize/quantization_utils.cc +++ b/tensorflow/lite/tools/optimize/quantization_utils.cc @@ -117,7 +117,7 @@ void SymmetricPerChannelQuantization(const float* const input, // Calculate scales per channel std::vector scale_invs(channel_dim_size); const float half_scale = kMaxQuantizedValue; - for (size_t channel_idx = 0; channel_idx < channel_dim_size; channel_idx++) { + for (int channel_idx = 0; channel_idx < channel_dim_size; channel_idx++) { const float half_range = std::max(std::abs(min_vals[channel_idx]), std::abs(max_vals[channel_idx])); output_scales->at(channel_idx) = half_range / half_scale;