• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1load("//tensorflow:tensorflow.bzl", "py_test")
2
3package(
4    default_visibility = ["//tensorflow:internal"],
5    licenses = ["notice"],
6)
7
8py_library(
9    name = "nest",
10    srcs = ["nest.py"],
11    srcs_version = "PY3",
12    deps = ["//tensorflow/python:util"],
13)
14
15py_test(
16    name = "nest_test",
17    size = "small",
18    srcs = ["nest_test.py"],
19    python_version = "PY3",
20    srcs_version = "PY3",
21    deps = [
22        ":nest",
23        "//tensorflow/python:array_ops",
24        "//tensorflow/python:client_testlib",
25        "//tensorflow/python:framework_for_generated_wrappers",
26        "//tensorflow/python:math_ops",
27        "//tensorflow/python/data/kernel_tests:test_base",
28        "//tensorflow/python/ops/ragged",
29        "//third_party/py/numpy",
30    ],
31)
32
33py_library(
34    name = "sparse",
35    srcs = ["sparse.py"],
36    srcs_version = "PY3",
37    deps = [
38        ":nest",
39        "//tensorflow/python:dtypes",
40        "//tensorflow/python:sparse_ops",
41        "//tensorflow/python:sparse_tensor",
42        "//tensorflow/python:tensor_shape",
43        "//tensorflow/python:util",
44        "//tensorflow/python/user_ops:ops",
45    ],
46)
47
48py_test(
49    name = "sparse_test",
50    size = "small",
51    srcs = ["sparse_test.py"],
52    python_version = "PY3",
53    srcs_version = "PY3",
54    deps = [
55        ":nest",
56        ":sparse",
57        "//tensorflow/python:client_testlib",
58        "//tensorflow/python:dtypes",
59        "//tensorflow/python:sparse_tensor",
60        "//tensorflow/python:tensor_shape",
61        "//tensorflow/python/data/kernel_tests:test_base",
62    ],
63)
64
65py_library(
66    name = "structure",
67    srcs = ["structure.py"],
68    srcs_version = "PY3",
69    deps = [
70        ":nest",
71        "//tensorflow/python:dtypes",
72        "//tensorflow/python:framework_ops",
73        "//tensorflow/python:sparse_ops",
74        "//tensorflow/python:sparse_tensor",
75        "//tensorflow/python:tensor_array_ops",
76        "//tensorflow/python:tensor_shape",
77        "//tensorflow/python:tensor_util",
78        "//tensorflow/python:util",
79        "//tensorflow/python/ops/ragged:ragged_tensor",
80        "//tensorflow/python/user_ops:ops",
81        "@wrapt",
82    ],
83)
84
85py_test(
86    name = "structure_test",
87    size = "small",
88    srcs = ["structure_test.py"],
89    python_version = "PY3",
90    srcs_version = "PY3",
91    deps = [
92        ":nest",
93        ":structure",
94        "//tensorflow/python:array_ops",
95        "//tensorflow/python:client_testlib",
96        "//tensorflow/python:constant_op",
97        "//tensorflow/python:dtypes",
98        "//tensorflow/python:framework_ops",
99        "//tensorflow/python:framework_test_lib",
100        "//tensorflow/python:sparse_tensor",
101        "//tensorflow/python:tensor_array_ops",
102        "//tensorflow/python:tensor_shape",
103        "//tensorflow/python:tensor_spec",
104        "//tensorflow/python:type_spec",
105        "//tensorflow/python:variables",
106        "//tensorflow/python/data/kernel_tests:test_base",
107        "//tensorflow/python/ops/ragged:ragged_factory_ops",
108        "//tensorflow/python/ops/ragged:ragged_tensor",
109        "//tensorflow/python/ops/ragged:ragged_tensor_value",
110        "//third_party/py/numpy",
111        "@absl_py//absl/testing:parameterized",
112        "@wrapt",
113    ],
114)
115
116py_library(
117    name = "options",
118    srcs = ["options.py"],
119    srcs_version = "PY3",
120)
121
122py_test(
123    name = "options_test",
124    size = "small",
125    srcs = ["options_test.py"],
126    python_version = "PY3",
127    srcs_version = "PY3",
128    deps = [
129        ":options",
130        "//tensorflow/python:client_testlib",
131        "//tensorflow/python/data/kernel_tests:test_base",
132    ],
133)
134
135py_library(
136    name = "convert",
137    srcs = ["convert.py"],
138    srcs_version = "PY3",
139    deps = [
140        "//tensorflow/python:constant_op",
141        "//tensorflow/python:dtypes",
142        "//tensorflow/python:framework_ops",
143        "//tensorflow/python:tensor_shape",
144    ],
145)
146
147py_test(
148    name = "convert_test",
149    size = "small",
150    srcs = ["convert_test.py"],
151    python_version = "PY3",
152    srcs_version = "PY3",
153    deps = [
154        ":convert",
155        "//tensorflow/python:client_testlib",
156        "//tensorflow/python:framework_for_generated_wrappers",
157        "//tensorflow/python:util",
158        "//tensorflow/python/data/kernel_tests:test_base",
159    ],
160)
161
162py_library(
163    name = "random_seed",
164    srcs = ["random_seed.py"],
165    srcs_version = "PY3",
166    deps = [
167        "//tensorflow/python:constant_op",
168        "//tensorflow/python:dtypes",
169        "//tensorflow/python:framework",
170    ],
171)
172
173py_test(
174    name = "random_seed_test",
175    size = "small",
176    srcs = ["random_seed_test.py"],
177    python_version = "PY3",
178    srcs_version = "PY3",
179    deps = [
180        ":random_seed",
181        "//tensorflow/python:client_testlib",
182        "//tensorflow/python:framework_for_generated_wrappers",
183        "//tensorflow/python:util",
184        "//tensorflow/python/data/kernel_tests:test_base",
185    ],
186)
187
188py_library(
189    name = "traverse",
190    srcs = ["traverse.py"],
191    srcs_version = "PY3",
192    deps = [
193    ],
194)
195
196py_test(
197    name = "traverse_test",
198    size = "small",
199    srcs = ["traverse_test.py"],
200    python_version = "PY3",
201    srcs_version = "PY3",
202    deps = [
203        ":traverse",
204        "//tensorflow/python:client_testlib",
205        "//tensorflow/python:framework_for_generated_wrappers",
206        "//tensorflow/python/data/experimental/kernel_tests/service:test_base",
207        "//tensorflow/python/data/experimental/ops:dataset_ops",
208        "//tensorflow/python/data/kernel_tests:test_base",
209        "//tensorflow/python/data/ops:dataset_ops",
210    ],
211)
212