Automated Code Change

PiperOrigin-RevId: 651292266

tensorflow/lite/tools/signature/BUILD

@@ -31,8 +31,10 @@ cc_library(
         "//tensorflow/lite/core/c:common",
         "//tensorflow/lite/schema:schema_fbs",
         "@com_google_absl//absl/memory",
+        "@com_google_absl//absl/status",
         "@com_google_protobuf//:protobuf",
         "@flatbuffers",
+        "@local_tsl//tsl/platform:status",
     ],
 )
 
@@ -50,16 +52,18 @@ cc_test(
     deps = [
         ":signature_def_util",
         "//tensorflow/cc/saved_model:signature_constants",
-        "//tensorflow/core:tflite_portable_logging",
         "//tensorflow/core/platform:errors",
+        "//tensorflow/core/protobuf:for_core_protos_cc",
         "//tensorflow/lite:framework",
+        "//tensorflow/lite:string",
         "//tensorflow/lite/c:c_api",
         "//tensorflow/lite/core:framework",
-        "//tensorflow/lite/core/c:c_api",
         "//tensorflow/lite/core/c:common",
         "//tensorflow/lite/schema:schema_fbs",
-        "//tensorflow/lite/testing:util",
+        "@com_google_absl//absl/status",
         "@com_google_googletest//:gtest_main",
+        "@flatbuffers//:runtime_cc",
+        "@local_tsl//tsl/platform:status",
     ],
 )
 
@@ -74,9 +78,11 @@ pybind_extension(
     enable_stub_generation = True,
     deps = [
         ":signature_def_util",
+        "//tensorflow/core/protobuf:for_core_protos_cc",
         "//tensorflow/lite:framework",
         "//tensorflow/lite/core:framework",
         "//tensorflow/python/lib/core:pybind11_lib",
+        "@com_google_absl//absl/status",
         "@pybind11",
     ],
 )

tensorflow/lite/tools/signature/signature_def_util.cc

@@ -19,14 +19,15 @@ limitations under the License.
 #include <string>
 #include <utility>
 
-#include "absl/memory/memory.h"
-#include "flatbuffers/flatbuffers.h"  // from @flatbuffers
+#include "absl/status/status.h"
+#include "flatbuffers/flatbuffer_builder.h"  // from @flatbuffers
 #include "flatbuffers/flexbuffers.h"  // from @flatbuffers
+#include "flatbuffers/vector.h"  // from @flatbuffers
 #include "tensorflow/core/lib/core/errors.h"
 #include "tensorflow/core/platform/errors.h"
 #include "tensorflow/core/protobuf/meta_graph.pb.h"
-#include "tensorflow/lite/core/model_builder.h"
 #include "tensorflow/lite/schema/schema_generated.h"
+#include "tsl/platform/status.h"
 
 namespace tflite {
 namespace {

tensorflow/lite/tools/signature/signature_def_util_test.cc

@@ -17,14 +17,15 @@ limitations under the License.
 #include <string>
 
 #include <gtest/gtest.h>
+#include "absl/status/status.h"
+#include "flatbuffers/buffer.h"  // from @flatbuffers
 #include "tensorflow/cc/saved_model/signature_constants.h"
 #include "tensorflow/core/platform/errors.h"
-#include "tensorflow/core/platform/logging.h"
-#include "tensorflow/lite/core/c/c_api.h"
-#include "tensorflow/lite/core/c/common.h"
+#include "tensorflow/core/protobuf/meta_graph.pb.h"
 #include "tensorflow/lite/core/model_builder.h"
 #include "tensorflow/lite/schema/schema_generated.h"
-#include "tensorflow/lite/testing/util.h"
+#include "tensorflow/lite/string_type.h"
+#include "tsl/platform/status.h"
 
 namespace tflite {
 namespace {

tensorflow/lite/tools/signature/signature_def_util_wrapper_pybind11.cc

@@ -14,9 +14,11 @@ limitations under the License.
 ==============================================================================*/
 #include <string>
 
+#include "absl/status/status.h"
 #include "pybind11/pybind11.h"  // from @pybind11
 #include "pybind11/pytypes.h"  // from @pybind11
 #include "pybind11/stl.h"  // from @pybind11
+#include "tensorflow/core/protobuf/meta_graph.pb.h"
 #include "tensorflow/lite/core/model_builder.h"
 #include "tensorflow/lite/tools/signature/signature_def_util.h"
 #include "tensorflow/python/lib/core/pybind11_lib.h"