• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# Tests of TensorFlow kernels written using the Python API.
2
3load("//tensorflow:tensorflow.bzl", "cuda_py_test")
4
5package(
6    default_visibility = ["//tensorflow:internal"],
7    licenses = ["notice"],  # Apache 2.0
8)
9
10cuda_py_test(
11    name = "bijector_test",
12    size = "small",
13    srcs = ["bijector_test.py"],
14    deps = [
15        "//tensorflow/python:array_ops",
16        "//tensorflow/python:client_testlib",
17        "//tensorflow/python:framework_for_generated_wrappers",
18        "//tensorflow/python:framework_test_lib",
19        "//tensorflow/python:math_ops",
20        "//tensorflow/python:platform_test",
21        "//tensorflow/python/ops/distributions",
22        "//third_party/py/numpy",
23        "@six_archive//:six",
24    ],
25)
26
27cuda_py_test(
28    name = "util_test",
29    size = "medium",
30    srcs = ["util_test.py"],
31    shard_count = 3,
32    xla_enable_strict_auto_jit = False,  # TODO(b/144920376)
33    deps = [
34        "//tensorflow/python:array_ops",
35        "//tensorflow/python:client_testlib",
36        "//tensorflow/python:framework",
37        "//tensorflow/python:framework_for_generated_wrappers",
38        "//tensorflow/python:framework_test_lib",
39        "//tensorflow/python:gradients",
40        "//tensorflow/python:math_ops",
41        "//tensorflow/python:platform_test",
42        "//tensorflow/python/ops/distributions",
43        "//third_party/py/numpy",
44    ],
45)
46
47cuda_py_test(
48    name = "kullback_leibler_test",
49    size = "small",
50    srcs = ["kullback_leibler_test.py"],
51    deps = [
52        "//tensorflow/python:array_ops",
53        "//tensorflow/python:client_testlib",
54        "//tensorflow/python:platform_test",
55        "//tensorflow/python/ops/distributions",
56    ],
57)
58
59cuda_py_test(
60    name = "beta_test",
61    size = "small",
62    srcs = ["beta_test.py"],
63    deps = [
64        "//tensorflow/python:client",
65        "//tensorflow/python:client_testlib",
66        "//tensorflow/python:framework",
67        "//tensorflow/python:framework_for_generated_wrappers",
68        "//tensorflow/python:math_ops",
69        "//tensorflow/python:nn_ops",
70        "//tensorflow/python:platform_test",
71        "//tensorflow/python/ops/distributions",
72        "//third_party/py/numpy",
73    ],
74)
75
76cuda_py_test(
77    name = "bernoulli_test",
78    size = "small",
79    srcs = ["bernoulli_test.py"],
80    deps = [
81        "//tensorflow/python:array_ops",
82        "//tensorflow/python:client_testlib",
83        "//tensorflow/python:framework_for_generated_wrappers",
84        "//tensorflow/python:math_ops",
85        "//tensorflow/python:platform_test",
86        "//tensorflow/python/ops/distributions",
87        "//third_party/py/numpy",
88    ],
89)
90
91cuda_py_test(
92    name = "categorical_test",
93    size = "small",
94    srcs = ["categorical_test.py"],
95    deps = [
96        "//tensorflow/python:array_ops",
97        "//tensorflow/python:client_testlib",
98        "//tensorflow/python:framework",
99        "//tensorflow/python:framework_for_generated_wrappers",
100        "//tensorflow/python:framework_test_lib",
101        "//tensorflow/python:math_ops",
102        "//tensorflow/python:platform_test",
103        "//tensorflow/python:random_ops",
104        "//tensorflow/python/ops/distributions",
105        "//third_party/py/numpy",
106        "@absl_py//absl/testing:parameterized",
107    ],
108)
109
110cuda_py_test(
111    name = "dirichlet_test",
112    size = "small",
113    srcs = ["dirichlet_test.py"],
114    deps = [
115        "//tensorflow/python:client_testlib",
116        "//tensorflow/python:framework_for_generated_wrappers",
117        "//tensorflow/python:framework_test_lib",
118        "//tensorflow/python:platform_test",
119        "//tensorflow/python/ops/distributions",
120        "//third_party/py/numpy",
121    ],
122)
123
124cuda_py_test(
125    name = "dirichlet_multinomial_test",
126    size = "medium",
127    srcs = ["dirichlet_multinomial_test.py"],
128    tags = [
129        "noguitar",  # b/110489471
130        "notap",  # b/110489471
131    ],
132    deps = [
133        "//tensorflow/python:array_ops",
134        "//tensorflow/python:client_testlib",
135        "//tensorflow/python:framework_for_generated_wrappers",
136        "//tensorflow/python:framework_test_lib",
137        "//tensorflow/python:math_ops",
138        "//tensorflow/python:platform_test",
139        "//tensorflow/python/ops/distributions",
140        "//third_party/py/numpy",
141    ],
142)
143
144cuda_py_test(
145    name = "exponential_test",
146    srcs = ["exponential_test.py"],
147    deps = [
148        "//tensorflow/python:client",
149        "//tensorflow/python:client_testlib",
150        "//tensorflow/python:framework_for_generated_wrappers",
151        "//tensorflow/python:framework_test_lib",
152        "//tensorflow/python:nn_ops",
153        "//tensorflow/python:platform_test",
154        "//tensorflow/python/ops/distributions",
155        "//third_party/py/numpy",
156    ],
157)
158
159cuda_py_test(
160    name = "gamma_test",
161    srcs = ["gamma_test.py"],
162    deps = [
163        "//tensorflow/python:client",
164        "//tensorflow/python:client_testlib",
165        "//tensorflow/python:framework_for_generated_wrappers",
166        "//tensorflow/python:framework_test_lib",
167        "//tensorflow/python:nn_ops",
168        "//tensorflow/python:platform_test",
169        "//tensorflow/python/ops/distributions",
170        "//third_party/py/numpy",
171    ],
172)
173
174cuda_py_test(
175    name = "laplace_test",
176    srcs = ["laplace_test.py"],
177    deps = [
178        "//tensorflow/python:client",
179        "//tensorflow/python:client_testlib",
180        "//tensorflow/python:framework_for_generated_wrappers",
181        "//tensorflow/python:framework_test_lib",
182        "//tensorflow/python:nn_ops",
183        "//tensorflow/python:platform_test",
184        "//tensorflow/python/ops/distributions",
185        "//third_party/py/numpy",
186    ],
187)
188
189cuda_py_test(
190    name = "multinomial_test",
191    srcs = ["multinomial_test.py"],
192    tags = ["manual"],  # b/69001419
193    deps = [
194        "//tensorflow/python:array_ops",
195        "//tensorflow/python:client_testlib",
196        "//tensorflow/python:framework_for_generated_wrappers",
197        "//tensorflow/python:framework_test_lib",
198        "//tensorflow/python:math_ops",
199        "//tensorflow/python:platform_test",
200        "//tensorflow/python/ops/distributions",
201        "//third_party/py/numpy",
202    ],
203)
204
205cuda_py_test(
206    name = "student_t_test",
207    size = "small",
208    srcs = ["student_t_test.py"],
209    tags = [
210        # TODO(b/121223043): Re-enable this test after fixing "mean not defined"
211        # errors.
212        "no_mac",
213        "no_oss",
214        # disable to avoid false positives from scipy.
215        "nomsan",
216    ],
217    deps = [
218        "//tensorflow/python:client_testlib",
219        "//tensorflow/python:framework",
220        "//tensorflow/python:framework_for_generated_wrappers",
221        "//tensorflow/python:framework_test_lib",
222        "//tensorflow/python:math_ops",
223        "//tensorflow/python:nn_ops",
224        "//tensorflow/python:platform_test",
225        "//tensorflow/python/ops/distributions",
226        "//third_party/py/numpy",
227    ],
228)
229
230cuda_py_test(
231    name = "uniform_test",
232    size = "small",
233    srcs = ["uniform_test.py"],
234    deps = [
235        "//tensorflow/python:array_ops",
236        "//tensorflow/python:client_testlib",
237        "//tensorflow/python:errors",
238        "//tensorflow/python:framework_for_generated_wrappers",
239        "//tensorflow/python:framework_test_lib",
240        "//tensorflow/python:math_ops",
241        "//tensorflow/python/ops/distributions",
242        "//third_party/py/numpy",
243    ],
244)
245
246cuda_py_test(
247    name = "normal_test",
248    size = "medium",
249    srcs = ["normal_test.py"],
250    deps = [
251        "//tensorflow/python:array_ops",
252        "//tensorflow/python:client_testlib",
253        "//tensorflow/python:framework_for_generated_wrappers",
254        "//tensorflow/python:framework_test_lib",
255        "//tensorflow/python:gradients",
256        "//tensorflow/python:nn_ops",
257        "//tensorflow/python:platform_test",
258        "//tensorflow/python:variables",
259        "//tensorflow/python/ops/distributions",
260        "//third_party/py/numpy",
261    ],
262)
263
264cuda_py_test(
265    name = "special_math_test",
266    size = "medium",
267    srcs = ["special_math_test.py"],
268    deps = [
269        "//tensorflow/python:client_testlib",
270        "//tensorflow/python:framework_for_generated_wrappers",
271        "//tensorflow/python:framework_test_lib",
272        "//tensorflow/python:gradients",
273        "//tensorflow/python:platform_test",
274        "//tensorflow/python:variables",
275        "//tensorflow/python/ops/distributions",
276        "//third_party/py/numpy",
277    ],
278)
279
280cuda_py_test(
281    name = "identity_bijector_test",
282    size = "small",
283    srcs = ["identity_bijector_test.py"],
284    deps = [
285        "//tensorflow/python:array_ops",
286        "//tensorflow/python:client_testlib",
287        "//tensorflow/python:framework_for_generated_wrappers",
288        "//tensorflow/python:framework_test_lib",
289        "//tensorflow/python:math_ops",
290        "//tensorflow/python:platform_test",
291        "//tensorflow/python/ops/distributions",
292        "//third_party/py/numpy",
293        "@six_archive//:six",
294    ],
295)
296