Clean up include and build file

PiperOrigin-RevId: 635488859
tensorflow · May 24, 2024 · a9f6952 · a9f6952
1 parent 3068334
commit a9f6952
Show file tree

Hide file tree

Showing 52 changed files with 4,366 additions and 1,061 deletions.
diff --git a/tensorflow/compiler/mlir/lite/schema/BUILD b/tensorflow/compiler/mlir/lite/schema/BUILD
@@ -1,4 +1,5 @@
 load("@flatbuffers//:build_defs.bzl", "flatbuffer_cc_library")
+load("//tensorflow:tensorflow.bzl", "tf_cc_test")
 load("//tensorflow:tensorflow.default.bzl", "get_compatible_with_portable")
 
 package(
@@ -69,3 +70,25 @@ cc_library(
         "@flatbuffers",
     ],
 )
+
+# Schema test to make sure we don't introduce backward incompatible changes
+# to schemas.
+tf_cc_test(
+    name = "flatbuffer_compatibility_test",
+    size = "small",
+    srcs = ["flatbuffer_compatibility_test.cc"],
+    data = [
+        "schema.fbs",
+        "schema_v3b.fbs",
+    ],
+    tags = [
+        "no_oss",
+        "tflite_not_portable_android",
+        "tflite_not_portable_ios",
+    ],
+    deps = [
+        "//tensorflow/core/platform",
+        "@com_google_googletest//:gtest_main",
+        "@flatbuffers//:flatc_library",
+    ],
+)
diff --git a/...e/schema/flatbuffer_compatibility_test.cc → ...e/schema/flatbuffer_compatibility_test.cc b/...e/schema/flatbuffer_compatibility_test.cc → ...e/schema/flatbuffer_compatibility_test.cc
@@ -63,9 +63,10 @@ TEST(SchemaTest, TestCompatibility) {
   // Read file contents of schemas into strings
   // TODO(aselle): Need a reliable way to load files.
   std::string base_contents, current_contents;
-  const char *base_filename = TFLITE_TF_PREFIX "lite/schema/schema_v3b.fbs";
+  const char *base_filename = TFLITE_TF_PREFIX
+      "compiler/mlir/lite/schema/schema_v3b.fbs";
   const char *current_filename =
-      TFLITE_TF_PREFIX "lite/schema/schema.fbs";
+      TFLITE_TF_PREFIX "compiler/mlir/lite/schema/schema.fbs";
 
   ASSERT_TRUE(LoadFileRaw(base_filename, &base_contents));
   ASSERT_TRUE(LoadFileRaw(current_filename, &current_contents));