• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1package(
2    default_visibility = ["//tensorflow:internal"],
3    licenses = ["notice"],
4)
5
6py_library(
7    name = "estimator_py",
8    srcs = [
9        "estimator_lib.py",
10    ],
11    srcs_version = "PY3",
12    visibility = [
13        "//tensorflow:__pkg__",
14        "//tensorflow:internal",
15    ],
16    deps = [
17        ":baseline",
18        ":dnn",
19        ":dnn_linear_combined",
20        ":estimator",
21        ":expect_tensorflow_estimator_installed",
22        ":export",
23        ":exporter",
24        ":inputs",
25        ":keras",
26        ":linear",
27        ":model_fn",
28        ":parsing_utils",
29        ":run_config",
30        ":training",
31        "//tensorflow:tensorflow_py_no_contrib",
32    ],
33)
34
35py_library(
36    name = "exporter",
37    srcs = ["exporter.py"],
38    srcs_version = "PY3",
39    deps = [
40        ":expect_tensorflow_estimator_installed",
41        ":gc",
42        ":metric_keys",
43        ":util",
44        "//tensorflow:tensorflow_py_no_contrib",
45    ],
46)
47
48py_library(
49    name = "gc",
50    srcs = ["gc.py"],
51    srcs_version = "PY3",
52    deps = [
53        ":expect_tensorflow_estimator_installed",
54        "//tensorflow:tensorflow_py_no_contrib",
55    ],
56)
57
58py_library(
59    name = "model_fn",
60    srcs = ["model_fn.py"],
61    srcs_version = "PY3",
62    deps = [
63        ":expect_tensorflow_estimator_installed",
64        ":export_output",
65        "//tensorflow:tensorflow_py_no_contrib",
66        "@six_archive//:six",
67    ],
68)
69
70py_library(
71    name = "training",
72    srcs = ["training.py"],
73    srcs_version = "PY3",
74    deps = [
75        ":estimator",
76        ":expect_tensorflow_estimator_installed",
77        ":exporter",
78        ":run_config",
79        "//tensorflow:tensorflow_py_no_contrib",
80        "@six_archive//:six",
81    ],
82)
83
84py_library(
85    name = "run_config",
86    srcs = ["run_config.py"],
87    srcs_version = "PY3",
88    deps = [
89        ":expect_tensorflow_estimator_installed",
90        "//tensorflow:tensorflow_py_no_contrib",
91        "@six_archive//:six",
92    ],
93)
94
95py_library(
96    name = "baseline",
97    srcs = ["canned/baseline.py"],
98    srcs_version = "PY3",
99    deps = [
100        ":estimator",
101        ":expect_tensorflow_estimator_installed",
102        ":head",
103        ":model_fn",
104        ":optimizers",
105        "//tensorflow:tensorflow_py_no_contrib",
106        "@six_archive//:six",
107    ],
108)
109
110py_library(
111    name = "dnn",
112    srcs = ["canned/dnn.py"],
113    srcs_version = "PY3",
114    deps = [
115        ":estimator",
116        ":expect_tensorflow_estimator_installed",
117        ":head",
118        ":model_fn",
119        ":optimizers",
120        "//tensorflow:tensorflow_py_no_contrib",
121        "@six_archive//:six",
122    ],
123)
124
125py_library(
126    name = "dnn_testing_utils",
127    testonly = 1,
128    srcs = ["canned/dnn_testing_utils.py"],
129    srcs_version = "PY3",
130    deps = [
131        ":estimator",
132        ":expect_tensorflow_estimator_installed",
133        ":head",
134        ":metric_keys",
135        ":model_fn",
136        ":numpy_io",
137        ":prediction_keys",
138        "//tensorflow:tensorflow_py_no_contrib",
139        "//third_party/py/numpy",
140        "@absl_py//absl/testing:parameterized",
141        "@six_archive//:six",
142    ],
143)
144
145py_library(
146    name = "dnn_linear_combined",
147    srcs = ["canned/dnn_linear_combined.py"],
148    srcs_version = "PY3",
149    deps = [
150        ":dnn",
151        ":estimator",
152        ":expect_tensorflow_estimator_installed",
153        ":head",
154        ":linear",
155        ":model_fn",
156        ":optimizers",
157        "//tensorflow:tensorflow_py_no_contrib",
158        "@six_archive//:six",
159    ],
160)
161
162py_library(
163    name = "util",
164    srcs = [
165        "util.py",
166    ],
167    srcs_version = "PY3",
168    deps = [
169        ":expect_tensorflow_estimator_installed",
170        "//tensorflow:tensorflow_py_no_contrib",
171    ],
172)
173
174py_library(
175    name = "estimator",
176    srcs = [
177        "estimator.py",
178    ],
179    srcs_version = "PY3",
180    deps = [
181        ":expect_tensorflow_estimator_installed",
182        ":export_export",
183        ":model_fn",
184        ":run_config",
185        ":util",
186        "//tensorflow:tensorflow_py_no_contrib",
187        "//third_party/py/numpy",
188        "@six_archive//:six",
189    ],
190)
191
192py_library(
193    name = "parsing_utils",
194    srcs = [
195        "canned/parsing_utils.py",
196    ],
197    srcs_version = "PY3",
198    deps = [
199        ":expect_tensorflow_estimator_installed",
200        "//tensorflow:tensorflow_py_no_contrib",
201        "@six_archive//:six",
202    ],
203)
204
205py_library(
206    name = "export_output",
207    srcs = ["export/export_output.py"],
208    srcs_version = "PY3",
209    deps = [
210        ":expect_tensorflow_estimator_installed",
211        "//tensorflow:tensorflow_py_no_contrib",
212        "@six_archive//:six",
213    ],
214)
215
216py_library(
217    name = "export",
218    srcs = [
219        "export/export_lib.py",
220    ],
221    srcs_version = "PY3",
222    deps = [
223        ":expect_tensorflow_estimator_installed",
224        ":export_export",
225        ":export_output",
226        "//tensorflow:tensorflow_py_no_contrib",
227    ],
228)
229
230py_library(
231    name = "export_export",
232    srcs = [
233        "export/export.py",
234    ],
235    srcs_version = "PY3",
236    deps = [
237        ":expect_tensorflow_estimator_installed",
238        ":util",
239        "//tensorflow:tensorflow_py_no_contrib",
240        "@six_archive//:six",
241    ],
242)
243
244py_library(
245    name = "head",
246    srcs = ["canned/head.py"],
247    srcs_version = "PY3",
248    deps = [
249        ":expect_tensorflow_estimator_installed",
250        ":export_output",
251        ":metric_keys",
252        ":model_fn",
253        ":prediction_keys",
254        "//tensorflow:tensorflow_py_no_contrib",
255        "@six_archive//:six",
256    ],
257)
258
259py_library(
260    name = "inputs",
261    srcs = ["inputs/inputs.py"],
262    srcs_version = "PY3",
263    deps = [
264        ":expect_tensorflow_estimator_installed",
265        ":numpy_io",
266        ":pandas_io",
267        "//tensorflow:tensorflow_py_no_contrib",
268    ],
269)
270
271py_library(
272    name = "linear",
273    srcs = ["canned/linear.py"],
274    srcs_version = "PY3",
275    deps = [
276        ":estimator",
277        ":expect_tensorflow_estimator_installed",
278        ":head",
279        ":optimizers",
280        "//tensorflow:tensorflow_py_no_contrib",
281        "@six_archive//:six",
282    ],
283)
284
285py_library(
286    name = "linear_testing_utils",
287    testonly = 1,
288    srcs = ["canned/linear_testing_utils.py"],
289    srcs_version = "PY3",
290    deps = [
291        ":estimator",
292        ":expect_tensorflow_estimator_installed",
293        ":export_export",
294        ":linear",
295        ":metric_keys",
296        ":numpy_io",
297        ":pandas_io",
298        ":run_config",
299        "//tensorflow:tensorflow_py_no_contrib",
300        "@six_archive//:six",
301    ],
302)
303
304py_library(
305    name = "metric_keys",
306    srcs = ["canned/metric_keys.py"],
307    srcs_version = "PY3",
308    deps = [
309        ":expect_tensorflow_estimator_installed",
310        ":model_fn",
311    ],
312)
313
314py_library(
315    name = "numpy_io",
316    srcs = ["inputs/numpy_io.py"],
317    srcs_version = "PY3",
318    deps = [
319        ":expect_tensorflow_estimator_installed",
320        ":inputs_queues",
321    ],
322)
323
324py_library(
325    name = "optimizers",
326    srcs = ["canned/optimizers.py"],
327    srcs_version = "PY3",
328    deps = [
329        ":expect_tensorflow_estimator_installed",
330        "//tensorflow:tensorflow_py_no_contrib",
331        "@six_archive//:six",
332    ],
333)
334
335py_library(
336    name = "pandas_io",
337    srcs = ["inputs/pandas_io.py"],
338    srcs_version = "PY3",
339    deps = [
340        ":expect_tensorflow_estimator_installed",
341        ":inputs_queues",
342    ],
343)
344
345py_library(
346    name = "prediction_keys",
347    srcs = ["canned/prediction_keys.py"],
348    srcs_version = "PY3",
349    deps = [
350        ":expect_tensorflow_estimator_installed",
351    ],
352)
353
354py_library(
355    name = "inputs_queues",
356    srcs = [
357        "inputs/queues/feeding_functions.py",
358        "inputs/queues/feeding_queue_runner.py",
359    ],
360    srcs_version = "PY3",
361    deps = [
362        ":expect_tensorflow_estimator_installed",
363        "//tensorflow:tensorflow_py_no_contrib",
364        "@six_archive//:six",
365    ],
366)
367
368py_library(
369    name = "keras",
370    srcs = ["keras.py"],
371    srcs_version = "PY3",
372    deps = [
373        ":estimator",
374        ":expect_tensorflow_estimator_installed",
375        ":export_export",
376        ":model_fn",
377        ":run_config",
378        "//tensorflow:tensorflow_py_no_contrib",
379    ],
380)
381
382py_library(
383    name = "expect_tensorflow_estimator_installed",
384    srcs_version = "PY3",
385    # This is a dummy rule used as a dependency in open-source.
386    # We expect tensorflow_estimator to already be installed.
387    visibility = ["//visibility:public"],
388)
389