Google Git
Sign in
android / platform / external / tensorflow / d7b71bb36cb / . / tensorflow / python / debug / BUILD
blob: 613f1f8d93c65a91d02f00c2d02801ae25c6fe8b [file] [log] [blame]
# Description:
# Python Client Code of the TensorFlow Debugger (tfdbg).
#
# Public target(s):
#
# ":debug_py": Public Python methods and classes of tfdbg.
# For API documentation, see https://www.tensorflow.org/api_docs/python/tfdbg
# For a user interface walkthrough, see https://www.tensorflow.org/guide/debugger
# ":grpc_debug_server": Server interface for grpc:// debug URLs.
load("//tensorflow:tensorflow.bzl", "cuda_py_test")
load("//tensorflow:tensorflow.bzl", "if_not_v2", "if_not_windows", "py_binary", "py_test")
package(
default_visibility = ["//tensorflow:internal"],
licenses = ["notice"],
)
py_library(
name = "debug_py",
srcs = ["__init__.py"],
srcs_version = "PY3",
visibility = ["//visibility:public"],
deps = [
":check_numerics_callback",
":debug_data",
":debug_events_monitors",
":debug_events_reader",
":debug_events_writer",
":debug_gradients",
":debug_graphs",
":debug_utils",
":dumping_callback",
":dumping_callback_test_lib",
":grpc_debug_server",
":grpc_debug_test_server",
":hooks",
":local_cli_wrapper",
"//tensorflow/python:debug_ops_gen",
"//tensorflow/python:util",
],
)
# Transitive dependencies of this target will be included in the pip package.
py_library(
name = "debug_pip",
data = [":grpc_tensorflow_server"],
srcs_version = "PY3",
deps = [
":cli_test_utils",
":debug_py",
":grpc_debug_test_server",
":grpc_tensorflow_server_lib",
":offline_analyzer_lib",
":session_debug_testlib",
":source_remote",
] + if_not_windows([
":debug_examples_v1",
":debug_examples_v2",
]),
)
py_library(
name = "op_callbacks_common",
srcs = ["lib/op_callbacks_common.py"],
srcs_version = "PY3",
)
py_library(
name = "check_numerics_callback",
srcs = ["lib/check_numerics_callback.py"],
srcs_version = "PY3",
deps = [
":op_callbacks_common",
":source_utils",
"//tensorflow/python:array_ops",
"//tensorflow/python:debug_ops_gen",
"//tensorflow/python:op_callbacks",
"//third_party/py/numpy",
],
)
py_library(
name = "dumping_callback",
srcs = ["lib/dumping_callback.py"],
srcs_version = "PY3",
deps = [
":debug_events_writer",
":op_callbacks_common",
":source_utils",
"//tensorflow/python:array_ops",
"//tensorflow/python:debug_ops_gen",
"//tensorflow/python:op_callbacks",
"//third_party/py/numpy",
],
)
py_library(
name = "dumping_callback_test_lib",
srcs = ["lib/dumping_callback_test_lib.py"],
srcs_version = "PY3",
deps = [
":check_numerics_callback",
":debug_events_reader",
":dumping_callback",
"//third_party/py/numpy",
],
)
py_library(
name = "common",
srcs = ["lib/common.py"],
srcs_version = "PY3",
)
py_library(
name = "debug_events_reader",
srcs = ["lib/debug_events_reader.py"],
srcs_version = "PY3",
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:framework",
"//tensorflow/python:lib",
"@six_archive//:six",
],
)
py_library(
name = "debug_events_monitors",
srcs = ["lib/debug_events_monitors.py"],
srcs_version = "PY3",
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:framework",
"//tensorflow/python:lib",
"//third_party/py/numpy",
],
)
py_library(
name = "debug_events_writer",
srcs = ["lib/debug_events_writer.py"],
srcs_version = "PY3",
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:framework",
],
)
py_library(
name = "debug_graphs",
srcs = ["lib/debug_graphs.py"],
srcs_version = "PY3",
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:framework",
"//tensorflow/python:op_def_registry",
"//tensorflow/python:platform",
"//tensorflow/python:tensor_util",
"@six_archive//:six",
],
)
py_library(
name = "debug_data",
srcs = ["lib/debug_data.py"],
srcs_version = "PY3",
visibility = [
"//tensorflow:internal",
"//third_party/py/tf_slim:__subpackages__",
],
deps = [
":debug_graphs",
"//tensorflow/core:protos_all_py",
"//tensorflow/python:framework",
"//tensorflow/python:op_def_registry",
"//tensorflow/python:platform",
"//tensorflow/python:tensor_util",
"//third_party/py/numpy",
"@six_archive//:six",
],
)
py_library(
name = "debug_gradients",
srcs = ["lib/debug_gradients.py"],
srcs_version = "PY3",
deps = [
":debug_data",
":debug_graphs",
"//tensorflow/python:array_ops",
"//tensorflow/python:framework",
"//tensorflow/python:platform",
"@six_archive//:six",
],
)
py_library(
name = "debug_utils",
srcs = ["lib/debug_utils.py"],
srcs_version = "PY3",
)
py_binary(
name = "grpc_tensorflow_server",
srcs = ["lib/grpc_tensorflow_server.py"],
python_version = "PY3",
srcs_version = "PY3",
deps = [":grpc_tensorflow_server_lib"],
)
py_library(
name = "grpc_tensorflow_server_lib",
srcs = [
"lib/grpc_tensorflow_server.py",
],
srcs_version = "PY3",
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:platform",
"//tensorflow/python:training",
],
)
py_library(
name = "source_utils",
srcs = ["lib/source_utils.py"],
srcs_version = "PY3",
deps = [
":profiling",
"//third_party/py/numpy",
],
)
py_library(
name = "source_remote",
srcs = ["lib/source_remote.py"],
srcs_version = "PY3",
deps = [
":common",
":debug_service_pb2_grpc",
"//tensorflow/core/debug:debug_service_proto_py",
"//tensorflow/python/profiler:tfprof_logger",
],
)
py_library(
name = "framework",
srcs = ["wrappers/framework.py"],
srcs_version = "PY3",
deps = [
":debug_utils",
"//tensorflow/core:protos_all_py",
"//tensorflow/python:client",
"//tensorflow/python:errors",
"//tensorflow/python:framework_ops",
"//tensorflow/python:training",
],
)
py_library(
name = "debugger_cli_common",
srcs = ["cli/debugger_cli_common.py"],
srcs_version = "PY3",
deps = [
"//tensorflow/python:platform",
"//tensorflow/python:pywrap_tf_session",
"//third_party/py/numpy",
"@six_archive//:six",
],
)
py_library(
name = "cli_config",
srcs = ["cli/cli_config.py"],
srcs_version = "PY3",
deps = [":debugger_cli_common"],
)
py_library(
name = "command_parser",
srcs = ["cli/command_parser.py"],
srcs_version = "PY3",
)
py_library(
name = "tensor_format",
srcs = ["cli/tensor_format.py"],
srcs_version = "PY3",
deps = [
":debug_data",
":debugger_cli_common",
"//third_party/py/numpy",
],
)
py_library(
name = "cli_shared",
srcs = ["cli/cli_shared.py"],
srcs_version = "PY3",
deps = [
":command_parser",
":common",
":debugger_cli_common",
":tensor_format",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:platform",
"//tensorflow/python:variables",
"//third_party/py/numpy",
"@six_archive//:six",
],
)
py_library(
name = "evaluator",
srcs = ["cli/evaluator.py"],
srcs_version = "PY3",
deps = [
":debug_data",
"//third_party/py/numpy",
],
)
py_library(
name = "analyzer_cli",
srcs = ["cli/analyzer_cli.py"],
srcs_version = "PY3",
deps = [
":cli_config",
":cli_shared",
":command_parser",
":debug_graphs",
":debugger_cli_common",
":evaluator",
":source_utils",
":ui_factory",
"@six_archive//:six",
],
)
py_library(
name = "profiling",
srcs = ["lib/profiling.py"],
srcs_version = "PY3",
)
py_library(
name = "profile_analyzer_cli",
srcs = ["cli/profile_analyzer_cli.py"],
srcs_version = "PY3",
deps = [
":cli_shared",
":command_parser",
":debugger_cli_common",
":profiling",
":source_utils",
":ui_factory",
"//third_party/py/numpy",
],
)
py_library(
name = "base_ui",
srcs = ["cli/base_ui.py"],
srcs_version = "PY3",
deps = [
":cli_config",
":command_parser",
":debugger_cli_common",
],
)
py_library(
name = "curses_widgets",
srcs = ["cli/curses_widgets.py"],
srcs_version = "PY3",
deps = [":debugger_cli_common"],
)
py_library(
name = "curses_ui",
srcs = ["cli/curses_ui.py"],
srcs_version = "PY3",
deps = [
":base_ui",
":cli_shared",
":command_parser",
":curses_widgets",
":debugger_cli_common",
":tensor_format",
"@six_archive//:six",
],
)
py_library(
name = "readline_ui",
srcs = ["cli/readline_ui.py"],
srcs_version = "PY3",
deps = [
":base_ui",
":debugger_cli_common",
],
)
py_library(
name = "ui_factory",
srcs = ["cli/ui_factory.py"],
srcs_version = "PY3",
deps = [
":curses_ui",
":readline_ui",
],
)
py_library(
name = "dumping_wrapper",
srcs = ["wrappers/dumping_wrapper.py"],
srcs_version = "PY3",
visibility = [
"//tensorflow:internal",
"//third_party/py/tf_slim:__subpackages__",
],
deps = [
":debug_data",
":framework",
"//tensorflow/core:protos_all_py",
"//tensorflow/python:platform",
],
)
py_library(
name = "grpc_wrapper",
srcs = ["wrappers/grpc_wrapper.py"],
srcs_version = "PY3",
deps = [
":common",
":framework",
":source_remote",
],
)
py_library(
name = "local_cli_wrapper",
srcs = ["wrappers/local_cli_wrapper.py"],
srcs_version = "PY3",
deps = [
":analyzer_cli",
":cli_shared",
":command_parser",
":common",
":debug_data",
":debugger_cli_common",
":framework",
":profile_analyzer_cli",
":tensor_format",
":ui_factory",
],
)
py_library(
name = "hooks",
srcs = ["wrappers/hooks.py"],
srcs_version = "PY3",
visibility = [
"//tensorflow:internal",
"//third_party/py/tf_slim:__subpackages__",
],
deps = [
":debug_utils",
":dumping_wrapper",
":framework",
":grpc_wrapper",
":local_cli_wrapper",
"//tensorflow/core:protos_all_py",
"//tensorflow/python:training",
],
)
py_binary(
name = "offline_analyzer",
srcs = ["cli/offline_analyzer.py"],
python_version = "PY3",
srcs_version = "PY3",
deps = [":offline_analyzer_lib"],
)
py_library(
name = "offline_analyzer_lib",
srcs = ["cli/offline_analyzer.py"],
srcs_version = "PY3",
deps = [
":analyzer_cli",
":debug_data",
"//tensorflow/python", # TODO(b/34059704): remove when fixed
"//tensorflow/python:platform",
],
)
py_library(
name = "debug_examples_v1",
srcs_version = "PY3",
deps = [
":debug_errors_lib",
":debug_fibonacci_lib",
":debug_keras_lib",
] + if_not_v2([
":debug_mnist_lib",
":debug_tflearn_iris_lib",
]),
)
py_library(
name = "debug_examples_v2",
srcs_version = "PY3",
deps = [
":debug_fibonacci_lib",
":debug_mnist_lib",
],
)
py_binary(
name = "debug_fibonacci",
srcs = ["examples/v1/debug_fibonacci.py"],
python_version = "PY3",
srcs_version = "PY3",
deps = [":debug_fibonacci_lib"],
)
py_binary(
name = "debug_fibonacci_v2",
srcs = ["examples/v2/debug_fibonacci_v2.py"],
python_version = "PY3",
srcs_version = "PY3",
deps = [":debug_fibonacci_lib"],
)
py_library(
name = "debug_fibonacci_lib",
srcs = [
"examples/v1/debug_fibonacci.py",
"examples/v2/debug_fibonacci_v2.py",
],
srcs_version = "PY3",
deps = [
":debug_py",
"//tensorflow:tensorflow_py",
"//third_party/py/numpy",
"@six_archive//:six",
],
)
py_binary(
name = "debug_errors",
srcs = ["examples/v1/debug_errors.py"],
python_version = "PY3",
srcs_version = "PY3",
deps = [":debug_errors_lib"],
)
py_library(
name = "debug_errors_lib",
srcs = ["examples/v1/debug_errors.py"],
srcs_version = "PY3",
deps = [
":debug_py",
"//tensorflow:tensorflow_py",
"//third_party/py/numpy",
],
)
py_binary(
name = "debug_mnist",
srcs = ["examples/debug_mnist.py"],
python_version = "PY3",
srcs_version = "PY3",
deps = [":debug_mnist_lib"],
)
py_binary(
name = "debug_mnist_v1",
srcs = ["examples/v1/debug_mnist_v1.py"],
python_version = "PY3",
srcs_version = "PY3",
deps = [":debug_mnist_lib"],
)
py_binary(
name = "debug_mnist_v2",
srcs = ["examples/v2/debug_mnist_v2.py"],
python_version = "PY3",
srcs_version = "PY3",
deps = [":debug_mnist_lib"],
)
py_library(
name = "debug_mnist_lib",
srcs = [
"examples/debug_mnist.py",
"examples/v1/debug_mnist_v1.py",
"examples/v2/debug_mnist_v2.py",
],
srcs_version = "PY3",
deps = [
":debug_py",
"//tensorflow:tensorflow_py",
],
)
py_binary(
name = "debug_tflearn_iris",
srcs = ["examples/v1/debug_tflearn_iris.py"],
python_version = "PY3",
srcs_version = "PY3",
deps = [":debug_tflearn_iris_lib"],
)
py_library(
name = "debug_tflearn_iris_lib",
srcs = ["examples/v1/debug_tflearn_iris.py"],
srcs_version = "PY3",
deps = [
":debug_py",
"//tensorflow:tensorflow_py",
"@six_archive//:six",
],
)
py_binary(
name = "debug_keras",
srcs = ["examples/v1/debug_keras.py"],
python_version = "PY3",
srcs_version = "PY3",
deps = [":debug_keras_lib"],
)
py_library(
name = "debug_keras_lib",
srcs = ["examples/v1/debug_keras.py"],
srcs_version = "PY3",
deps = [
":debug_py",
"//tensorflow:tensorflow_py",
"//third_party/py/numpy",
],
)
py_test(
name = "common_test",
size = "small",
srcs = ["lib/common_test.py"],
python_version = "PY3",
srcs_version = "PY3",
deps = [
":common",
"//tensorflow/python:client",
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
"//tensorflow/python:platform_test",
],
)
py_test(
name = "debug_events_monitors_test",
size = "medium",
srcs = ["lib/debug_events_monitors_test.py"],
python_version = "PY3",
srcs_version = "PY3",
tags = [
"no_windows", # b/142475891
],
deps = [
":debug_events_monitors",
":debug_events_reader",
":debug_events_writer",
":dumping_callback",
":dumping_callback_test_lib",
"//tensorflow/core:protos_all_py",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_test_lib",
"//third_party/py/numpy",
],
)
py_test(
name = "debug_events_writer_test",
size = "medium",
srcs = ["lib/debug_events_writer_test.py"],
python_version = "PY3",
srcs_version = "PY3",
tags = [
"no_windows", # b/142475891
],
deps = [
":debug_events_reader",
":debug_events_writer",
":dumping_callback_test_lib",
"//tensorflow/core:protos_all_py",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_test_lib",
],
)
py_test(
name = "debug_graphs_test",
size = "small",
srcs = ["lib/debug_graphs_test.py"],
python_version = "PY3",
srcs_version = "PY3",
tags = ["no_windows"], # TODO(b/184424727): Enable this test on Windows.
deps = [
":debug_graphs",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_test_lib",
],
)
py_test(
name = "debug_data_test",
size = "small",
srcs = ["lib/debug_data_test.py"],
python_version = "PY3",
srcs_version = "PY3",
tags = ["no_windows"], # TODO(b/184424727): Enable this test on Windows.
deps = [
":debug_data",
"//tensorflow/core:protos_all_py",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:platform",
"//tensorflow/python:platform_test",
"//third_party/py/numpy",
],
)
cuda_py_test(
name = "check_numerics_callback_test",
size = "medium",
srcs = ["lib/check_numerics_callback_test.py"],
python_version = "PY3",
tags = [
"no_mac", # TODO(b/175322370): Detected Infinity or NaN in output 0 of graph op "RealDiv"
"no_windows",
],
deps = [
":check_numerics_callback",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:variables",
"//third_party/py/numpy",
],
)
cuda_py_test(
name = "dumping_callback_test",
size = "medium",
srcs = ["lib/dumping_callback_test.py"],
python_version = "PY3",
shard_count = 4,
tags = [
"no_windows", # TODO(b/142475891): Enable this test on Windows.
],
xla_enable_strict_auto_jit = False, # Node names are different with autojit
deps = [
":debug_events_reader",
":debug_events_writer",
":dumping_callback",
":dumping_callback_test_lib",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:variables",
"//third_party/py/numpy",
],
)
cuda_py_test(
name = "debug_v2_ops_test",
size = "medium",
srcs = ["lib/debug_v2_ops_test.py"],
python_version = "PY3",
tags = ["no_windows_gpu"],
deps = [
":debug_events_reader",
":debug_events_writer",
":dumping_callback_test_lib",
"//tensorflow/python:debug_ops_gen",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:gradients",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:training",
"//tensorflow/python:variables",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
],
)
cuda_py_test(
name = "debug_gradients_test",
size = "small",
srcs = ["lib/debug_gradients_test.py"],
python_version = "PY3",
xla_enable_strict_auto_jit = False, # Node names are different with autojit
deps = [
":debug_data",
":debug_gradients",
":debug_utils",
"//tensorflow/python:client",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:gradients",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:training",
"//tensorflow/python:variables",
],
)
py_test(
name = "debug_utils_test",
size = "small",
srcs = ["lib/debug_utils_test.py"],
python_version = "PY3",
srcs_version = "PY3",
deps = [
":debug_utils",
"//tensorflow/core:protos_all_py",
"//tensorflow/python:client",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:resource_variable_ops",
"//tensorflow/python:variables",
"//third_party/py/numpy",
],
)
py_test(
name = "source_utils_test",
size = "small",
srcs = ["lib/source_utils_test.py"],
python_version = "PY3",
srcs_version = "PY3",
tags = [
"no_windows",
],
deps = [
":debug_data",
":debug_utils",
":source_utils",
"//tensorflow/core:protos_all_py",
"//tensorflow/python:client",
"//tensorflow/python:cond_v2",
"//tensorflow/python:constant_op",
"//tensorflow/python:control_flow_ops",
"//tensorflow/python:framework_ops",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:resource_variable_ops",
"//tensorflow/python:util",
"//tensorflow/python:variables",
"//tensorflow/python:while_v2",
"//third_party/py/numpy",
],
)
py_test(
name = "source_remote_test",
size = "small",
srcs = ["lib/source_remote_test.py"],
python_version = "PY3",
srcs_version = "PY3",
tags = [
"no_windows",
"oss_serial",
],
deps = [
":grpc_debug_test_server",
":source_remote",
":source_utils",
"//tensorflow/core:protos_all_py",
"//tensorflow/python:client",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_ops",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:resource_variable_ops",
"//tensorflow/python:util",
"//tensorflow/python:variables",
],
)
py_test(
name = "framework_test",
size = "medium",
srcs = ["wrappers/framework_test.py"],
python_version = "PY3",
srcs_version = "PY3",
tags = ["no_rocm"],
deps = [
":debug_data",
":framework",
"//tensorflow/core:protos_all_py",
"//tensorflow/python:array_ops",
"//tensorflow/python:client",
"//tensorflow/python:errors",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:resource_variable_ops",
"//tensorflow/python:training",
"//tensorflow/python:util",
"//tensorflow/python:variables",
"//third_party/py/numpy",
],
)
py_test(
name = "profiling_test",
size = "small",
srcs = ["lib/profiling_test.py"],
python_version = "PY3",
srcs_version = "PY3",
deps = [
":profiling",
"//tensorflow/core:protos_all_py",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:platform_test",
],
)
py_test(
name = "curses_ui_test",
size = "small",
srcs = ["cli/curses_ui_test.py"],
python_version = "PY3",
srcs_version = "PY3",
tags = [
"no_windows",
],
deps = [
":cli_test_utils",
":curses_ui",
":debugger_cli_common",
":tensor_format",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:platform",
"//tensorflow/python:platform_test",
"//third_party/py/numpy",
],
)
py_test(
name = "readline_ui_test",
size = "small",
srcs = ["cli/readline_ui_test.py"],
python_version = "PY3",
srcs_version = "PY3",
deps = [
":cli_config",
":debugger_cli_common",
":readline_ui",
":ui_factory",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:platform",
"//tensorflow/python:platform_test",
],
)
py_library(
name = "session_debug_testlib",
srcs = ["lib/session_debug_testlib.py"],
srcs_version = "PY3",
deps = [
":debug_data",
":debug_graphs",
":debug_utils",
"//tensorflow/core:protos_all_py",
"//tensorflow/python:array_ops",
"//tensorflow/python:client",
"//tensorflow/python:client_testlib",
"//tensorflow/python:control_flow_ops",
"//tensorflow/python:data_flow_ops",
"//tensorflow/python:errors",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python:parsing_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:rnn",
"//tensorflow/python:rnn_cell",
"//tensorflow/python:state_ops",
"//tensorflow/python:tensor_array_grad",
"//tensorflow/python:training",
"//tensorflow/python:variables",
"//third_party/py/numpy",
"@six_archive//:six",
],
)
py_library(
name = "debug_service_pb2_grpc",
srcs = ["lib/debug_service_pb2_grpc.py"],
srcs_version = "PY3",
deps = [
"//tensorflow/core/debug:debug_service_proto_py",
],
)
py_library(
name = "grpc_debug_server",
srcs = ["lib/grpc_debug_server.py"],
srcs_version = "PY3",
visibility = ["//visibility:public"],
deps = [
":debug_graphs",
":debug_service_pb2_grpc",
"//tensorflow/core/debug:debug_service_proto_py",
"@six_archive//:six",
],
)
py_library(
name = "grpc_debug_test_server",
srcs = ["lib/grpc_debug_test_server.py"],
srcs_version = "PY3",
deps = [
":debug_data",
":debug_utils",
":grpc_debug_server",
"//tensorflow/core:protos_all_py",
"//tensorflow/python:client",
"//tensorflow/python:constant_op",
"//tensorflow/python:errors",
"//tensorflow/python:variables",
],
)
cuda_py_test(
name = "debug_grappler_test",
size = "small",
srcs = ["lib/debug_grappler_test.py"],
python_version = "PY3",
xla_enable_strict_auto_jit = False, # Tests TF:Classic implementation.
deps = [
":debug_data",
":debug_utils",
"//tensorflow/python:client",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:variables",
],
)
cuda_py_test(
name = "session_debug_file_test",
size = "small",
srcs = ["lib/session_debug_file_test.py"],
python_version = "PY3",
tags = ["notsan"],
xla_enable_strict_auto_jit = False, # Node names are different with autojit
deps = [
":debug_data",
":debug_utils",
":session_debug_testlib",
"//tensorflow/python:client",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:variables",
],
)
cuda_py_test(
name = "debug_graph_reconstruction_test",
size = "small",
srcs = ["lib/debug_graph_reconstruction_test.py"],
python_version = "PY3",
xla_enable_strict_auto_jit = False, # Node names are different with autojit
deps = [
":debug_data",
":debug_utils",
"//tensorflow/python:client",
"//tensorflow/python:client_testlib",
"//tensorflow/python:control_flow_ops",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python:training",
"//tensorflow/python:variables",
],
)
cuda_py_test(
name = "session_debug_multi_gpu_test",
size = "small",
srcs = ["lib/session_debug_multi_gpu_test.py"],
python_version = "PY3",
tags = ["no_windows_gpu"],
xla_enable_strict_auto_jit = False, # Node names are different with autojit
deps = [
":debug_data",
":debug_utils",
"//tensorflow/python:client",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:variables",
],
)
py_test(
name = "debugger_cli_common_test",
size = "small",
srcs = ["cli/debugger_cli_common_test.py"],
python_version = "PY3",
srcs_version = "PY3",
deps = [
":debugger_cli_common",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:platform",
"//tensorflow/python:platform_test",
"//tensorflow/python:pywrap_tf_session",
"//third_party/py/numpy",
],
)
py_test(
name = "cli_config_test",
size = "small",
srcs = ["cli/cli_config_test.py"],
python_version = "PY3",
srcs_version = "PY3",
deps = [
":cli_config",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:platform",
"//tensorflow/python:platform_test",
],
)
py_test(
name = "command_parser_test",
size = "small",
srcs = ["cli/command_parser_test.py"],
python_version = "PY3",
srcs_version = "PY3",
deps = [
":command_parser",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:platform_test",
],
)
py_test(
name = "tensor_format_test",
size = "small",
srcs = ["cli/tensor_format_test.py"],
python_version = "PY3",
srcs_version = "PY3",
deps = [
":cli_test_utils",
":debug_data",
":tensor_format",
"//tensorflow/core:protos_all_py",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:platform_test",
"//third_party/py/numpy",
"@six_archive//:six",
],
)
py_test(
name = "cli_shared_test",
size = "small",
srcs = ["cli/cli_shared_test.py"],
python_version = "PY3",
srcs_version = "PY3",
deps = [
":cli_shared",
":debugger_cli_common",
"//tensorflow/python:errors",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:platform_test",
"//tensorflow/python:sparse_tensor",
"//tensorflow/python:variables",
],
)
py_test(
name = "evaluator_test",
size = "small",
srcs = [
"cli/evaluator_test.py",
],
python_version = "PY3",
srcs_version = "PY3",
tags = ["no_windows"], # TODO(b/184424727): Enable this test on Windows.
deps = [
":debug_data",
":evaluator",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_test_lib",
"//third_party/py/numpy",
],
)
py_library(
name = "cli_test_utils",
srcs = ["cli/cli_test_utils.py"],
srcs_version = "PY3",
)
cuda_py_test(
name = "analyzer_cli_test",
size = "small",
srcs = ["cli/analyzer_cli_test.py"],
python_version = "PY3",
tags = ["no_windows"], # TODO: needs investigation on Windows
xla_enable_strict_auto_jit = False, # Node names are different with autojit
deps = [
":analyzer_cli",
":cli_config",
":cli_test_utils",
":command_parser",
":debug_data",
":debug_utils",
":debugger_cli_common",
":source_utils",
"//tensorflow:tensorflow_py",
"//tensorflow/python:array_ops",
"//tensorflow/python:client",
"//tensorflow/python:client_testlib",
"//tensorflow/python:control_flow_ops",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:util",
"//tensorflow/python:variables",
"//third_party/py/numpy",
"@six_archive//:six",
],
)
py_test(
name = "profile_analyzer_cli_test",
size = "small",
srcs = ["cli/profile_analyzer_cli_test.py"],
python_version = "PY3",
srcs_version = "PY3",
deps = [
":debugger_cli_common",
":profile_analyzer_cli",
"//tensorflow/core:protos_all_py",
"//tensorflow/python:client_testlib",
"//tensorflow/python:control_flow_ops",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:session",
"//tensorflow/python:util",
],
)
cuda_py_test(
name = "session_debug_grpc_test",
size = "medium",
srcs = ["lib/session_debug_grpc_test.py"],
python_version = "PY3",
tags = [
"no_oss", # Test flaky due to port collisions.
"no_windows",
"notap", # TODO(b/160925973): Test flaky due to port collisions.
"notsan",
"oss_serial",
],
xla_enable_strict_auto_jit = False, # Node names are different with autojit
deps = [
":debug_data",
":debug_utils",
":dumping_wrapper",
":grpc_debug_test_server",
":grpc_wrapper",
":hooks",
":session_debug_testlib",
"//tensorflow/python:client",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:variables",
],
)
cuda_py_test(
name = "grpc_large_data_test",
size = "medium",
srcs = ["lib/grpc_large_data_test.py"],
python_version = "PY3",
tags = [
"no_oss", # Test flaky due to port collisions.
"no_windows",
"noasan", # Times out due to size of test (b/73731462).
"notap", # TODO(b/153898535) flaky
"optonly", # Test flaky (b/80130873)
"oss_serial",
],
xla_enable_strict_auto_jit = False, # Node names are different with autojit
deps = [
":dumping_wrapper",
":grpc_debug_test_server",
":grpc_wrapper",
":session_debug_testlib",
"//tensorflow/python:client",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:platform_test",
"//tensorflow/python:variables",
"//third_party/py/numpy",
],
)
# TODO(cais): Run the test in OSS, perhaps through a sh_test.
cuda_py_test(
name = "dist_session_debug_grpc_test",
size = "medium",
srcs = ["lib/dist_session_debug_grpc_test.py"],
grpc_enabled = True,
python_version = "PY3",
tags = [
"no_oss", # Incompatible with bazel_pip.
"no_windows",
"notap", # TODO(b/160822489) flaky timeouts
"notsan",
],
xla_enable_strict_auto_jit = False, # Node names are different with autojit
deps = [
":debug_data",
":debug_utils",
":dumping_wrapper",
":grpc_debug_test_server",
":grpc_wrapper",
":hooks",
"//tensorflow/python:client",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:variables",
],
)
py_test(
name = "dumping_wrapper_test",
size = "small",
srcs = ["wrappers/dumping_wrapper_test.py"],
python_version = "PY3",
srcs_version = "PY3",
deps = [
":debug_data",
":dumping_wrapper",
":framework",
":hooks",
"//tensorflow/python:array_ops",
"//tensorflow/python:client",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:platform",
"//tensorflow/python:platform_test",
"//tensorflow/python:state_ops",
"//tensorflow/python:training",
"//tensorflow/python:variables",
],
)
py_test(
name = "local_cli_wrapper_test",
size = "small",
srcs = ["wrappers/local_cli_wrapper_test.py"],
python_version = "PY3",
srcs_version = "PY3",
deps = [
":cli_shared",
":debugger_cli_common",
":local_cli_wrapper",
":ui_factory",
"//tensorflow/core:protos_all_py",
"//tensorflow/python:array_ops",
"//tensorflow/python:client",
"//tensorflow/python:control_flow_ops",
"//tensorflow/python:errors",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:resource_variable_ops",
"//tensorflow/python:state_ops",
"//tensorflow/python:training",
"//tensorflow/python:variables",
"//third_party/py/numpy",
],
)
py_test(
name = "disk_usage_test",
size = "small",
srcs = ["wrappers/disk_usage_test.py"],
python_version = "PY3",
srcs_version = "PY3",
deps = [
":dumping_wrapper",
":hooks",
"//tensorflow/python:client",
"//tensorflow/python:errors",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:platform_test",
"//tensorflow/python:training",
"//tensorflow/python:variables",
],
)
sh_test(
name = "examples_v1_test",
size = "medium",
srcs = ["examples/v1/examples_v1_test.sh"],
data = [
":debug_errors",
":debug_fibonacci",
":debug_keras",
":debug_mnist_v1",
":debug_tflearn_iris",
":offline_analyzer",
],
tags = [
"no_windows",
"noasan", # TODO(b/143150907)
"nomsan", # TODO(b/143150907)
"v1only",
],
)
sh_test(
name = "examples_v2_test",
size = "medium",
srcs = ["examples/v2/examples_v2_test.sh"],
data = [
":debug_fibonacci_v2",
":debug_mnist_v2",
],
tags = [
"no_windows",
],
)