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