From c072556f70fe6854f80d43af566e12e30b46754b Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Fri, 12 Jul 2019 18:40:04 -0700 Subject: [PATCH] Updates visibility for TFLite Swift and ObjC library targets. PiperOrigin-RevId: 257906035 --- tensorflow/lite/experimental/objc/BUILD.apple | 2 ++ tensorflow/lite/experimental/swift/BUILD.apple | 5 ++--- tensorflow/lite/special_rules.bzl | 4 ++++ 3 files changed, 8 insertions(+), 3 deletions(-) diff --git a/tensorflow/lite/experimental/objc/BUILD.apple b/tensorflow/lite/experimental/objc/BUILD.apple index 0bd6bcd2da5..032cba0cc86 100644 --- a/tensorflow/lite/experimental/objc/BUILD.apple +++ b/tensorflow/lite/experimental/objc/BUILD.apple @@ -1,5 +1,6 @@ # TensorFlow Lite for Objective-C +load("//tensorflow/lite:special_rules.bzl", "ios_visibility_whitelist") load("//tensorflow/lite/experimental/ios:ios.bzl", "TFL_DEFAULT_TAGS", "TFL_DISABLED_SANITIZER_TAGS", "TFL_MINIMUM_OS_VERSION") load("@build_bazel_rules_apple//apple:ios.bzl", "ios_application", "ios_unit_test") @@ -60,6 +61,7 @@ objc_library( hdrs = API_HEADERS, copts = RELEASE_COPTS, tags = TFL_DEFAULT_TAGS, + visibility = ios_visibility_whitelist(), deps = [ "//tensorflow/lite/experimental/c:c_api", ], diff --git a/tensorflow/lite/experimental/swift/BUILD.apple b/tensorflow/lite/experimental/swift/BUILD.apple index b2a0aadca15..0a2126be310 100644 --- a/tensorflow/lite/experimental/swift/BUILD.apple +++ b/tensorflow/lite/experimental/swift/BUILD.apple @@ -1,5 +1,6 @@ # TensorFlow Lite for Swift +load("//tensorflow/lite:special_rules.bzl", "ios_visibility_whitelist") load("//tensorflow/lite/experimental/ios:ios.bzl", "TFL_DEFAULT_TAGS", "TFL_DISABLED_SANITIZER_TAGS", "TFL_MINIMUM_OS_VERSION") load("@build_bazel_rules_apple//apple:ios.bzl", "ios_application", "ios_unit_test") load("@build_bazel_rules_swift//swift:swift.bzl", "swift_library") @@ -14,9 +15,7 @@ swift_library( srcs = glob(["Sources/*.swift"]), module_name = "TensorFlowLite", tags = TFL_DEFAULT_TAGS, - visibility = [ - "//third_party/py/tensorflow_examples/lite:__subpackages__", - ], + visibility = ios_visibility_whitelist(), deps = [ "//tensorflow/lite/experimental/c:c_api", ], diff --git a/tensorflow/lite/special_rules.bzl b/tensorflow/lite/special_rules.bzl index e10af3d240e..f2a4e2f7a57 100644 --- a/tensorflow/lite/special_rules.bzl +++ b/tensorflow/lite/special_rules.bzl @@ -4,3 +4,7 @@ def tflite_portable_test_suite(**kwargs): """This is a no-op outside of Google.""" _ignore = [kwargs] pass + +def ios_visibility_whitelist(): + """This is a no-op outside of Google.""" + pass