From 019cd527de85762993f183081e85cfb845dae8e4 Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Thu, 13 Aug 2020 11:56:05 -0700
Subject: [PATCH] Export GPU allowlist database and flatbuffer java library
 BUILD targets.

PiperOrigin-RevId: 326493266
Change-Id: Ib87006ee2ee3e3723f422ef7bf20a382a4de34ef
---
 .../lite/experimental/acceleration/compatibility/BUILD | 10 +++++++++-
 .../acceleration/compatibility/database.fbs            | 10 +++++-----
 2 files changed, 14 insertions(+), 6 deletions(-)

diff --git a/tensorflow/lite/experimental/acceleration/compatibility/BUILD b/tensorflow/lite/experimental/acceleration/compatibility/BUILD
index 559abc44a4d..6c5a32b0795 100644
--- a/tensorflow/lite/experimental/acceleration/compatibility/BUILD
+++ b/tensorflow/lite/experimental/acceleration/compatibility/BUILD
@@ -13,7 +13,7 @@
 # limitations under the License.
 # ==============================================================================
 
-load("@flatbuffers//:build_defs.bzl", "flatbuffer_cc_library")
+load("@flatbuffers//:build_defs.bzl", "flatbuffer_cc_library", "flatbuffer_java_library")
 load("//tensorflow/lite:special_rules.bzl", "tflite_extra_gles_deps", "tflite_portable_test_suite")
 
 package(
@@ -30,6 +30,12 @@ flatbuffer_cc_library(
 
 exports_files(srcs = ["database.fbs"])
 
+flatbuffer_java_library(
+    name = "database_fbs_java",
+    srcs = ["database.fbs"],
+    package_prefix = "org.tensorflow",
+)
+
 cc_library(
     name = "devicedb",
     srcs = [
@@ -118,6 +124,8 @@ cc_test(
     ],
 )
 
+exports_files(["gpu_compatibility.bin"])
+
 genrule(
     name = "gpu_compatibility_binary",
     srcs = ["gpu_compatibility.bin"],
diff --git a/tensorflow/lite/experimental/acceleration/compatibility/database.fbs b/tensorflow/lite/experimental/acceleration/compatibility/database.fbs
index cf5aaa6d795..8c77718068a 100644
--- a/tensorflow/lite/experimental/acceleration/compatibility/database.fbs
+++ b/tensorflow/lite/experimental/acceleration/compatibility/database.fbs
@@ -28,15 +28,15 @@ enum Comparison : byte {
 // The structure describes a decision tree, with multiple matching branches.
 // The branches are applied depth-first.
 table DeviceDatabase {
-  root:[tflite.acceleration.DeviceDecisionTreeNode];
+  root:[DeviceDecisionTreeNode];
 }
 
 table DeviceDecisionTreeNode {
   // The variables are strings, as we have multiple clients that want to
   // introduce their own fields. Known variables are listed in variables.h.
   variable:string (shared);
-  comparison:tflite.acceleration.Comparison;
-  items:[tflite.acceleration.DeviceDecisionTreeEdge];
+  comparison:Comparison;
+  items:[DeviceDecisionTreeEdge];
 }
 
 table DeviceDecisionTreeEdge {
@@ -44,9 +44,9 @@ table DeviceDecisionTreeEdge {
   value:string (key, shared);
   // Which child branches should also be consulted and used to override this
   // node.
-  children:[tflite.acceleration.DeviceDecisionTreeNode];
+  children:[DeviceDecisionTreeNode];
   // What information can be derived about this device.
-  derived_properties:[tflite.acceleration.DerivedProperty];
+  derived_properties:[DerivedProperty];
 }
 
 // Derived variable value to combine with detected variables.