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