• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# Tests of TensorFlow array ops kernels written using the Python API.
2
3load("//tensorflow:tensorflow.bzl", "cuda_py_test")
4
5package(
6    default_visibility = ["//tensorflow:internal"],
7    licenses = ["notice"],
8)
9
10cuda_py_test(
11    name = "batch_gather_op_test",
12    srcs = ["batch_gather_op_test.py"],
13    deps = [
14        "//tensorflow/python:array_ops",
15        "//tensorflow/python:client_testlib",
16        "//tensorflow/python:constant_op",
17        "//tensorflow/python:dtypes",
18        "@absl_py//absl/testing:parameterized",
19    ],
20)
21
22cuda_py_test(
23    name = "unstack_op_test",
24    size = "small",
25    srcs = ["unstack_op_test.py"],
26    tags = [
27        "no_cuda_asan",  # b/173806679
28    ],
29    xla_tags = [
30        "no_cuda_asan",  # times out
31    ],
32    deps = [
33        "//tensorflow/python:array_ops",
34        "//tensorflow/python:client_testlib",
35        "//tensorflow/python:errors",
36        "//tensorflow/python:framework_for_generated_wrappers",
37        "//tensorflow/python:gradients",
38        "//third_party/py/numpy",
39    ],
40)
41
42cuda_py_test(
43    name = "slice_op_test",
44    size = "medium",
45    srcs = ["slice_op_test.py"],
46    deps = [
47        "//tensorflow/python:array_ops",
48        "//tensorflow/python:client_testlib",
49        "//tensorflow/python:errors",
50        "//tensorflow/python:framework_for_generated_wrappers",
51        "//tensorflow/python:gradients",
52        "//third_party/py/numpy",
53    ],
54)
55
56cuda_py_test(
57    name = "gather_op_test",
58    size = "medium",
59    srcs = ["gather_op_test.py"],
60    deps = [
61        "//tensorflow/python:array_ops",
62        "//tensorflow/python:client_testlib",
63        "//tensorflow/python:framework_for_generated_wrappers",
64        "//tensorflow/python:gradients",
65        "//third_party/py/numpy",
66        "@absl_py//absl/testing:parameterized",
67    ],
68)
69
70cuda_py_test(
71    name = "scatter_nd_ops_test",
72    size = "medium",
73    srcs = ["scatter_nd_ops_test.py"],
74    tags = ["noasan"],  # http://b/32635055
75    deps = [
76        "//tensorflow/python:array_ops",
77        "//tensorflow/python:client",
78        "//tensorflow/python:client_testlib",
79        "//tensorflow/python:framework_for_generated_wrappers",
80        "//tensorflow/python:gradients",
81        "//tensorflow/python:math_ops",
82        "//tensorflow/python:resource_variable_ops",
83        "//tensorflow/python:state_ops",
84        "//tensorflow/python:variables",
85        "//third_party/py/numpy",
86    ],
87)
88
89cuda_py_test(
90    name = "stack_op_test",
91    size = "medium",
92    srcs = ["stack_op_test.py"],
93    deps = [
94        "//tensorflow/python:array_ops",
95        "//tensorflow/python:client_testlib",
96        "//tensorflow/python:errors",
97        "//tensorflow/python:framework_for_generated_wrappers",
98        "//tensorflow/python:variables",
99        "//third_party/py/numpy",
100    ],
101)
102