• 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 = "framework",
10    srcs = ["framework.py"],
11    srcs_version = "PY3",
12    deps = [
13        "//tensorflow/core:protos_all_py",
14        "//tensorflow/python:client",
15        "//tensorflow/python:errors",
16        "//tensorflow/python:framework_ops",
17        "//tensorflow/python:training",
18        "//tensorflow/python/debug/lib:debug_utils",
19    ],
20)
21
22py_library(
23    name = "dumping_wrapper",
24    srcs = ["dumping_wrapper.py"],
25    srcs_version = "PY3",
26    visibility = ["//tensorflow:internal"],
27    deps = [
28        ":framework",
29        "//tensorflow/core:protos_all_py",
30        "//tensorflow/python:platform",
31        "//tensorflow/python/debug/lib:debug_data",
32    ],
33)
34
35py_library(
36    name = "grpc_wrapper",
37    srcs = ["grpc_wrapper.py"],
38    srcs_version = "PY3",
39    deps = [
40        ":framework",
41        "//tensorflow/python/debug/lib:common",
42        "//tensorflow/python/debug/lib:source_remote",
43    ],
44)
45
46py_library(
47    name = "local_cli_wrapper",
48    srcs = ["local_cli_wrapper.py"],
49    srcs_version = "PY3",
50    deps = [
51        ":framework",
52        "//tensorflow/python/debug/cli:analyzer_cli",
53        "//tensorflow/python/debug/cli:cli_shared",
54        "//tensorflow/python/debug/cli:command_parser",
55        "//tensorflow/python/debug/cli:debugger_cli_common",
56        "//tensorflow/python/debug/cli:profile_analyzer_cli",
57        "//tensorflow/python/debug/cli:tensor_format",
58        "//tensorflow/python/debug/cli:ui_factory",
59        "//tensorflow/python/debug/lib:common",
60        "//tensorflow/python/debug/lib:debug_data",
61    ],
62)
63
64py_library(
65    name = "hooks",
66    srcs = ["hooks.py"],
67    srcs_version = "PY3",
68    visibility = [
69        "//tensorflow:internal",
70        "//third_party/py/tf_slim:__subpackages__",
71    ],
72    deps = [
73        ":dumping_wrapper",
74        ":framework",
75        ":grpc_wrapper",
76        ":local_cli_wrapper",
77        "//tensorflow/core:protos_all_py",
78        "//tensorflow/python:training",
79        "//tensorflow/python/debug/lib:debug_utils",
80    ],
81)
82
83py_test(
84    name = "framework_test",
85    size = "medium",
86    srcs = ["framework_test.py"],
87    python_version = "PY3",
88    srcs_version = "PY3",
89    tags = [
90        "no_rocm",
91    ],
92    deps = [
93        ":framework",
94        "//tensorflow/core:protos_all_py",
95        "//tensorflow/python:array_ops",
96        "//tensorflow/python:client",
97        "//tensorflow/python:errors",
98        "//tensorflow/python:framework_for_generated_wrappers",
99        "//tensorflow/python:framework_test_lib",
100        "//tensorflow/python:math_ops",
101        "//tensorflow/python:platform_test",
102        "//tensorflow/python:resource_variable_ops",
103        "//tensorflow/python:training",
104        "//tensorflow/python:util",
105        "//tensorflow/python:variables",
106        "//tensorflow/python/debug/lib:debug_data",
107        "//third_party/py/numpy",
108    ],
109)
110
111py_test(
112    name = "dumping_wrapper_test",
113    size = "small",
114    srcs = ["dumping_wrapper_test.py"],
115    python_version = "PY3",
116    srcs_version = "PY3",
117    deps = [
118        ":dumping_wrapper",
119        ":framework",
120        ":hooks",
121        "//tensorflow/python:array_ops",
122        "//tensorflow/python:client",
123        "//tensorflow/python:framework_for_generated_wrappers",
124        "//tensorflow/python:framework_test_lib",
125        "//tensorflow/python:platform",
126        "//tensorflow/python:platform_test",
127        "//tensorflow/python:state_ops",
128        "//tensorflow/python:training",
129        "//tensorflow/python:variables",
130        "//tensorflow/python/debug/lib:debug_data",
131    ],
132)
133
134py_test(
135    name = "local_cli_wrapper_test",
136    size = "small",
137    srcs = ["local_cli_wrapper_test.py"],
138    python_version = "PY3",
139    srcs_version = "PY3",
140    deps = [
141        ":local_cli_wrapper",
142        "//tensorflow/core:protos_all_py",
143        "//tensorflow/python:array_ops",
144        "//tensorflow/python:client",
145        "//tensorflow/python:control_flow_ops",
146        "//tensorflow/python:errors",
147        "//tensorflow/python:framework_for_generated_wrappers",
148        "//tensorflow/python:framework_test_lib",
149        "//tensorflow/python:math_ops",
150        "//tensorflow/python:platform_test",
151        "//tensorflow/python:resource_variable_ops",
152        "//tensorflow/python:state_ops",
153        "//tensorflow/python:training",
154        "//tensorflow/python:variables",
155        "//tensorflow/python/debug/cli:cli_shared",
156        "//tensorflow/python/debug/cli:debugger_cli_common",
157        "//tensorflow/python/debug/cli:ui_factory",
158        "//third_party/py/numpy",
159    ],
160)
161
162py_test(
163    name = "disk_usage_test",
164    size = "small",
165    srcs = ["disk_usage_test.py"],
166    python_version = "PY3",
167    srcs_version = "PY3",
168    deps = [
169        ":dumping_wrapper",
170        ":hooks",
171        "//tensorflow/python:client",
172        "//tensorflow/python:errors",
173        "//tensorflow/python:framework_test_lib",
174        "//tensorflow/python:platform_test",
175        "//tensorflow/python:training",
176        "//tensorflow/python:variables",
177    ],
178)
179