Automated Code Change

PiperOrigin-RevId: 702230075
google-ai-edge · Dec 3, 2024 · 143d03f · 143d03f
1 parent b878ebc
commit 143d03f
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 4 deletions.
diff --git a/tflite/tools/evaluation/stages/image_classification_stage.cc b/tflite/tools/evaluation/stages/image_classification_stage.cc
@@ -15,9 +15,11 @@ limitations under the License.
 #include "tflite/tools/evaluation/stages/image_classification_stage.h"
 
 #include <algorithm>
+#include <cstddef>
 #include <iterator>
 #include <memory>
 #include <string>
+#include <vector>
 
 #include "tensorflow/core/platform/logging.h"
 #include "tflite/c/c_api_types.h"

diff --git a/tflite/tools/evaluation/stages/image_preprocessing_stage.cc b/tflite/tools/evaluation/stages/image_preprocessing_stage.cc
@@ -21,7 +21,6 @@ limitations under the License.
 #include <ios>
 #include <iterator>
 #include <memory>
-#include <streambuf>
 #include <string>
 #include <vector>
 

diff --git a/tflite/tools/evaluation/stages/image_preprocessing_stage_test.cc b/tflite/tools/evaluation/stages/image_preprocessing_stage_test.cc
@@ -14,7 +14,7 @@ limitations under the License.
 ==============================================================================*/
 #include "tflite/tools/evaluation/stages/image_preprocessing_stage.h"
 
-#include <memory>
+#include <cstdint>
 #include <string>
 
 #include <gtest/gtest.h>

diff --git a/tflite/tools/evaluation/stages/inference_profiler_stage.cc b/tflite/tools/evaluation/stages/inference_profiler_stage.cc
@@ -15,10 +15,12 @@ limitations under the License.
 #include "tflite/tools/evaluation/stages/inference_profiler_stage.h"
 
 #include <cmath>
+#include <cstddef>
 #include <cstdint>
 #include <limits>
 #include <memory>
 #include <random>
+#include <vector>
 
 #include "fp16.h"  // from @FP16
 #include "tensorflow/core/platform/logging.h"

diff --git a/tflite/tools/evaluation/stages/inference_profiler_stage_test.cc b/tflite/tools/evaluation/stages/inference_profiler_stage_test.cc
@@ -16,8 +16,6 @@ limitations under the License.
 
 #include <stdint.h>
 
-#include <string>
-
 #include <gtest/gtest.h>
 #include "tflite/core/c/common.h"
 #include "tflite/tools/evaluation/proto/evaluation_config.pb.h"