1# Description:
2# Package for TensorFlow.
3
4load("//tensorflow/python/tools/api/generator:api_gen.bzl", "gen_api_init_files")
5load("//tensorflow/python/tools/api/generator:api_init_files.bzl", "KERAS_API_INIT_FILES")
6load("//tensorflow/python/tools/api/generator:api_init_files_v1.bzl", "KERAS_API_INIT_FILES_V1")
7
8package(
9    default_visibility = [
10        "//tensorflow:tensorflow_py",
11    ],
12    licenses = ["notice"],  # Apache 2.0 License
13)
14
15keras_packages = [
16    "tensorflow.python",
17    "tensorflow.python.keras",
18    "tensorflow.python.keras.activations",
19    "tensorflow.python.keras.applications.densenet",
20    "tensorflow.python.keras.applications.efficientnet",
21    "tensorflow.python.keras.applications.imagenet_utils",
22    "tensorflow.python.keras.applications.inception_resnet_v2",
23    "tensorflow.python.keras.applications.inception_v3",
24    "tensorflow.python.keras.applications.mobilenet",
25    "tensorflow.python.keras.applications.mobilenet_v2",
26    "tensorflow.python.keras.applications.mobilenet_v3",
27    "tensorflow.python.keras.applications.nasnet",
28    "tensorflow.python.keras.applications.resnet",
29    "tensorflow.python.keras.applications.resnet_v2",
30    "tensorflow.python.keras.applications.vgg16",
31    "tensorflow.python.keras.applications.vgg19",
32    "tensorflow.python.keras.applications.xception",
33    "tensorflow.python.keras.backend",
34    "tensorflow.python.keras.backend_config",
35    "tensorflow.python.keras.callbacks",
36    "tensorflow.python.keras.callbacks_v1",
37    "tensorflow.python.keras.constraints",
38    "tensorflow.python.keras.datasets.boston_housing",
39    "tensorflow.python.keras.datasets.cifar10",
40    "tensorflow.python.keras.datasets.cifar100",
41    "tensorflow.python.keras.datasets.fashion_mnist",
42    "tensorflow.python.keras.datasets.imdb",
43    "tensorflow.python.keras.datasets.mnist",
44    "tensorflow.python.keras.datasets.reuters",
45    "tensorflow.python.keras.engine.base_layer",
46    "tensorflow.python.keras.engine.data_adapter",
47    "tensorflow.python.keras.engine.input_layer",
48    "tensorflow.python.keras.engine.input_spec",
49    "tensorflow.python.keras.engine.sequential",
50    "tensorflow.python.keras.engine.training",
51    "tensorflow.python.keras.estimator",
52    "tensorflow.python.keras.feature_column.sequence_feature_column",
53    "tensorflow.python.keras.initializers",
54    "tensorflow.python.keras.initializers.initializers_v1",
55    "tensorflow.python.keras.initializers.initializers_v2",
56    "tensorflow.python.keras.layers.advanced_activations",
57    "tensorflow.python.keras.layers.convolutional",
58    "tensorflow.python.keras.layers.convolutional_recurrent",
59    "tensorflow.python.keras.layers.core",
60    "tensorflow.python.keras.layers.cudnn_recurrent",
61    "tensorflow.python.keras.layers.dense_attention",
62    "tensorflow.python.keras.layers.embeddings",
63    "tensorflow.python.keras.layers.local",
64    "tensorflow.python.keras.layers.merge",
65    "tensorflow.python.keras.layers.noise",
66    "tensorflow.python.keras.layers.normalization",
67    "tensorflow.python.keras.layers.normalization_v2",
68    "tensorflow.python.keras.layers.preprocessing",
69    "tensorflow.python.keras.layers.pooling",
70    "tensorflow.python.keras.layers.recurrent",
71    "tensorflow.python.keras.layers.recurrent_v2",
72    "tensorflow.python.keras.layers.serialization",
73    "tensorflow.python.keras.layers.wrappers",
74    "tensorflow.python.keras.losses",
75    "tensorflow.python.keras.metrics",
76    "tensorflow.python.keras.mixed_precision.get_layer_policy",
77    "tensorflow.python.keras.mixed_precision.loss_scale_optimizer",
78    "tensorflow.python.keras.mixed_precision.policy",
79    "tensorflow.python.keras.models",
80    "tensorflow.python.keras.optimizer_v2.adadelta",
81    "tensorflow.python.keras.optimizer_v2.adagrad",
82    "tensorflow.python.keras.optimizer_v2.adam",
83    "tensorflow.python.keras.optimizer_v2.adamax",
84    "tensorflow.python.keras.optimizer_v2.ftrl",
85    "tensorflow.python.keras.optimizer_v2.gradient_descent",
86    "tensorflow.python.keras.optimizer_v2.learning_rate_schedule",
87    "tensorflow.python.keras.optimizer_v2.nadam",
88    "tensorflow.python.keras.optimizer_v2.optimizer_v2",
89    "tensorflow.python.keras.optimizer_v2.rmsprop",
90    "tensorflow.python.keras.optimizers",
91    "tensorflow.python.keras.premade.linear",
92    "tensorflow.python.keras.premade.wide_deep",
93    "tensorflow.python.keras.preprocessing.image",
94    "tensorflow.python.keras.preprocessing.sequence",
95    "tensorflow.python.keras.preprocessing.text",
96    "tensorflow.python.keras.regularizers",
97    "tensorflow.python.keras.saving.model_config",
98    "tensorflow.python.keras.saving.save",
99    "tensorflow.python.keras.saving.saved_model_experimental",
100    "tensorflow.python.keras.utils.data_utils",
101    "tensorflow.python.keras.utils.generic_utils",
102    "tensorflow.python.keras.utils.io_utils",
103    "tensorflow.python.keras.utils.layer_utils",
104    "tensorflow.python.keras.utils.losses_utils",
105    "tensorflow.python.keras.utils.multi_gpu_utils",
106    "tensorflow.python.keras.utils.np_utils",
107    "tensorflow.python.keras.utils.vis_utils",
108    "tensorflow.python.keras.wrappers.scikit_learn",
109]
110
111gen_api_init_files(
112    name = "keras_python_api_gen",
113    api_name = "keras",
114    api_version = 1,
115    output_files = KERAS_API_INIT_FILES_V1,
116    output_package = "tensorflow.python.keras.api",
117    package_deps = [
118        "//tensorflow/python/keras",
119        "//tensorflow/python:no_contrib",
120    ],
121    packages = keras_packages,
122)
123
124gen_api_init_files(
125    name = "keras_python_api_gen_compat_v1",
126    api_name = "keras",
127    api_version = 1,
128    output_dir = "_v1/",
129    output_files = KERAS_API_INIT_FILES_V1,
130    output_package = "tensorflow.python.keras.api._v1",
131    package_deps = [
132        "//tensorflow/python/keras",
133        "//tensorflow/python:no_contrib",
134    ],
135    packages = keras_packages,
136)
137
138gen_api_init_files(
139    name = "keras_python_api_gen_compat_v2",
140    api_name = "keras",
141    api_version = 2,
142    output_dir = "_v2/",
143    output_files = KERAS_API_INIT_FILES,
144    output_package = "tensorflow.python.keras.api._v2",
145    package_deps = [
146        "//tensorflow/python/keras",
147        "//tensorflow/python:no_contrib",
148    ],
149    packages = keras_packages,
150)
151