Automated Code Change

PiperOrigin-RevId: 632474571
tensorflow · May 11, 2024 · ed7e2c8 · ed7e2c8
1 parent 67846fb
commit ed7e2c8
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 1 deletion.
diff --git a/tensorflow/lite/delegates/gpu/common/data_type_test.cc b/tensorflow/lite/delegates/gpu/common/data_type_test.cc
@@ -15,7 +15,6 @@ limitations under the License.
 
 #include "tensorflow/lite/delegates/gpu/common/data_type.h"
 
-#include <gmock/gmock.h>
 #include <gtest/gtest.h>
 
 namespace tflite {

diff --git a/tensorflow/lite/delegates/gpu/common/flops_util.cc b/tensorflow/lite/delegates/gpu/common/flops_util.cc
@@ -15,6 +15,8 @@ limitations under the License.
 
 #include "tensorflow/lite/delegates/gpu/common/flops_util.h"
 
+#include "tensorflow/lite/delegates/gpu/common/shape.h"
+
 namespace tflite {
 namespace gpu {
 

diff --git a/tensorflow/tools/optimization/BUILD b/tensorflow/tools/optimization/BUILD
@@ -29,6 +29,7 @@ tf_cuda_library(
         "//tensorflow/core:protos_all_cc",
         "//tensorflow/core:tensorflow",
         "@com_google_absl//absl/memory",
+        "@com_google_absl//absl/status",
         "@com_google_absl//absl/strings",
         "@com_google_absl//absl/strings:string_view",
     ],

diff --git a/tensorflow/tools/optimization/optimization_pass_runner.cc b/tensorflow/tools/optimization/optimization_pass_runner.cc
@@ -24,6 +24,7 @@ limitations under the License.
 #include <vector>
 
 #include "absl/memory/memory.h"
+#include "absl/status/status.h"
 #include "absl/strings/str_cat.h"
 #include "absl/strings/string_view.h"
 #include "tensorflow/core/common_runtime/device_set.h"