• 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"],  # Apache 2.0
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    tags = ["no_cuda_asan"],  # b/173806733
61    deps = [
62        "//tensorflow/python:array_ops",
63        "//tensorflow/python:client_testlib",
64        "//tensorflow/python:framework_for_generated_wrappers",
65        "//tensorflow/python:gradients",
66        "//third_party/py/numpy",
67        "@absl_py//absl/testing:parameterized",
68    ],
69)
70
71cuda_py_test(
72    name = "scatter_nd_ops_test",
73    size = "medium",
74    srcs = ["scatter_nd_ops_test.py"],
75    tags = ["noasan"],  # http://b/32635055
76    deps = [
77        "//tensorflow/python:array_ops",
78        "//tensorflow/python:client",
79        "//tensorflow/python:client_testlib",
80        "//tensorflow/python:framework_for_generated_wrappers",
81        "//tensorflow/python:gradients",
82        "//tensorflow/python:math_ops",
83        "//tensorflow/python:resource_variable_ops",
84        "//tensorflow/python:state_ops",
85        "//tensorflow/python:variables",
86        "//third_party/py/numpy",
87    ],
88)
89
90cuda_py_test(
91    name = "stack_op_test",
92    size = "medium",
93    srcs = ["stack_op_test.py"],
94    deps = [
95        "//tensorflow/python:array_ops",
96        "//tensorflow/python:client_testlib",
97        "//tensorflow/python:errors",
98        "//tensorflow/python:framework_for_generated_wrappers",
99        "//tensorflow/python:variables",
100        "//third_party/py/numpy",
101    ],
102)
103