1package(
2    default_visibility = ["//tensorflow:internal"],
3    licenses = ["notice"],  # Apache 2.0
4)
5
6exports_files(["LICENSE"])
7
8py_library(
9    name = "batching",
10    srcs = ["batching.py"],
11    srcs_version = "PY3",
12    deps = [
13        "//tensorflow/python:dtypes",
14        "//tensorflow/python:experimental_dataset_ops_gen",
15        "//tensorflow/python:framework_ops",
16        "//tensorflow/python:tensor_shape",
17        "//tensorflow/python:tensor_util",
18        "//tensorflow/python:util",
19        "//tensorflow/python/data/ops:dataset_ops",
20        "//tensorflow/python/data/util:convert",
21        "//tensorflow/python/data/util:nest",
22        "//tensorflow/python/data/util:structure",
23    ],
24)
25
26py_library(
27    name = "cardinality",
28    srcs = ["cardinality.py"],
29    srcs_version = "PY3",
30    deps = [
31        "//tensorflow/python:experimental_dataset_ops_gen",
32        "//tensorflow/python:util",
33    ],
34)
35
36py_library(
37    name = "compression_ops",
38    srcs = ["compression_ops.py"],
39    srcs_version = "PY3",
40    deps = [
41        "//tensorflow/python:experimental_dataset_ops_gen",
42    ],
43)
44
45py_library(
46    name = "counter",
47    srcs = ["counter.py"],
48    srcs_version = "PY3",
49    deps = [
50        ":scan_ops",
51        "//tensorflow/python:dtypes",
52        "//tensorflow/python:framework_ops",
53        "//tensorflow/python:util",
54        "//tensorflow/python/data/ops:dataset_ops",
55    ],
56)
57
58py_library(
59    name = "data_service_ops",
60    srcs = [
61        "data_service_ops.py",
62    ],
63    srcs_version = "PY3",
64    deps = [
65        "//tensorflow/python:experimental_dataset_ops_gen",
66        "//tensorflow/python:framework_ops",
67        "//tensorflow/python/data/ops:dataset_ops",
68        "//tensorflow/python/data/util:nest",
69    ],
70)
71
72py_library(
73    name = "distribute",
74    srcs = [
75        "distribute.py",
76    ],
77    srcs_version = "PY3",
78    deps = [
79        "//tensorflow/python:experimental_dataset_ops_gen",
80        "//tensorflow/python:framework_ops",
81        "//tensorflow/python/data/ops:dataset_ops",
82        "//tensorflow/python/data/util:nest",
83    ],
84)
85
86py_library(
87    name = "distribute_options",
88    srcs = ["distribute_options.py"],
89    srcs_version = "PY3",
90    deps = [
91        "//tensorflow/python:util",
92        "//tensorflow/python/data/util:options",
93    ],
94)
95
96py_library(
97    name = "enumerate_ops",
98    srcs = ["enumerate_ops.py"],
99    srcs_version = "PY3",
100    deps = [
101        "//tensorflow/python:dtypes",
102        "//tensorflow/python/data/ops:dataset_ops",
103    ],
104)
105
106py_library(
107    name = "error_ops",
108    srcs = ["error_ops.py"],
109    srcs_version = "PY3",
110    deps = [
111        "//tensorflow/python:experimental_dataset_ops_gen",
112        "//tensorflow/python/data/ops:dataset_ops",
113        "//tensorflow/python/data/util:nest",
114        "//tensorflow/python/data/util:sparse",
115    ],
116)
117
118py_library(
119    name = "get_single_element",
120    srcs = ["get_single_element.py"],
121    srcs_version = "PY3",
122    deps = [
123        "//tensorflow/python:dataset_ops_gen",
124        "//tensorflow/python/data/ops:dataset_ops",
125        "//tensorflow/python/data/util:nest",
126        "//tensorflow/python/data/util:sparse",
127    ],
128)
129
130py_library(
131    name = "grouping",
132    srcs = ["grouping.py"],
133    srcs_version = "PY3",
134    deps = [
135        "//tensorflow/python:array_ops",
136        "//tensorflow/python:check_ops",
137        "//tensorflow/python:dtypes",
138        "//tensorflow/python:framework_ops",
139        "//tensorflow/python:function",
140        "//tensorflow/python:math_ops",
141        "//tensorflow/python:tensor_shape",
142        "//tensorflow/python/data/ops:dataset_ops",
143        "//tensorflow/python/data/util:nest",
144        "//tensorflow/python/data/util:structure",
145    ],
146)
147
148py_library(
149    name = "interleave_ops",
150    srcs = ["interleave_ops.py"],
151    srcs_version = "PY3",
152    deps = [
153        ":random_ops",
154        "//tensorflow/python:array_ops",
155        "//tensorflow/python:dtypes",
156        "//tensorflow/python:experimental_dataset_ops_gen",
157        "//tensorflow/python:framework_ops",
158        "//tensorflow/python:math_ops",
159        "//tensorflow/python:stateless_random_ops_gen",
160        "//tensorflow/python:util",
161        "//tensorflow/python/data/ops:readers",
162        "//tensorflow/python/data/util:nest",
163        "//tensorflow/python/data/util:sparse",
164    ],
165)
166
167py_library(
168    name = "io",
169    srcs = [
170        "io.py",
171    ],
172    srcs_version = "PY3",
173    deps = [
174        "//tensorflow/python:experimental_dataset_ops_gen",
175        "//tensorflow/python/data/ops:dataset_ops",
176    ],
177)
178
179py_library(
180    name = "iterator_ops",
181    srcs = [
182        "iterator_ops.py",
183    ],
184    srcs_version = "PY3",
185    deps = [
186        "//tensorflow/python:basic_session_run_hooks",
187        "//tensorflow/python:checkpoint_management",
188        "//tensorflow/python:dataset_ops_gen",
189        "//tensorflow/python:framework_ops",
190        "//tensorflow/python:saver",
191        "//tensorflow/python:session_run_hook",
192        "//tensorflow/python/data/ops:iterator_ops",
193        "//tensorflow/python/data/ops:optional_ops",
194    ],
195)
196
197py_library(
198    name = "map_defun",
199    srcs = ["map_defun.py"],
200    srcs_version = "PY3",
201    deps = [
202        "//tensorflow/python:dataset_ops_gen",
203        "//tensorflow/python:framework_ops",
204        "//tensorflow/python:tensor_shape",
205    ],
206)
207
208py_library(
209    name = "matching_files",
210    srcs = ["matching_files.py"],
211    srcs_version = "PY3",
212    deps = [
213        "//tensorflow/python:dataset_ops_gen",
214        "//tensorflow/python:framework_ops",
215        "//tensorflow/python:tensor_shape",
216        "//tensorflow/python/data/ops:dataset_ops",
217        "//tensorflow/python/data/util:structure",
218    ],
219)
220
221py_library(
222    name = "optimization",
223    srcs = ["optimization.py"],
224    srcs_version = "PY3",
225    deps = [
226        "//tensorflow/python:dtypes",
227        "//tensorflow/python:experimental_dataset_ops_gen",
228        "//tensorflow/python:framework_ops",
229        "//tensorflow/python/data/util:nest",
230        "//tensorflow/python/data/util:sparse",
231    ],
232)
233
234py_library(
235    name = "optimization_options",
236    srcs = ["optimization_options.py"],
237    srcs_version = "PY3",
238    deps = [
239        "//tensorflow/python:util",
240        "//tensorflow/python/data/util:options",
241    ],
242)
243
244py_library(
245    name = "parsing_ops",
246    srcs = ["parsing_ops.py"],
247    srcs_version = "PY3",
248    deps = [
249        "//tensorflow/python:dataset_ops_gen",
250        "//tensorflow/python:dtypes",
251        "//tensorflow/python:framework_ops",
252        "//tensorflow/python:parsing_ops",
253        "//tensorflow/python:sparse_tensor",
254        "//tensorflow/python:tensor_shape",
255        "//tensorflow/python/data/ops:dataset_ops",
256        "//tensorflow/python/data/util:structure",
257    ],
258)
259
260py_library(
261    name = "prefetching_ops",
262    srcs = ["prefetching_ops.py"],
263    srcs_version = "PY3",
264    deps = [
265        "//tensorflow/python:experimental_dataset_ops_gen",
266        "//tensorflow/python:framework_ops",
267        "//tensorflow/python:tensor_spec",
268        "//tensorflow/python/data/ops:dataset_ops",
269        "//tensorflow/python/data/util:nest",
270        "//tensorflow/python/data/util:sparse",
271        "//tensorflow/python/eager:context",
272        "//tensorflow/python/eager:function",
273    ],
274)
275
276py_library(
277    name = "random_ops",
278    srcs = [
279        "random_ops.py",
280    ],
281    srcs_version = "PY3",
282    deps = [
283        "//tensorflow/python:constant_op",
284        "//tensorflow/python:dtypes",
285        "//tensorflow/python:experimental_dataset_ops_gen",
286        "//tensorflow/python:framework_ops",
287        "//tensorflow/python:random_seed",
288        "//tensorflow/python:tensor_shape",
289        "//tensorflow/python/data/ops:dataset_ops",
290        "//tensorflow/python/data/util:structure",
291    ],
292)
293
294py_library(
295    name = "readers",
296    srcs = [
297        "readers.py",
298    ],
299    srcs_version = "PY3",
300    deps = [
301        ":batching",
302        ":error_ops",
303        ":interleave_ops",
304        ":optimization",
305        ":parsing_ops",
306        ":shuffle_ops",
307        "//tensorflow/python:constant_op",
308        "//tensorflow/python:dataset_ops_gen",
309        "//tensorflow/python:dtypes",
310        "//tensorflow/python:experimental_dataset_ops_gen",
311        "//tensorflow/python:framework_ops",
312        "//tensorflow/python:io_ops",
313        "//tensorflow/python:lib",
314        "//tensorflow/python:platform",
315        "//tensorflow/python:tensor_shape",
316        "//tensorflow/python:util",
317        "//tensorflow/python/data/ops:dataset_ops",
318        "//tensorflow/python/data/ops:readers",
319        "//tensorflow/python/data/util:convert",
320        "//tensorflow/python/data/util:nest",
321        "//third_party/py/numpy",
322    ],
323)
324
325py_library(
326    name = "resampling",
327    srcs = ["resampling.py"],
328    srcs_version = "PY3",
329    deps = [
330        ":batching",
331        ":interleave_ops",
332        ":scan_ops",
333        "//tensorflow/python:array_ops",
334        "//tensorflow/python:control_flow_ops",
335        "//tensorflow/python:dtypes",
336        "//tensorflow/python:framework_ops",
337        "//tensorflow/python:logging_ops",
338        "//tensorflow/python:math_ops",
339        "//tensorflow/python:random_ops",
340        "//tensorflow/python/data/ops:dataset_ops",
341        "//third_party/py/numpy",
342    ],
343)
344
345py_library(
346    name = "scan_ops",
347    srcs = ["scan_ops.py"],
348    srcs_version = "PY3",
349    deps = [
350        "//tensorflow/python:experimental_dataset_ops_gen",
351        "//tensorflow/python:framework_ops",
352        "//tensorflow/python:function",
353        "//tensorflow/python:tensor_array_ops",
354        "//tensorflow/python/data/ops:dataset_ops",
355        "//tensorflow/python/data/util:nest",
356        "//tensorflow/python/data/util:sparse",
357        "//tensorflow/python/data/util:structure",
358    ],
359)
360
361py_library(
362    name = "shuffle_ops",
363    srcs = [
364        "shuffle_ops.py",
365    ],
366    srcs_version = "PY3",
367    deps = [
368        "//tensorflow/python/data/ops:dataset_ops",
369    ],
370)
371
372py_library(
373    name = "snapshot",
374    srcs = [
375        "snapshot.py",
376    ],
377    srcs_version = "PY3",
378    visibility = ["//tensorflow:internal"],
379    deps = [
380        "//tensorflow/python:experimental_dataset_ops_gen",
381        "//tensorflow/python/data/ops:dataset_ops",
382        "//tensorflow/python/data/util:structure",
383    ],
384)
385
386py_library(
387    name = "stats_aggregator",
388    srcs = ["stats_aggregator.py"],
389    srcs_version = "PY3",
390    deps = [
391        "//tensorflow/python:dataset_ops_gen",
392        "//tensorflow/python:util",
393    ],
394)
395
396py_library(
397    name = "stats_ops",
398    srcs = ["stats_ops.py"],
399    srcs_version = "PY3",
400    deps = [
401        "//tensorflow/python:dataset_ops_gen",
402        "//tensorflow/python:dtypes",
403        "//tensorflow/python:framework_ops",
404        "//tensorflow/python/data/ops:dataset_ops",
405        "//tensorflow/python/data/ops:iterator_ops",
406        "//tensorflow/python/data/util:nest",
407        "//tensorflow/python/data/util:sparse",
408    ],
409)
410
411py_library(
412    name = "stats_options",
413    srcs = ["stats_options.py"],
414    srcs_version = "PY3",
415    deps = [
416        ":stats_aggregator",
417        "//tensorflow/python:util",
418        "//tensorflow/python/data/util:options",
419    ],
420)
421
422py_library(
423    name = "take_while_ops",
424    srcs = ["take_while_ops.py"],
425    srcs_version = "PY3",
426    deps = [
427        "//tensorflow/python:dtypes",
428        "//tensorflow/python:experimental_dataset_ops_gen",
429        "//tensorflow/python:framework_ops",
430        "//tensorflow/python:function",
431        "//tensorflow/python/data/ops:dataset_ops",
432    ],
433)
434
435py_library(
436    name = "testing",
437    srcs = ["testing.py"],
438    srcs_version = "PY3",
439    deps = [
440        "//tensorflow/python:experimental_dataset_ops_gen",
441        "//tensorflow/python:framework_ops",
442        "//tensorflow/python/data/ops:dataset_ops",
443    ],
444)
445
446py_library(
447    name = "threading_options",
448    srcs = ["threading_options.py"],
449    srcs_version = "PY3",
450    deps = [
451        "//tensorflow/python:util",
452        "//tensorflow/python/data/util:options",
453    ],
454)
455
456py_library(
457    name = "threadpool",
458    srcs = ["threadpool.py"],
459    srcs_version = "PY3",
460    deps = [
461        "//tensorflow/python:experimental_dataset_ops_gen",
462        "//tensorflow/python:resource_variable_ops",
463        "//tensorflow/python:util",
464        "//tensorflow/python/data/ops:dataset_ops",
465        "//tensorflow/python/eager:context",
466    ],
467)
468
469py_library(
470    name = "unique",
471    srcs = [
472        "unique.py",
473    ],
474    srcs_version = "PY3",
475    deps = [
476        "//tensorflow/python:dtypes",
477        "//tensorflow/python:experimental_dataset_ops_gen",
478        "//tensorflow/python/data/ops:dataset_ops",
479        "//tensorflow/python/data/util:nest",
480        "//tensorflow/python/data/util:sparse",
481    ],
482)
483
484py_library(
485    name = "writers",
486    srcs = [
487        "writers.py",
488    ],
489    srcs_version = "PY3",
490    deps = [
491        "//tensorflow/python:dtypes",
492        "//tensorflow/python/data/ops:dataset_ops",
493    ],
494)
495
496py_library(
497    name = "dataset_ops",
498    srcs_version = "PY3",
499    deps = [
500        ":batching",
501        ":cardinality",
502        ":compression_ops",
503        ":counter",
504        ":data_service_ops",
505        ":distribute",
506        ":enumerate_ops",
507        ":error_ops",
508        ":get_single_element",
509        ":grouping",
510        ":interleave_ops",
511        ":io",
512        ":map_defun",
513        ":matching_files",
514        ":optimization",
515        ":prefetching_ops",
516        ":readers",
517        ":resampling",
518        ":scan_ops",
519        ":shuffle_ops",
520        ":snapshot",
521        ":stats_ops",
522        ":take_while_ops",
523        ":threadpool",
524        ":unique",
525        ":writers",
526        "//tensorflow/python:dataset_ops_gen",
527        "//tensorflow/python:util",
528        "//tensorflow/python/data/ops:dataset_ops",
529        "//tensorflow/python/data/util:nest",
530    ],
531)
532