• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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"],
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    # Placeholder for internal API
54    "tensorflow.python.keras.initializers",
55    "tensorflow.python.keras.initializers.initializers_v1",
56    "tensorflow.python.keras.initializers.initializers_v2",
57    "tensorflow.python.keras.layers.advanced_activations",
58    "tensorflow.python.keras.layers.convolutional",
59    "tensorflow.python.keras.layers.convolutional_recurrent",
60    "tensorflow.python.keras.layers.core",
61    "tensorflow.python.keras.layers.cudnn_recurrent",
62    "tensorflow.python.keras.layers.dense_attention",
63    "tensorflow.python.keras.layers.embeddings",
64    "tensorflow.python.keras.layers.legacy_rnn.rnn_cell_impl",
65    "tensorflow.python.keras.layers.local",
66    "tensorflow.python.keras.layers.merge",
67    "tensorflow.python.keras.layers.noise",
68    "tensorflow.python.keras.layers.normalization.batch_normalization",
69    "tensorflow.python.keras.layers.normalization.batch_normalization_v1",
70    "tensorflow.python.keras.layers.normalization.layer_normalization",
71    "tensorflow.python.keras.layers.preprocessing",
72    "tensorflow.python.keras.layers.pooling",
73    "tensorflow.python.keras.layers.recurrent",
74    "tensorflow.python.keras.layers.recurrent_v2",
75    "tensorflow.python.keras.layers.serialization",
76    "tensorflow.python.keras.layers.wrappers",
77    "tensorflow.python.keras.legacy_tf_layers.base",
78    "tensorflow.python.keras.legacy_tf_layers.convolutional",
79    "tensorflow.python.keras.legacy_tf_layers.core",
80    "tensorflow.python.keras.legacy_tf_layers.normalization",
81    "tensorflow.python.keras.legacy_tf_layers.pooling",
82    "tensorflow.python.keras.losses",
83    "tensorflow.python.keras.metrics",
84    "tensorflow.python.keras.mixed_precision.get_layer_policy",
85    "tensorflow.python.keras.mixed_precision.loss_scale_optimizer",
86    "tensorflow.python.keras.mixed_precision.policy",
87    "tensorflow.python.keras.models",
88    "tensorflow.python.keras.optimizer_v2.adadelta",
89    "tensorflow.python.keras.optimizer_v2.adagrad",
90    "tensorflow.python.keras.optimizer_v2.adam",
91    "tensorflow.python.keras.optimizer_v2.adamax",
92    "tensorflow.python.keras.optimizer_v2.ftrl",
93    "tensorflow.python.keras.optimizer_v2.gradient_descent",
94    "tensorflow.python.keras.optimizer_v2.learning_rate_schedule",
95    "tensorflow.python.keras.optimizer_v2.nadam",
96    "tensorflow.python.keras.optimizer_v2.optimizer_v2",
97    "tensorflow.python.keras.optimizer_v2.rmsprop",
98    "tensorflow.python.keras.optimizers",
99    "tensorflow.python.keras.premade.linear",
100    "tensorflow.python.keras.premade.wide_deep",
101    "tensorflow.python.keras.preprocessing.image",
102    "tensorflow.python.keras.preprocessing.sequence",
103    "tensorflow.python.keras.preprocessing.text",
104    "tensorflow.python.keras.regularizers",
105    "tensorflow.python.keras.saving.model_config",
106    "tensorflow.python.keras.saving.save",
107    "tensorflow.python.keras.saving.saved_model_experimental",
108    "tensorflow.python.keras.utils.data_utils",
109    "tensorflow.python.keras.utils.generic_utils",
110    "tensorflow.python.keras.utils.io_utils",
111    "tensorflow.python.keras.utils.layer_utils",
112    "tensorflow.python.keras.utils.losses_utils",
113    "tensorflow.python.keras.utils.multi_gpu_utils",
114    "tensorflow.python.keras.utils.np_utils",
115    "tensorflow.python.keras.utils.tf_utils",
116    "tensorflow.python.keras.utils.vis_utils",
117    "tensorflow.python.keras.wrappers.scikit_learn",
118]
119
120gen_api_init_files(
121    name = "keras_python_api_gen",
122    api_name = "keras",
123    api_version = 1,
124    output_files = KERAS_API_INIT_FILES_V1,
125    output_package = "tensorflow.python.keras.api",
126    package_deps = [
127        "//tensorflow/python/keras",
128        "//tensorflow/python:no_contrib",
129    ],
130    packages = keras_packages,
131)
132
133gen_api_init_files(
134    name = "keras_python_api_gen_compat_v1",
135    api_name = "keras",
136    api_version = 1,
137    output_dir = "_v1/",
138    output_files = KERAS_API_INIT_FILES_V1,
139    output_package = "tensorflow.python.keras.api._v1",
140    package_deps = [
141        "//tensorflow/python/keras",
142        "//tensorflow/python:no_contrib",
143    ],
144    packages = keras_packages,
145)
146
147gen_api_init_files(
148    name = "keras_python_api_gen_compat_v2",
149    api_name = "keras",
150    api_version = 2,
151    output_dir = "_v2/",
152    output_files = KERAS_API_INIT_FILES,
153    output_package = "tensorflow.python.keras.api._v2",
154    package_deps = [
155        "//tensorflow/python/keras",
156        "//tensorflow/python:no_contrib",
157    ],
158    packages = keras_packages,
159)
160