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