Update package visibility for all keras sub packages.

PiperOrigin-RevId: 301829736
Change-Id: If1cd07bd48ff4c2f2ec850cc71460f70718a5dc0

tensorflow/python/keras/api/BUILD

@@ -6,7 +6,9 @@ load("//tensorflow/python/tools/api/generator:api_init_files.bzl", "KERAS_API_IN
 load("//tensorflow/python/tools/api/generator:api_init_files_v1.bzl", "KERAS_API_INIT_FILES_V1")
 
 package(
-    default_visibility = ["//visibility:public"],
+    default_visibility = [
+        "//tensorflow:tensorflow_py",
+    ],
     licenses = ["notice"],  # Apache 2.0 License
 )
 

tensorflow/python/keras/applications/BUILD

@@ -4,7 +4,11 @@
 load("//tensorflow:tensorflow.bzl", "tf_py_test")
 
 package(
-    default_visibility = ["//visibility:public"],
+    default_visibility = [
+        # Remove this deps to integration test.
+        "//tensorflow/lite/experimental/tf_runtime:__pkg__",
+        "//tensorflow/python/keras:__subpackages__",
+    ],
     licenses = ["notice"],  # Apache 2.0
 )
 

tensorflow/python/keras/datasets/BUILD

@@ -2,7 +2,9 @@
 #   Contains the Keras datasets package (internal TensorFlow version).
 
 package(
-    default_visibility = ["//visibility:public"],
+    default_visibility = [
+        "//tensorflow/python/keras:__subpackages__",
+    ],
     licenses = ["notice"],  # Apache 2.0
 )
 

tensorflow/python/keras/distribute/BUILD

@@ -7,7 +7,12 @@ load("//tensorflow:tensorflow.bzl", "py_test")
 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
 
 package(
-    default_visibility = ["//visibility:public"],
+    # TODO(scottzhu): Remove this deps when distribute test are converted to integration test.
+    default_visibility = [
+        "//tensorflow/python/distribute:__pkg__",
+        "//tensorflow/python/keras:__subpackages__",
+        "//tensorflow/tools/pip_package:__pkg__",
+    ],
     licenses = ["notice"],  # Apache 2.0
 )
 

tensorflow/python/keras/engine/BUILD

@@ -5,7 +5,12 @@ load("//tensorflow:tensorflow.bzl", "tf_py_test")
 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
 
 package(
-    default_visibility = ["//visibility:public"],
+    # TODO(scottzhu): Remove non-keras deps from TF.
+    default_visibility = [
+        "//tensorflow/python:__pkg__",
+        "//tensorflow/python/feature_column:__pkg__",
+        "//tensorflow/python/keras:__subpackages__",
+    ],
     licenses = ["notice"],  # Apache 2.0
 )
 

tensorflow/python/keras/layers/BUILD

@@ -5,7 +5,15 @@ load("//tensorflow:tensorflow.bzl", "tf_py_test")
 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
 
 package(
-    default_visibility = ["//visibility:public"],
+    # TODO(scottzhu): Remove non-keras deps from TF.
+    default_visibility = [
+        "//tensorflow/python/distribute:__pkg__",
+        "//tensorflow/python/feature_column:__pkg__",
+        "//tensorflow/python/keras:__subpackages__",
+        "//tensorflow/python/kernel_tests:__pkg__",
+        "//tensorflow/python/training/tracking:__pkg__",
+        "//tensorflow/tools/pip_package:__pkg__",
+    ],
     licenses = ["notice"],  # Apache 2.0
 )
 

tensorflow/python/keras/layers/preprocessing/BUILD

@@ -5,7 +5,10 @@ load("//tensorflow:tensorflow.bzl", "tf_py_test")
 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
 
 package(
-    default_visibility = ["//visibility:public"],
+    default_visibility = [
+        "//tensorflow/python/keras:__subpackages__",
+        "//tensorflow/tools/pip_package:__pkg__",
+    ],
     licenses = ["notice"],  # Apache 2.0
 )
 

tensorflow/python/keras/mixed_precision/experimental/BUILD

@@ -20,7 +20,13 @@ load("//tensorflow:tensorflow.bzl", "py_test")
 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
 
 package(
-    default_visibility = ["//visibility:public"],
+    default_visibility = [
+        # TODO(scottzhu): Remove these two deps and convert the test to integration test.
+        "//tensorflow/python:__pkg__",  # For loss_scale_optimizer_test
+        "//tensorflow/python/distribute:__pkg__",  # For collective_all_reduce_strategy_test
+        "//tensorflow/python/keras:__subpackages__",
+        "//tensorflow/tools/pip_package:__pkg__",
+    ],
     licenses = ["notice"],  # Apache 2.0
 )
 

tensorflow/python/keras/optimizer_v2/BUILD

@@ -4,7 +4,13 @@
 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
 
 package(
-    default_visibility = ["//visibility:public"],
+    # TODO(scottzhu): Remove non-keras deps from TF.
+    default_visibility = [
+        "//tensorflow/python:__pkg__",
+        "//tensorflow/python/distribute:__pkg__",
+        "//tensorflow/python/keras:__subpackages__",
+        "//tensorflow/python/training/tracking:__pkg__",
+    ],
     licenses = ["notice"],  # Apache 2.0
 )
 

tensorflow/python/keras/premade/BUILD

@@ -1,15 +1,16 @@
 # Description:
 #   Contains the Keras Premade Models (internal TensorFlow version).
+load("//tensorflow:tensorflow.bzl", "py_test")
 
 package(
-    default_visibility = ["//visibility:public"],
+    default_visibility = [
+        "//tensorflow/python/keras:__subpackages__",
+    ],
     licenses = ["notice"],  # Apache 2.0
 )
 
 exports_files(["LICENSE"])
 
-load("//tensorflow:tensorflow.bzl", "py_test")
-
 py_library(
     name = "premade",
     srcs = [

tensorflow/python/keras/preprocessing/BUILD

@@ -4,7 +4,12 @@
 load("//tensorflow:tensorflow.bzl", "tf_py_test")
 
 package(
-    default_visibility = ["//visibility:public"],
+    default_visibility = [
+        # TODO(scottzhu): Remove non-keras deps from TF.
+        "//tensorflow/lite/experimental/tf_runtime:__pkg__",
+        "//tensorflow/tools/docs:__pkg__",
+        "//tensorflow/python/keras:__subpackages__",
+    ],
     licenses = ["notice"],  # Apache 2.0
 )
 

tensorflow/python/keras/saving/BUILD

@@ -4,7 +4,11 @@
 load("//tensorflow:tensorflow.bzl", "tf_py_test")
 
 package(
-    default_visibility = ["//visibility:public"],
+    # TODO(scottzhu): Remove non-keras deps from TF.
+    default_visibility = [
+        "//tensorflow/python/distribute:__pkg__",
+        "//tensorflow/python/keras:__subpackages__",
+    ],
     licenses = ["notice"],  # Apache 2.0
 )
 

tensorflow/python/keras/tests/BUILD

@@ -4,7 +4,9 @@
 load("//tensorflow:tensorflow.bzl", "tf_py_test")
 
 package(
-    default_visibility = ["//visibility:public"],
+    default_visibility = [
+        "//tensorflow/tools/pip_package:__pkg__",
+    ],
     licenses = ["notice"],  # Apache 2.0
 )
 

tensorflow/python/keras/type/BUILD

@@ -1,7 +1,7 @@
 load("//tensorflow:tensorflow.bzl", "py_strict_library")
 
 package(
-    default_visibility = ["//tensorflow:__subpackages__"],
+    default_visibility = ["//tensorflow/python/keras:__subpackages__"],
     licenses = ["notice"],  # Apache 2.0
 )
 

tensorflow/python/keras/utils/BUILD

@@ -5,7 +5,11 @@ load("//tensorflow:tensorflow.bzl", "tf_py_test")
 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
 
 package(
-    default_visibility = ["//visibility:public"],
+    # TODO(scottzhu): Remove non-keras deps from TF.
+    default_visibility = [
+        "//tensorflow/python/feature_column:__pkg__",
+        "//tensorflow/python/keras:__subpackages__",
+    ],
     licenses = ["notice"],  # Apache 2.0
 )
 

tensorflow/python/keras/wrappers/BUILD

@@ -4,7 +4,7 @@
 load("//tensorflow:tensorflow.bzl", "tf_py_test")
 
 package(
-    default_visibility = ["//visibility:public"],
+    default_visibility = ["//tensorflow/python/keras:__subpackages__"],
     licenses = ["notice"],  # Apache 2.0
 )