From 98c434cd50237765837cea71444aca0d85c41883 Mon Sep 17 00:00:00 2001
From: Juhyun Lee <impjdi@google.com>
Date: Wed, 15 Apr 2020 15:27:01 -0700
Subject: [PATCH] Employ tf_platform_alias for TFLite GPU custom_parsers.

PiperOrigin-RevId: 306731712
Change-Id: I58bfc7bd622891ff218ef0e1ac4a78aa06d87725
---
 tensorflow/lite/delegates/gpu/common/BUILD    | 27 +++++++------------
 .../lite/delegates/gpu/common/default/BUILD   | 16 +++++++++++
 .../common/{ => default}/custom_parsers.cc    |  3 ++-
 3 files changed, 28 insertions(+), 18 deletions(-)
 create mode 100644 tensorflow/lite/delegates/gpu/common/default/BUILD
 rename tensorflow/lite/delegates/gpu/common/{ => default}/custom_parsers.cc (95%)

diff --git a/tensorflow/lite/delegates/gpu/common/BUILD b/tensorflow/lite/delegates/gpu/common/BUILD
index d80a2fb0a4a..bb988a443a6 100644
--- a/tensorflow/lite/delegates/gpu/common/BUILD
+++ b/tensorflow/lite/delegates/gpu/common/BUILD
@@ -1,3 +1,5 @@
+load("//tensorflow/core/platform:build_config.bzl", "tf_platform_alias")
+
 package(
     default_visibility = ["//visibility:public"],
     licenses = ["notice"],  # Apache 2.0
@@ -19,17 +21,9 @@ cc_library(
     ],
 )
 
-cc_library(
-    name = "custom_parsers",
-    srcs = ["custom_parsers.cc"],
-    hdrs = ["custom_parsers.h"],
-    deps = [
-        ":shape",
-        ":status",
-        "@com_google_absl//absl/strings",
-        "@com_google_absl//absl/types:any",
-        "@flatbuffers",
-    ],
+exports_files(
+    ["custom_parsers.h"],
+    visibility = ["//tensorflow/lite/delegates/gpu/common:__subpackages__"],
 )
 
 cc_library(
@@ -126,7 +120,6 @@ cc_library(
     srcs = ["model_builder.cc"],
     hdrs = ["model_builder.h"],
     deps = [
-        ":custom_parsers",
         ":data_type",
         ":model",
         ":model_builder_helper",
@@ -134,6 +127,10 @@ cc_library(
         ":shape",
         ":status",
         ":tensor",
+        "@FP16",
+        "@com_google_absl//absl/memory",
+        "@com_google_absl//absl/strings",
+        "@com_google_absl//absl/strings:str_format",
         "//tensorflow/lite:context",
         "//tensorflow/lite:kernel_api",
         "//tensorflow/lite:util",
@@ -143,11 +140,7 @@ cc_library(
         "//tensorflow/lite/kernels/internal:reference_base",
         "//tensorflow/lite/kernels/internal:tensor",
         "//tensorflow/lite/schema:schema_fbs",
-        "@FP16",
-        "@com_google_absl//absl/memory",
-        "@com_google_absl//absl/strings",
-        "@com_google_absl//absl/strings:str_format",
-    ],
+    ] + tf_platform_alias("custom_parsers", "//tensorflow/lite/delegates/gpu/common/"),
 )
 
 cc_test(
diff --git a/tensorflow/lite/delegates/gpu/common/default/BUILD b/tensorflow/lite/delegates/gpu/common/default/BUILD
new file mode 100644
index 00000000000..b085f68fcfb
--- /dev/null
+++ b/tensorflow/lite/delegates/gpu/common/default/BUILD
@@ -0,0 +1,16 @@
+package(
+    default_visibility = ["//visibility:public"],
+    licenses = ["notice"],  # Apache 2.0
+)
+
+cc_library(
+    name = "custom_parsers",
+    srcs = ["custom_parsers.cc"],
+    hdrs = ["//tensorflow/lite/delegates/gpu/common:custom_parsers.h"],
+    deps = [
+        "//tensorflow/lite/delegates/gpu/common:shape",
+        "//tensorflow/lite/delegates/gpu/common:status",
+        "@com_google_absl//absl/strings",
+        "@com_google_absl//absl/types:any",
+    ],
+)
diff --git a/tensorflow/lite/delegates/gpu/common/custom_parsers.cc b/tensorflow/lite/delegates/gpu/common/default/custom_parsers.cc
similarity index 95%
rename from tensorflow/lite/delegates/gpu/common/custom_parsers.cc
rename to tensorflow/lite/delegates/gpu/common/default/custom_parsers.cc
index e43cba05525..9844b8d8aee 100644
--- a/tensorflow/lite/delegates/gpu/common/custom_parsers.cc
+++ b/tensorflow/lite/delegates/gpu/common/default/custom_parsers.cc
@@ -1,4 +1,4 @@
-/* Copyright 2019 The TensorFlow Authors. All Rights Reserved.
+/* Copyright 2020 The TensorFlow Authors. All Rights Reserved.
 
 Licensed under the Apache License, Version 2.0 (the "License");
 you may not use this file except in compliance with the License.
@@ -12,6 +12,7 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 See the License for the specific language governing permissions and
 limitations under the License.
 ==============================================================================*/
+
 #include "tensorflow/lite/delegates/gpu/common/custom_parsers.h"
 
 #include <string>