Adding tf_mobile_aware_deps macro.

PiperOrigin-RevId: 270791843
xialinxi · Sep 24, 2019 · 014df5e · 014df5e
1 parent 40fb2e3
commit 014df5e
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/tensorflow/core/platform/default/build_refactor.bzl b/tensorflow/core/platform/default/build_refactor.bzl
@@ -17,7 +17,7 @@ load(
     "tf_copts",
 )
 
-TF_PLATFORM_LIBRARIES = {
+TF_DEFAULT_PLATFORM_LIBRARIES = {
     "context": {
         "name": "context_impl",
         "hdrs": ["//tensorflow/core/platform:context.h"],
@@ -482,11 +482,14 @@ def tf_instantiate_platform_libraries(names = []):
                 tags = ["no_oss", "manual"],
             )
         else:
-            if name in TF_PLATFORM_LIBRARIES:
-                native.cc_library(**TF_PLATFORM_LIBRARIES[name])
+            if name in TF_DEFAULT_PLATFORM_LIBRARIES:
+                native.cc_library(**TF_DEFAULT_PLATFORM_LIBRARIES[name])
             if name in TF_WINDOWS_PLATFORM_LIBRARIES:
                 native.cc_library(**TF_WINDOWS_PLATFORM_LIBRARIES[name])
 
+def tf_mobile_aware_deps(name):
+    return [":" + name]
+
 def tf_platform_helper_deps(name):
     return select({
         "//tensorflow:windows": [":windows_" + name],