diff --git a/tensorflow/lite/tools/evaluation/stages/object_detection_average_precision_stage.cc b/tensorflow/lite/tools/evaluation/stages/object_detection_average_precision_stage.cc
index f3b2ad463f6d3b..6e4bfe595ef2d6 100644
--- a/tensorflow/lite/tools/evaluation/stages/object_detection_average_precision_stage.cc
+++ b/tensorflow/lite/tools/evaluation/stages/object_detection_average_precision_stage.cc
@@ -16,7 +16,7 @@ limitations under the License.
 
 #include <stdint.h>
 
-#include <numeric>
+#include <vector>
 
 #include "tensorflow/core/platform/logging.h"
 #include "tensorflow/lite/c/c_api_types.h"
diff --git a/tensorflow/lite/tools/evaluation/stages/object_detection_average_precision_stage_test.cc b/tensorflow/lite/tools/evaluation/stages/object_detection_average_precision_stage_test.cc
index d952823ff7c39f..31c6027bb72248 100644
--- a/tensorflow/lite/tools/evaluation/stages/object_detection_average_precision_stage_test.cc
+++ b/tensorflow/lite/tools/evaluation/stages/object_detection_average_precision_stage_test.cc
@@ -16,8 +16,6 @@ limitations under the License.
 
 #include <stdint.h>
 
-#include <string>
-
 #include <gtest/gtest.h>
 #include "tensorflow/lite/c/c_api_types.h"
 #include "tensorflow/lite/tools/evaluation/proto/evaluation_config.pb.h"
diff --git a/tensorflow/lite/tools/evaluation/stages/object_detection_stage.cc b/tensorflow/lite/tools/evaluation/stages/object_detection_stage.cc
index f7e4d945dc892d..cd7b04931765bb 100644
--- a/tensorflow/lite/tools/evaluation/stages/object_detection_stage.cc
+++ b/tensorflow/lite/tools/evaluation/stages/object_detection_stage.cc
@@ -15,8 +15,10 @@ limitations under the License.
 #include "tensorflow/lite/tools/evaluation/stages/object_detection_stage.h"
 
 #include <fstream>
+#include <iterator>
 #include <memory>
 #include <string>
+#include <vector>
 
 #include "absl/container/flat_hash_map.h"
 #include "tensorflow/core/platform/logging.h"
diff --git a/tensorflow/lite/tools/evaluation/stages/tflite_inference_stage.cc b/tensorflow/lite/tools/evaluation/stages/tflite_inference_stage.cc
index 9d6f7f9ba118ba..fdf80a5f2c03cf 100644
--- a/tensorflow/lite/tools/evaluation/stages/tflite_inference_stage.cc
+++ b/tensorflow/lite/tools/evaluation/stages/tflite_inference_stage.cc
@@ -14,11 +14,12 @@ limitations under the License.
 ==============================================================================*/
 #include "tensorflow/lite/tools/evaluation/stages/tflite_inference_stage.h"
 
-#include <cstring>
+#include <cstdint>
 #include <fstream>
 #include <memory>
 #include <string>
 #include <utility>
+#include <vector>
 
 #include "absl/base/attributes.h"
 #include "tensorflow/core/platform/logging.h"
diff --git a/tensorflow/lite/tools/evaluation/stages/tflite_inference_stage.h b/tensorflow/lite/tools/evaluation/stages/tflite_inference_stage.h
index aae93082a1abd7..ca8ffc688fea7f 100644
--- a/tensorflow/lite/tools/evaluation/stages/tflite_inference_stage.h
+++ b/tensorflow/lite/tools/evaluation/stages/tflite_inference_stage.h
@@ -17,6 +17,7 @@ limitations under the License.
 
 #include <stdint.h>
 
+#include <memory>
 #include <vector>
 
 #include "xla/tsl/util/stats_calculator.h"
diff --git a/tensorflow/lite/tools/evaluation/stages/tflite_inference_stage_test.cc b/tensorflow/lite/tools/evaluation/stages/tflite_inference_stage_test.cc
index b1a3b67e295eda..b23ce24e3d64ad 100644
--- a/tensorflow/lite/tools/evaluation/stages/tflite_inference_stage_test.cc
+++ b/tensorflow/lite/tools/evaluation/stages/tflite_inference_stage_test.cc
@@ -16,7 +16,8 @@ limitations under the License.
 
 #include <stdint.h>
 
-#include <string>
+#include <utility>
+#include <vector>
 
 #include <gtest/gtest.h>
 #include "tensorflow/lite/core/c/common.h"
diff --git a/tensorflow/lite/tools/evaluation/stages/topk_accuracy_eval_stage.cc b/tensorflow/lite/tools/evaluation/stages/topk_accuracy_eval_stage.cc
index 1d8bb0595cc662..e25d2aa95b9ceb 100644
--- a/tensorflow/lite/tools/evaluation/stages/topk_accuracy_eval_stage.cc
+++ b/tensorflow/lite/tools/evaluation/stages/topk_accuracy_eval_stage.cc
@@ -17,7 +17,9 @@ limitations under the License.
 #include <stdint.h>
 
 #include <algorithm>
+#include <cstddef>
 #include <numeric>
+#include <vector>
 
 #include "tensorflow/core/platform/logging.h"
 #include "tensorflow/lite/c/c_api_types.h"
diff --git a/tensorflow/lite/tools/evaluation/stages/topk_accuracy_eval_stage_test.cc b/tensorflow/lite/tools/evaluation/stages/topk_accuracy_eval_stage_test.cc
index 01e3e607ab50c0..acccf8948f8731 100644
--- a/tensorflow/lite/tools/evaluation/stages/topk_accuracy_eval_stage_test.cc
+++ b/tensorflow/lite/tools/evaluation/stages/topk_accuracy_eval_stage_test.cc
@@ -17,6 +17,7 @@ limitations under the License.
 #include <stdint.h>
 
 #include <string>
+#include <vector>
 
 #include <gtest/gtest.h>
 #include "tensorflow/lite/c/c_api_types.h"