Automated Code Change

PiperOrigin-RevId: 691291363
ROCm · Oct 30, 2024 · 225fb6d · 225fb6d
1 parent f86990d
commit 225fb6d
Show file tree

Hide file tree

Showing 8 changed files with 11 additions and 5 deletions.
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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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"