• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1package(
2    default_visibility = ["//tensorflow:internal"],
3    licenses = ["notice"],  # Apache 2.0
4)
5
6exports_files(["LICENSE"])
7
8py_library(
9    name = "dataset_ops",
10    srcs = ["dataset_ops.py"],
11    srcs_version = "PY3",
12    deps = [
13        ":iterator_ops",
14        "//tensorflow/python:constant_op",
15        "//tensorflow/python:control_flow_ops",
16        "//tensorflow/python:dataset_ops_gen",
17        "//tensorflow/python:dtypes",
18        "//tensorflow/python:experimental_dataset_ops_gen",
19        "//tensorflow/python:framework_ops",
20        "//tensorflow/python:function",
21        "//tensorflow/python:math_ops",
22        "//tensorflow/python:random_seed",
23        "//tensorflow/python:script_ops",
24        "//tensorflow/python:smart_cond",
25        "//tensorflow/python:sparse_tensor",
26        "//tensorflow/python:string_ops",
27        "//tensorflow/python:tensor_shape",
28        "//tensorflow/python:tensor_util",
29        "//tensorflow/python:util",
30        "//tensorflow/python/data/experimental/ops:distribute_options",
31        "//tensorflow/python/data/experimental/ops:optimization_options",
32        "//tensorflow/python/data/experimental/ops:stats_options",
33        "//tensorflow/python/data/experimental/ops:threading_options",
34        "//tensorflow/python/data/util:convert",
35        "//tensorflow/python/data/util:nest",
36        "//tensorflow/python/data/util:options",
37        "//tensorflow/python/data/util:random_seed",
38        "//tensorflow/python/data/util:sparse",
39        "//tensorflow/python/data/util:structure",
40        "//tensorflow/python/data/util:traverse",
41        "//tensorflow/python/ops/ragged",
42        "//third_party/py/numpy",
43    ],
44)
45
46py_library(
47    name = "readers",
48    srcs = ["readers.py"],
49    srcs_version = "PY3",
50    deps = [
51        ":dataset_ops",
52        "//tensorflow/python:array_ops",
53        "//tensorflow/python:dataset_ops_gen",
54        "//tensorflow/python:dtypes",
55        "//tensorflow/python:experimental_dataset_ops_gen",
56        "//tensorflow/python:framework_ops",
57        "//tensorflow/python:tensor_shape",
58        "//tensorflow/python:tensor_spec",
59        "//tensorflow/python:tf2",
60        "//tensorflow/python:util",
61        "//tensorflow/python/data/util:convert",
62        "//tensorflow/python/util:tf_export",
63    ],
64)
65
66py_library(
67    name = "iterator_ops",
68    srcs = ["iterator_ops.py"],
69    srcs_version = "PY3",
70    deps = [
71        ":optional_ops",
72        "//tensorflow/python:dataset_ops_gen",
73        "//tensorflow/python:dtypes",
74        "//tensorflow/python:framework_ops",
75        "//tensorflow/python:resource_variable_ops",
76        "//tensorflow/python:saver",
77        "//tensorflow/python:tensor_shape",
78        "//tensorflow/python/compat",
79        "//tensorflow/python/data/util:nest",
80        "//tensorflow/python/data/util:sparse",
81        "//tensorflow/python/data/util:structure",
82        "//tensorflow/python/eager:context",
83        "//tensorflow/python/training/tracking:base",
84    ],
85)
86
87py_library(
88    name = "optional_ops",
89    srcs = ["optional_ops.py"],
90    srcs_version = "PY3",
91    deps = [
92        "//tensorflow/python:dataset_ops_gen",
93        "//tensorflow/python:dtypes",
94        "//tensorflow/python:framework_ops",
95        "//tensorflow/python:resource_variable_ops",
96        "//tensorflow/python:tensor_shape",
97        "//tensorflow/python:util",
98        "//tensorflow/python/data/util:structure",
99    ],
100)
101
102py_library(
103    name = "multi_device_iterator_ops",
104    srcs = ["multi_device_iterator_ops.py"],
105    srcs_version = "PY3",
106    deps = [
107        ":dataset_ops",
108        ":iterator_ops",
109        "//tensorflow/python:array_ops",
110        "//tensorflow/python:control_flow_ops",
111        "//tensorflow/python:dataset_ops_gen",
112        "//tensorflow/python:dtypes",
113        "//tensorflow/python:framework_ops",
114        "//tensorflow/python:functional_ops",
115        "//tensorflow/python:tensor_spec",
116        "//tensorflow/python/data/util:nest",
117        "//tensorflow/python/data/util:sparse",
118        "//tensorflow/python/eager:context",
119        "//tensorflow/python/eager:function",
120    ],
121)
122