• 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
5# buildifier: disable=same-origin-load
6load("//tensorflow:tensorflow.bzl", "tf_py_test")
7
8package(
9    default_visibility = ["//tensorflow:internal"],
10    licenses = ["notice"],
11)
12
13cuda_py_test(
14    name = "summary_ops_test",
15    size = "medium",
16    srcs = ["summary_ops_test.py"],
17    deps = [
18        "//tensorflow/core:protos_all_py",
19        "//tensorflow/python:client_testlib",
20        "//tensorflow/python:constant_op",
21        "//tensorflow/python:dtypes",
22        "//tensorflow/python:errors",
23        "//tensorflow/python:framework_ops",
24        "//tensorflow/python:framework_test_lib",
25        "//tensorflow/python:lib",
26        "//tensorflow/python:math_ops",
27        "//tensorflow/python:platform",
28        "//tensorflow/python:summary_ops_v2",
29        "//tensorflow/python:tensor_spec",
30        "//tensorflow/python:tensor_util",
31        "//tensorflow/python:variables",
32        "//tensorflow/python/eager:context",
33        "//tensorflow/python/eager:function",
34        "//tensorflow/python/saved_model:load",
35        "//tensorflow/python/saved_model:loader",
36        "//tensorflow/python/saved_model:tag_constants",
37    ],
38)
39
40cuda_py_test(
41    name = "summary_v1_audio_op_test",
42    size = "small",
43    srcs = ["summary_v1_audio_op_test.py"],
44    deps = [
45        "//tensorflow/core:protos_all_py",
46        "//tensorflow/python:client_testlib",
47        "//tensorflow/python:framework_for_generated_wrappers",
48        "//tensorflow/python/summary",
49        "//third_party/py/numpy",
50    ],
51)
52
53cuda_py_test(
54    name = "summary_v1_image_op_test",
55    size = "small",
56    srcs = ["summary_v1_image_op_test.py"],
57    deps = [
58        "//tensorflow/core:protos_all_py",
59        "//tensorflow/python:client_testlib",
60        "//tensorflow/python:framework_for_generated_wrappers",
61        "//tensorflow/python:image_ops",
62        "//tensorflow/python:nn_grad",
63        "//tensorflow/python/summary",
64        "//third_party/py/numpy",
65    ],
66)
67
68tf_py_test(
69    name = "summary_v1_ops_test",
70    size = "small",
71    srcs = ["summary_v1_ops_test.py"],
72    deps = [
73        "//tensorflow/core:protos_all_py",
74        "//tensorflow/python:client_testlib",
75        "//tensorflow/python:framework_for_generated_wrappers",
76        "//tensorflow/python:logging_ops",
77        "//tensorflow/python/summary",
78    ],
79)
80
81tf_py_test(
82    name = "summary_v1_tensor_op_test",
83    size = "small",
84    srcs = ["summary_v1_tensor_op_test.py"],
85    deps = [
86        "//tensorflow/core:protos_all_py",
87        "//tensorflow/python:array_ops",
88        "//tensorflow/python:client_testlib",
89        "//tensorflow/python:framework",
90        "//tensorflow/python:framework_for_generated_wrappers",
91        "//tensorflow/python/summary",
92        "//third_party/py/numpy",
93    ],
94)
95