experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Move additional_deps to deps for cuda_py_test.

Browse Source 
PiperOrigin-RevId: 285283853
Change-Id: I2534d9fb51955cc9a86d1900ec60fc265f451ddc
This commit is contained in:
Brian Atkinson 2019-12-12 15:24:41 -08:00 committed by TensorFlower Gardener
parent 356de62421
commit 2fcfa6085b
23 changed files with 1484 additions and 1484 deletions
tensorflow
compiler/tests
BUILD
examples/saved_model/integration_tests
BUILD
python
BUILD
compiler
tensorrt
BUILD
xla
BUILD
data
experimental/kernel_tests
BUILD
kernel_tests
BUILD
debug
BUILD
distribute
BUILD
keras
BUILD
distribute
BUILD
mixed_precision/experimental
BUILD
optimizer_v2
BUILD
kernel_tests
BUILD
distributions
BUILD
linalg
BUILD
sparse
BUILD
random
BUILD
ops/parallel_for
BUILD
profiler
BUILD
internal
BUILD
saved_model
BUILD
training/saving
BUILD

22
tensorflow/compiler/tests/BUILD
View File

@ -1271,7 +1271,8 @@ cuda_py_test(
name = "xla_device_gpu_test",
size = "small",
srcs = ["xla_device_gpu_test.py"],
additional_deps = [
xla_enable_strict_auto_jit = False,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client",
"//tensorflow/python:client_testlib",
@ -1279,16 +1280,16 @@ cuda_py_test(
"//tensorflow/python:framework",
"//tensorflow/python:math_ops",
],
xla_enable_strict_auto_jit = False,
)
cuda_py_test(
name = "jit_test",
size = "medium",
srcs = ["jit_test.py"],
additional_deps = [
shard_count = 5,
xla_enable_strict_auto_jit = False,
deps = [
":test_utils",
"//tensorflow/python/compiler/xla:compiler_py",
"//tensorflow/core:protos_all_py",
"//tensorflow/python:array_ops",
"//tensorflow/python:client",
@ -1298,25 +1299,24 @@ cuda_py_test(
"//tensorflow/python:gradients",
"//tensorflow/python:math_ops",
"//tensorflow/python:nn_ops",
"//tensorflow/python/compiler/xla:compiler_py",
],
shard_count = 5,
xla_enable_strict_auto_jit = False,
)
cuda_py_test(
name = "dense_layer_test",
size = "medium",
srcs = ["dense_layer_test.py"],
additional_deps = [
xla_enable_strict_auto_jit = False,
deps = [
":test_utils",
"//tensorflow/python/compiler/xla:compiler_py",
"//tensorflow/core:protos_all_py",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:layers",
"//tensorflow/python:variables",
"//tensorflow/python/compiler/xla:compiler_py",
],
xla_enable_strict_auto_jit = False,
)
cc_library(
@ -1391,7 +1391,8 @@ py_library(
cuda_py_test(
name = "lstm_test",
srcs = ["lstm_test.py"],
additional_deps = [
xla_enable_strict_auto_jit = False,
deps = [
":lstm",
":xla_test",
"//tensorflow/python:array_ops",
@ -1403,7 +1404,6 @@ cuda_py_test(
"//tensorflow/python:platform",
"//tensorflow/python:variables",
],
xla_enable_strict_auto_jit = False,
)
# An example of ahead-of-time compilation using tfcompile. The

14
tensorflow/examples/saved_model/integration_tests/BUILD
View File

@ -53,13 +53,6 @@ cuda_py_test(
srcs = [
"saved_model_test.py",
],
additional_deps = [
":distribution_strategy_utils",
":integration_scripts",
"@absl_py//absl/testing:parameterized",
"//tensorflow:tensorflow_py",
"//tensorflow/python/distribute:combinations",
],
shard_count = 4,
tags = [
"no_pip", # b/131697937 and b/132196869
@ -67,6 +60,13 @@ cuda_py_test(
"nomsan", # forge input size exceeded
"notsan", # forge input size exceeded
],
deps = [
":distribution_strategy_utils",
":integration_scripts",
"//tensorflow:tensorflow_py",
"//tensorflow/python/distribute:combinations",
"@absl_py//absl/testing:parameterized",
],
)
# b/132234211: Target added to support internal test target that runs the test

442
tensorflow/python/BUILD
View File

File diff suppressed because it is too large Load Diff

44
tensorflow/python/compiler/tensorrt/BUILD
View File

@ -79,21 +79,6 @@ py_library(
cuda_py_test(
name = "trt_convert_test",
srcs = ["trt_convert_test.py"],
additional_deps = [
":trt_convert_py",
"@absl_py//absl/testing:parameterized",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:graph_util",
"//tensorflow/python/saved_model:builder",
"//tensorflow/python/saved_model:loader",
"//tensorflow/python/saved_model:signature_constants",
"//tensorflow/python/saved_model:signature_def_utils",
"//tensorflow/python/saved_model:tag_constants",
"//tensorflow/python/saved_model:utils",
"//tensorflow/python/tools:freeze_graph_lib",
"//tensorflow/python/tools:saved_model_utils",
],
data = [
"test/testdata/tftrt_2.0_saved_model/saved_model.pb",
"test/testdata/tftrt_2.0_saved_model/variables/variables.data-00000-of-00002",
@ -108,6 +93,21 @@ cuda_py_test(
"no_windows",
"nomac",
],
deps = [
":trt_convert_py",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:graph_util",
"//tensorflow/python/saved_model:builder",
"//tensorflow/python/saved_model:loader",
"//tensorflow/python/saved_model:signature_constants",
"//tensorflow/python/saved_model:signature_def_utils",
"//tensorflow/python/saved_model:tag_constants",
"//tensorflow/python/saved_model:utils",
"//tensorflow/python/tools:freeze_graph_lib",
"//tensorflow/python/tools:saved_model_utils",
"@absl_py//absl/testing:parameterized",
],
)
cuda_py_tests(
@ -171,13 +171,6 @@ cuda_py_tests(
cuda_py_test(
name = "quantization_mnist_test",
srcs = ["test/quantization_mnist_test.py"],
additional_deps = [
":tf_trt_integration_test_base",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python/keras:keras",
"//tensorflow/python/estimator:estimator",
],
data = [
"test/testdata/mnist/checkpoint",
"test/testdata/mnist/model.ckpt-46900.data-00000-of-00001",
@ -193,4 +186,11 @@ cuda_py_test(
"no_windows",
"nomac",
],
deps = [
":tf_trt_integration_test_base",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python/estimator",
"//tensorflow/python/keras",
],
)

38
tensorflow/python/compiler/xla/BUILD
View File

@ -24,9 +24,9 @@ cuda_py_test(
name = "jit_test",
size = "small",
srcs = ["jit_test.py"],
additional_deps = [
xla_enabled = True,
deps = [
":compiler_py",
"//third_party/py/numpy",
"//tensorflow/python:array_ops",
"//tensorflow/python:client",
"//tensorflow/python:client_testlib",
@ -40,8 +40,8 @@ cuda_py_test(
"//tensorflow/python:random_ops",
"//tensorflow/python:variable_scope",
"//tensorflow/python:variables",
"//third_party/py/numpy",
],
xla_enabled = True,
)
py_library(
@ -68,26 +68,26 @@ py_library(
cuda_py_test(
name = "xla_test",
srcs = ["xla_test.py"],
additional_deps = [
":xla",
"@absl_py//absl/testing:parameterized",
"//tensorflow/compiler/tests:xla_test",
"//tensorflow/python/tpu:tpu_lib",
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
"//tensorflow/python:control_flow_ops",
"//tensorflow/python:control_flow_util",
"//tensorflow/python:math_ops",
"//tensorflow/python/estimator:estimator_py",
"//tensorflow/python:platform",
"//tensorflow/python:state_ops",
"//tensorflow/python:summary",
"//tensorflow/python:variable_scope",
],
tags = [
"no_mac",
"no_rocm", # XLA support is not enabled on the ROCm platform
"no_windows",
],
xla_enabled = True,
deps = [
":xla",
"//tensorflow/compiler/tests:xla_test",
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
"//tensorflow/python:control_flow_ops",
"//tensorflow/python:control_flow_util",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform",
"//tensorflow/python:state_ops",
"//tensorflow/python:summary",
"//tensorflow/python:variable_scope",
"//tensorflow/python/estimator:estimator_py",
"//tensorflow/python/tpu:tpu_lib",
"@absl_py//absl/testing:parameterized",
],
)

26
tensorflow/python/data/experimental/kernel_tests/BUILD
View File

@ -80,20 +80,20 @@ cuda_py_test(
name = "copy_to_device_test",
size = "small",
srcs = ["copy_to_device_test.py"],
additional_deps = [
"//tensorflow/python/data/experimental/ops:prefetching_ops",
tags = ["no_windows_gpu"],
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:client_testlib",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python:dtypes",
"//tensorflow/python:framework_ops",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python/compat:compat",
"//tensorflow/python/compat",
"//tensorflow/python/data/experimental/ops:prefetching_ops",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/ops:iterator_ops",
],
tags = ["no_windows_gpu"],
)
tf_py_test(
@ -497,17 +497,17 @@ cuda_py_test(
name = "prefetch_to_device_test",
size = "small",
srcs = ["prefetch_to_device_test.py"],
additional_deps = [
"//tensorflow/python/data/experimental/ops:prefetching_ops",
tags = ["no_windows_gpu"],
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:client_testlib",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python:dtypes",
"//tensorflow/python:framework_ops",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python/data/experimental/ops:prefetching_ops",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
],
tags = ["no_windows_gpu"],
)
tf_py_test(
@ -624,8 +624,8 @@ cuda_py_test(
name = "scan_test",
size = "small",
srcs = ["scan_test.py"],
additional_deps = [
"//third_party/py/numpy",
tags = ["no_pip"],
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
@ -641,8 +641,8 @@ cuda_py_test(
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/eager:context",
"//third_party/py/numpy",
],
tags = ["no_pip"],
)
tf_py_test(
@ -829,7 +829,7 @@ cuda_py_test(
name = "wrap_unwrap_test",
size = "small",
srcs = ["wrap_unwrap_test.py"],
additional_deps = [
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",

76
tensorflow/python/data/kernel_tests/BUILD
View File

@ -316,17 +316,10 @@ cuda_py_test(
name = "iterator_test",
size = "medium",
srcs = ["iterator_test.py"],
additional_deps = [
grpc_enabled = True,
deps = [
":test_base",
"@absl_py//absl/testing:parameterized",
"//third_party/py/numpy",
"//tensorflow/python/data/ops:readers",
"//tensorflow/core:protos_all_py",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/ops:iterator_ops",
"//tensorflow/python/data/util:structure",
"//tensorflow/python/eager:context",
"//tensorflow/python/training/tracking:util",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
@ -347,11 +340,18 @@ cuda_py_test(
"//tensorflow/python:sparse_tensor",
"//tensorflow/python:tensor_shape",
"//tensorflow/python:training",
"//tensorflow/python/compat:compat",
"//tensorflow/python:util",
"//tensorflow/python:variables",
"//tensorflow/python/compat",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/ops:iterator_ops",
"//tensorflow/python/data/ops:readers",
"//tensorflow/python/data/util:structure",
"//tensorflow/python/eager:context",
"//tensorflow/python/training/tracking:util",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
grpc_enabled = True,
)
tf_py_test(
@ -408,25 +408,25 @@ cuda_py_test(
name = "multi_device_iterator_test",
size = "medium",
srcs = ["multi_device_iterator_test.py"],
additional_deps = [
tags = [
"no_gpu", #TODO(b/141255188): Enable test after bug is resolved.
"no_oss",
"no_windows_gpu",
],
deps = [
":test_base",
"@absl_py//absl/testing:parameterized",
"//tensorflow/core:protos_all_py",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/ops:multi_device_iterator_ops",
"//tensorflow/python/data/ops:iterator_ops",
"//tensorflow/python/data/experimental/ops:testing",
"//tensorflow/python/data/experimental/ops:optimization_options",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
"//tensorflow/python:errors",
"//tensorflow/python:framework_test_lib",
],
tags = [
"no_gpu", #TODO(b/141255188): Enable test after bug is resolved.
"no_oss",
"no_windows_gpu",
"//tensorflow/python/data/experimental/ops:optimization_options",
"//tensorflow/python/data/experimental/ops:testing",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/ops:iterator_ops",
"//tensorflow/python/data/ops:multi_device_iterator_ops",
"@absl_py//absl/testing:parameterized",
],
)
@ -434,16 +434,16 @@ cuda_py_test(
name = "memory_cleanup_test",
size = "medium",
srcs = ["memory_cleanup_test.py"],
additional_deps = [
tags = ["notsan"], # b/144706539
deps = [
":test_base",
"@absl_py//absl/testing:parameterized",
"//tensorflow/core:protos_all_py",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/ops:multi_device_iterator_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/ops:multi_device_iterator_ops",
"@absl_py//absl/testing:parameterized",
],
tags = ["notsan"], # b/144706539
)
tf_py_test(
@ -467,20 +467,20 @@ cuda_py_test(
name = "optional_test",
size = "small",
srcs = ["optional_test.py"],
additional_deps = [
deps = [
":test_base",
"@absl_py//absl/testing:parameterized",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/ops:iterator_ops",
"//tensorflow/python/data/ops:optional_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
"//tensorflow/python:dtypes",
"//tensorflow/python:errors",
"//tensorflow/python:framework_ops",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:tensor_shape",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/ops:iterator_ops",
"//tensorflow/python/data/ops:optional_ops",
"@absl_py//absl/testing:parameterized",
],
)
@ -550,17 +550,17 @@ cuda_py_test(
name = "reduce_test",
size = "small",
srcs = ["reduce_test.py"],
additional_deps = [
deps = [
":test_base",
"@absl_py//absl/testing:parameterized",
"//third_party/py/numpy",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
"//tensorflow/python:math_ops",
"//tensorflow/python/data/experimental/ops:testing",
"//tensorflow/python:sparse_tensor",
"//tensorflow/python/data/experimental/ops:testing",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)

178
tensorflow/python/debug/BUILD
View File

@ -695,36 +695,21 @@ cuda_py_test(
name = "check_numerics_callback_test",
size = "medium",
srcs = ["lib/check_numerics_callback_test.py"],
additional_deps = [
python_version = "PY3",
deps = [
":check_numerics_callback",
"//third_party/py/numpy",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:variables",
"//third_party/py/numpy",
],
python_version = "PY3",
)
cuda_py_test(
name = "distributed_callbacks_test",
size = "medium",
srcs = ["lib/distributed_callbacks_test.py"],
additional_deps = [
":check_numerics_callback",
":debug_events_writer",
":dumping_callback",
":dumping_callback_test_lib",
"//third_party/py/numpy",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:variables",
"//tensorflow/python/distribute:combinations",
"//tensorflow/python/distribute:mirrored_strategy",
"//tensorflow/python/distribute:strategy_combinations",
"//tensorflow/python/keras",
],
python_version = "PY3",
shard_count = 4,
tags = [
@ -735,41 +720,59 @@ cuda_py_test(
"no_windows_gpu", # TODO(b/130551176)
],
xla_enable_strict_auto_jit = False, # Node names are different with autojit
deps = [
":check_numerics_callback",
":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",
"//tensorflow/python/distribute:combinations",
"//tensorflow/python/distribute:mirrored_strategy",
"//tensorflow/python/distribute:strategy_combinations",
"//tensorflow/python/keras",
"//third_party/py/numpy",
],
)
cuda_py_test(
name = "dumping_callback_test",
size = "medium",
srcs = ["lib/dumping_callback_test.py"],
additional_deps = [
":debug_events_reader",
":debug_events_writer",
":dumping_callback",
":dumping_callback_test_lib",
"//third_party/py/numpy",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:variables",
"//tensorflow/python/keras",
],
python_version = "PY3",
shard_count = 8,
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",
"//tensorflow/python/keras",
"//third_party/py/numpy",
],
)
cuda_py_test(
name = "debug_v2_ops_test",
size = "medium",
srcs = ["lib/debug_v2_ops_test.py"],
additional_deps = [
python_version = "PY3",
tags = [
"no_windows", # b/142475891
],
deps = [
":debug_events_reader",
":debug_events_writer",
":dumping_callback_test_lib",
"//third_party/py/numpy",
"//tensorflow/python:debug_ops_gen",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:gradients",
@ -779,10 +782,7 @@ cuda_py_test(
"//tensorflow/python:variables",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/keras",
],
python_version = "PY3",
tags = [
"no_windows", # b/142475891
"//third_party/py/numpy",
],
)
@ -790,7 +790,9 @@ cuda_py_test(
name = "debug_gradients_test",
size = "small",
srcs = ["lib/debug_gradients_test.py"],
additional_deps = [
python_version = "PY3",
xla_enable_strict_auto_jit = False, # Node names are different with autojit
deps = [
":debug_data",
":debug_gradients",
":debug_utils",
@ -802,8 +804,6 @@ cuda_py_test(
"//tensorflow/python:training",
"//tensorflow/python:variables",
],
python_version = "PY3",
xla_enable_strict_auto_jit = False, # Node names are different with autojit
)
py_test(
@ -1030,7 +1030,9 @@ cuda_py_test(
name = "debug_grappler_test",
size = "small",
srcs = ["lib/debug_grappler_test.py"],
additional_deps = [
python_version = "PY3",
xla_enable_strict_auto_jit = False, # Tests TF:Classic implementation.
deps = [
":debug_data",
":debug_utils",
"//tensorflow/python:client",
@ -1039,15 +1041,16 @@ cuda_py_test(
"//tensorflow/python:platform_test",
"//tensorflow/python:variables",
],
python_version = "PY3",
xla_enable_strict_auto_jit = False, # Tests TF:Classic implementation.
)
cuda_py_test(
name = "session_debug_file_test",
size = "small",
srcs = ["lib/session_debug_file_test.py"],
additional_deps = [
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",
@ -1057,16 +1060,15 @@ cuda_py_test(
"//tensorflow/python:platform_test",
"//tensorflow/python:variables",
],
python_version = "PY3",
tags = ["notsan"],
xla_enable_strict_auto_jit = False, # Node names are different with autojit
)
cuda_py_test(
name = "debug_graph_reconstruction_test",
size = "small",
srcs = ["lib/debug_graph_reconstruction_test.py"],
additional_deps = [
python_version = "PY3",
xla_enable_strict_auto_jit = False, # Node names are different with autojit
deps = [
":debug_data",
":debug_utils",
"//tensorflow/python:client",
@ -1077,15 +1079,16 @@ cuda_py_test(
"//tensorflow/python:training",
"//tensorflow/python:variables",
],
python_version = "PY3",
xla_enable_strict_auto_jit = False, # Node names are different with autojit
)
cuda_py_test(
name = "session_debug_multi_gpu_test",
size = "small",
srcs = ["lib/session_debug_multi_gpu_test.py"],
additional_deps = [
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",
@ -1095,9 +1098,6 @@ cuda_py_test(
"//tensorflow/python:platform_test",
"//tensorflow/python:variables",
],
python_version = "PY3",
tags = ["no_windows_gpu"],
xla_enable_strict_auto_jit = False, # Node names are different with autojit
)
py_test(
@ -1205,7 +1205,10 @@ cuda_py_test(
name = "analyzer_cli_test",
size = "small",
srcs = ["cli/analyzer_cli_test.py"],
additional_deps = [
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",
@ -1214,8 +1217,6 @@ cuda_py_test(
":debug_utils",
":debugger_cli_common",
":source_utils",
"//third_party/py/numpy",
"@six_archive//:six",
"//tensorflow:tensorflow_py",
"//tensorflow/python:array_ops",
"//tensorflow/python:client",
@ -1227,10 +1228,9 @@ cuda_py_test(
"//tensorflow/python:platform_test",
"//tensorflow/python:util",
"//tensorflow/python:variables",
"//third_party/py/numpy",
"@six_archive//:six",
],
python_version = "PY3",
tags = ["no_windows"], # TODO: needs investigation on Windows
xla_enable_strict_auto_jit = False, # Node names are different with autojit
)
py_test(
@ -1258,7 +1258,15 @@ cuda_py_test(
name = "session_debug_grpc_test",
size = "medium",
srcs = ["lib/session_debug_grpc_test.py"],
additional_deps = [
python_version = "PY3",
tags = [
"no_oss", # Test flaky due to port collisions.
"no_windows",
"notsan",
"oss_serial",
],
xla_enable_strict_auto_jit = False, # Node names are different with autojit
deps = [
":debug_data",
":debug_utils",
":dumping_wrapper",
@ -1273,32 +1281,12 @@ cuda_py_test(
"//tensorflow/python:platform_test",
"//tensorflow/python:variables",
],
python_version = "PY3",
tags = [
"no_oss", # Test flaky due to port collisions.
"no_windows",
"notsan",
"oss_serial",
],
xla_enable_strict_auto_jit = False, # Node names are different with autojit
)
cuda_py_test(
name = "grpc_large_data_test",
size = "medium",
srcs = ["lib/grpc_large_data_test.py"],
additional_deps = [
":dumping_wrapper",
":grpc_debug_test_server",
":grpc_wrapper",
":session_debug_testlib",
"//third_party/py/numpy",
"//tensorflow/python:client",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:platform_test",
"//tensorflow/python:variables",
],
python_version = "PY3",
tags = [
"no_oss", # Test flaky due to port collisions.
@ -1308,6 +1296,18 @@ cuda_py_test(
"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.
@ -1315,7 +1315,15 @@ cuda_py_test(
name = "dist_session_debug_grpc_test",
size = "medium",
srcs = ["lib/dist_session_debug_grpc_test.py"],
additional_deps = [
grpc_enabled = True,
python_version = "PY3",
tags = [
"no_oss", # Incompatible with bazel_pip.
"no_windows",
"notsan",
],
xla_enable_strict_auto_jit = False, # Node names are different with autojit
deps = [
":debug_data",
":debug_utils",
":dumping_wrapper",
@ -1329,14 +1337,6 @@ cuda_py_test(
"//tensorflow/python:platform_test",
"//tensorflow/python:variables",
],
grpc_enabled = True,
python_version = "PY3",
tags = [
"no_oss", # Incompatible with bazel_pip.
"no_windows",
"notsan",
],
xla_enable_strict_auto_jit = False, # Node names are different with autojit
)
py_test(

164
tensorflow/python/distribute/BUILD
View File

@ -113,7 +113,7 @@ py_library(
cuda_py_test(
name = "device_util_test",
srcs = ["device_util_test.py"],
additional_deps = [
deps = [
":device_util",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_ops",
@ -410,14 +410,13 @@ cuda_py_test(
srcs = [
"multi_worker_continuous_run_test.py",
],
additional_deps = [
deps = [
":collective_all_reduce_strategy",
":multi_process_runner",
":reduce_util",
"@absl_py//absl/testing:parameterized",
"//third_party/py/numpy",
"//tensorflow/python:array_ops",
"//tensorflow/python:errors",
"//tensorflow/python:framework_ops",
"//tensorflow/python:math_ops",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/distribute:combinations",
@ -425,7 +424,8 @@ cuda_py_test(
"//tensorflow/python/distribute:strategy_combinations",
"//tensorflow/python/eager:context",
"//tensorflow/python/eager:test",
"//tensorflow/python:framework_ops",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)
@ -487,17 +487,17 @@ py_library(
cuda_py_test(
name = "input_ops_test",
srcs = ["input_ops_test.py"],
additional_deps = [
deps = [
":input_ops",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/ops:readers",
"//tensorflow/python/data/util:structure",
"//tensorflow/python:errors",
"//tensorflow/python:client_testlib",
"//tensorflow/python:errors",
"//tensorflow/python:framework_ops",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:io_ops",
"//tensorflow/python:util",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/ops:readers",
"//tensorflow/python/data/util:structure",
],
)
@ -704,17 +704,17 @@ cuda_py_test(
name = "checkpoint_utils_test",
size = "medium",
srcs = ["checkpoint_utils_test.py"],
additional_deps = [
"//tensorflow/python/distribute:combinations",
"//tensorflow/python/distribute:strategy_combinations",
tags = [
"multi_and_single_gpu",
],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_ops",
"//tensorflow/python:training",
"//tensorflow/python:variable_scope",
"//tensorflow/python:variables",
],
tags = [
"multi_and_single_gpu",
"//tensorflow/python/distribute:combinations",
"//tensorflow/python/distribute:strategy_combinations",
],
)
@ -764,54 +764,54 @@ distribute_py_test(
cuda_py_test(
name = "cross_device_utils_test",
srcs = ["cross_device_utils_test.py"],
additional_deps = [
"//tensorflow/python/distribute:combinations",
"//tensorflow/python/distribute:strategy_combinations",
"@absl_py//absl/testing:parameterized",
deps = [
"//tensorflow/python:constant_op",
"//tensorflow/python:framework_ops",
"//tensorflow/python:math_ops",
"//tensorflow/python/distribute:combinations",
"//tensorflow/python/distribute:cross_device_utils",
"//tensorflow/python/distribute:strategy_combinations",
"//tensorflow/python/eager:context",
"//tensorflow/python/eager:test",
"@absl_py//absl/testing:parameterized",
],
)
cuda_py_test(
name = "cross_device_ops_test",
srcs = ["cross_device_ops_test.py"],
additional_deps = [
tags = [
# TODO(b/138143527): Re-enable after fixing Guitar failure.
# "multi_and_single_gpu",
],
deps = [
":collective_all_reduce_strategy",
"//tensorflow/python/distribute:combinations",
"//tensorflow/python/distribute:strategy_combinations",
"//tensorflow/python/distribute:multi_worker_test_base",
":mirrored_strategy",
"@absl_py//absl/testing:parameterized",
"//tensorflow/python:array_ops",
"//tensorflow/python:constant_op",
"//tensorflow/python:framework_ops",
"//tensorflow/python:math_ops",
"//tensorflow/python/distribute:combinations",
"//tensorflow/python/distribute:cross_device_ops",
"//tensorflow/python/distribute:multi_worker_test_base",
"//tensorflow/python/distribute:strategy_combinations",
"//tensorflow/python/distribute:values",
"//tensorflow/python/eager:context",
"//tensorflow/python/eager:test",
],
tags = [
# TODO(b/138143527): Re-enable after fixing Guitar failure.
# "multi_and_single_gpu",
"@absl_py//absl/testing:parameterized",
],
)
cuda_py_test(
name = "one_device_strategy_test",
srcs = ["one_device_strategy_test.py"],
additional_deps = [
grpc_enabled = True,
deps = [
":strategy_test_lib",
"//tensorflow/python/distribute:combinations",
"//tensorflow/python/distribute:strategy_combinations",
"//tensorflow/python/eager:test",
],
grpc_enabled = True,
)
py_library(
@ -1014,24 +1014,24 @@ cuda_py_test(
name = "warm_starting_util_test",
size = "medium",
srcs = ["warm_starting_util_test.py"],
additional_deps = [
"//tensorflow/python/distribute:combinations",
"//tensorflow/python/distribute:strategy_combinations",
tags = [
"multi_and_single_gpu",
],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_ops",
"//tensorflow/python:training",
"//tensorflow/python:variable_scope",
"//tensorflow/python:variables",
],
tags = [
"multi_and_single_gpu",
"//tensorflow/python/distribute:combinations",
"//tensorflow/python/distribute:strategy_combinations",
],
)
cuda_py_test(
name = "remote_mirrored_strategy_eager_test",
srcs = ["remote_mirrored_strategy_eager_test.py"],
additional_deps = [
deps = [
":combinations",
":mirrored_strategy",
":multi_worker_test_base",
@ -1054,27 +1054,6 @@ cuda_py_test(
cuda_py_test(
name = "mirrored_strategy_test",
srcs = ["mirrored_strategy_test.py"],
additional_deps = [
":combinations",
":strategy_combinations",
":mirrored_strategy",
":multi_worker_test_base",
":strategy_test_lib",
"//tensorflow/core:protos_all_py",
"//tensorflow/python:array_ops",
"//tensorflow/python/autograph/core:test_lib",
"//tensorflow/python:constant_op",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:layers",
"//tensorflow/python:state_ops",
"//tensorflow/python:tensor_shape",
"//tensorflow/python:tensor_util",
"//tensorflow/python:variable_scope",
"//tensorflow/python/distribute:distribute_lib",
"//tensorflow/python/distribute:values",
"//tensorflow/python/eager:context",
"//tensorflow/python/eager:test",
],
shard_count = 5,
tags = [
"guitar",
@ -1082,12 +1061,37 @@ cuda_py_test(
"no_rocm",
"no_windows_gpu", # TODO(b/130551176)
],
deps = [
":combinations",
":mirrored_strategy",
":multi_worker_test_base",
":strategy_combinations",
":strategy_test_lib",
"//tensorflow/core:protos_all_py",
"//tensorflow/python:array_ops",
"//tensorflow/python:constant_op",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:layers",
"//tensorflow/python:state_ops",
"//tensorflow/python:tensor_shape",
"//tensorflow/python:tensor_util",
"//tensorflow/python:variable_scope",
"//tensorflow/python/autograph/core:test_lib",
"//tensorflow/python/distribute:distribute_lib",
"//tensorflow/python/distribute:values",
"//tensorflow/python/eager:context",
"//tensorflow/python/eager:test",
],
)
cuda_py_test(
name = "mirrored_variable_test",
srcs = ["mirrored_variable_test.py"],
additional_deps = [
tags = [
"guitar",
"multi_and_single_gpu",
],
deps = [
":collective_all_reduce_strategy",
":combinations",
":strategy_combinations",
@ -1103,10 +1107,6 @@ cuda_py_test(
"//tensorflow/python/eager:context",
"//tensorflow/python/eager:test",
],
tags = [
"guitar",
"multi_and_single_gpu",
],
)
distribute_py_test(
@ -1255,14 +1255,16 @@ distribute_py_test(
cuda_py_test(
name = "collective_all_reduce_strategy_test",
srcs = ["collective_all_reduce_strategy_test.py"],
additional_deps = [
tags = [
"multi_and_single_gpu",
],
xla_enable_strict_auto_jit = False,
deps = [
":collective_all_reduce_strategy",
":combinations",
":strategy_combinations",
":multi_worker_test_base",
":strategy_combinations",
":strategy_test_lib",
"@absl_py//absl/testing:parameterized",
"//third_party/py/numpy",
"//tensorflow/core:protos_all_py",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
@ -1278,24 +1280,27 @@ cuda_py_test(
"//tensorflow/python/eager:context",
"//tensorflow/python/estimator:estimator_py",
"//tensorflow/python/keras/mixed_precision/experimental:test_util",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
tags = [
"multi_and_single_gpu",
],
xla_enable_strict_auto_jit = False,
)
cuda_py_test(
name = "parameter_server_strategy_test",
srcs = ["parameter_server_strategy_test.py"],
additional_deps = [
":parameter_server_strategy",
tags = [
"multi_and_single_gpu",
"no_oss", # TODO(b/133330625)
],
# b/141096229: Non-atomic AssignAdd
xla_enable_strict_auto_jit = False,
deps = [
":central_storage_strategy",
":combinations",
":strategy_combinations",
":multi_worker_test_base",
":parameter_server_strategy",
":strategy_combinations",
":strategy_test_lib",
"@absl_py//absl/testing:parameterized",
"//tensorflow/core:protos_all_py",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
@ -1313,13 +1318,8 @@ cuda_py_test(
"//tensorflow/python/distribute:values",
"//tensorflow/python/eager:context",
"//tensorflow/python/estimator:estimator_py",
"@absl_py//absl/testing:parameterized",
],
tags = [
"multi_and_single_gpu",
"no_oss", # TODO(b/133330625)
],
# b/141096229: Non-atomic AssignAdd
xla_enable_strict_auto_jit = False,
)
py_library(

142
tensorflow/python/keras/BUILD
View File

@ -770,59 +770,59 @@ cuda_py_test(
name = "convolutional_test",
size = "medium",
srcs = ["layers/convolutional_test.py"],
additional_deps = [
":keras",
"@absl_py//absl/testing:parameterized",
"//third_party/py/numpy",
"//tensorflow/python:client_testlib",
],
python_version = "PY3",
shard_count = 8,
deps = [
":keras",
"//tensorflow/python:client_testlib",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)
cuda_py_test(
name = "image_preprocessing_test",
size = "medium",
srcs = ["layers/preprocessing/image_preprocessing_test.py"],
additional_deps = [
":keras",
"@absl_py//absl/testing:parameterized",
"//third_party/py/numpy",
"//tensorflow/python:client_testlib",
],
python_version = "PY3",
shard_count = 4,
deps = [
":keras",
"//tensorflow/python:client_testlib",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)
cuda_py_test(
name = "convolutional_transpose_test",
size = "medium",
srcs = ["layers/convolutional_transpose_test.py"],
additional_deps = [
":keras",
"@absl_py//absl/testing:parameterized",
"//third_party/py/numpy",
"//tensorflow/python:client_testlib",
],
python_version = "PY3",
deps = [
":keras",
"//tensorflow/python:client_testlib",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)
cuda_py_test(
name = "cudnn_recurrent_test",
size = "medium",
srcs = ["layers/cudnn_recurrent_test.py"],
additional_deps = [
":keras",
"@absl_py//absl/testing:parameterized",
"//third_party/py/numpy",
"//tensorflow/python:client_testlib",
],
python_version = "PY3",
shard_count = 4,
tags = [
"no_rocm",
"no_windows_gpu",
],
deps = [
":keras",
"//tensorflow/python:client_testlib",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)
tf_py_test(
@ -902,12 +902,12 @@ cuda_py_test(
name = "embeddings_test",
size = "medium",
srcs = ["layers/embeddings_test.py"],
additional_deps = [
":keras",
"@absl_py//absl/testing:parameterized",
"//tensorflow/python:client_testlib",
],
python_version = "PY3",
deps = [
":keras",
"//tensorflow/python:client_testlib",
"@absl_py//absl/testing:parameterized",
],
)
tf_py_test(
@ -954,18 +954,18 @@ cuda_py_test(
name = "normalization_test",
size = "medium",
srcs = ["layers/normalization_test.py"],
additional_deps = [
":keras",
"@absl_py//absl/testing:parameterized",
"//third_party/py/numpy",
"//tensorflow/python:client_testlib",
],
python_version = "PY3",
shard_count = 4,
tags = [
"no_rocm",
"notsan",
],
deps = [
":keras",
"//tensorflow/python:client_testlib",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)
tf_py_test(
@ -1064,57 +1064,57 @@ cuda_py_test(
name = "recurrent_v2_test",
size = "medium",
srcs = ["layers/recurrent_v2_test.py"],
additional_deps = [
":keras",
"@absl_py//absl/testing:parameterized",
"//third_party/py/numpy",
"//tensorflow/python:client_testlib",
],
python_version = "PY3",
shard_count = 2,
deps = [
":keras",
"//tensorflow/python:client_testlib",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)
cuda_py_test(
name = "separable_convolutional_test",
size = "medium",
srcs = ["layers/separable_convolutional_test.py"],
additional_deps = [
":keras",
"@absl_py//absl/testing:parameterized",
"//third_party/py/numpy",
"//tensorflow/python:client_testlib",
],
python_version = "PY3",
deps = [
":keras",
"//tensorflow/python:client_testlib",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)
cuda_py_test(
name = "lstm_v2_test",
size = "medium",
srcs = ["layers/lstm_v2_test.py"],
additional_deps = [
":keras",
"@absl_py//absl/testing:parameterized",
"//third_party/py/numpy",
"//tensorflow/python:client_testlib",
],
python_version = "PY3",
shard_count = 12,
tags = ["no_rocm"],
deps = [
":keras",
"//tensorflow/python:client_testlib",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)
cuda_py_test(
name = "gru_v2_test",
size = "medium",
srcs = ["layers/gru_v2_test.py"],
additional_deps = [
":keras",
"@absl_py//absl/testing:parameterized",
"//third_party/py/numpy",
"//tensorflow/python:client_testlib",
],
python_version = "PY3",
shard_count = 12,
tags = ["no_rocm"],
deps = [
":keras",
"//tensorflow/python:client_testlib",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)
tf_py_test(
@ -1347,34 +1347,34 @@ tf_py_test(
cuda_py_test(
name = "multi_gpu_utils_test",
srcs = ["utils/multi_gpu_utils_test.py"],
additional_deps = [
":keras",
"@absl_py//absl/testing:parameterized",
"//third_party/py/numpy",
"//tensorflow/python:client_testlib",
],
python_version = "PY3",
tags = [
"guitar",
"multi_gpu",
],
xla_enable_strict_auto_jit = False, # b/142744009
deps = [
":keras",
"//tensorflow/python:client_testlib",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)
cuda_py_test(
name = "training_gpu_test",
size = "small",
srcs = ["engine/training_gpu_test.py"],
additional_deps = [
":keras",
"@absl_py//absl/testing:parameterized",
"//third_party/py/numpy",
"//tensorflow/python:client_testlib",
],
python_version = "PY3",
tags = [
"nomac", # TODO(mihaimaruseac): b/127695564
],
deps = [
":keras",
"//tensorflow/python:client_testlib",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)
tf_py_test(

88
tensorflow/python/keras/distribute/BUILD
View File

@ -63,7 +63,8 @@ py_library(
cuda_py_test(
name = "multi_worker_training_state_test",
srcs = ["multi_worker_training_state_test.py"],
additional_deps = [
shard_count = 4,
deps = [
":multi_worker_testing_utils",
":multi_worker_training_state",
"//tensorflow/python:platform",
@ -71,7 +72,6 @@ cuda_py_test(
"//tensorflow/python/distribute:multi_worker_test_base",
"//tensorflow/python/keras",
],
shard_count = 4,
)
py_library(
@ -305,23 +305,27 @@ py_library(
cuda_py_test(
name = "keras_optimizer_v2_test",
srcs = ["keras_optimizer_v2_test.py"],
additional_deps = [
":keras_test_lib",
],
shard_count = 4,
tags = [
"multi_and_single_gpu",
"no_oss", # http://b/119349471
"tf_integration_test",
],
deps = [
":keras_test_lib",
],
)
cuda_py_test(
name = "multi_worker_test",
srcs = ["multi_worker_test.py"],
additional_deps = [
shard_count = 32,
tags = [
"no_oss", # TODO(b/130369494): Investigate why it times out on OSS.
# TODO(b/123307453): Add "multi_and_single_gpu",
],
deps = [
":multi_worker_testing_utils",
"@absl_py//absl/testing:parameterized",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
@ -343,28 +347,13 @@ cuda_py_test(
"//tensorflow/python/keras:engine",
"//tensorflow/python/keras:optimizers",
"//tensorflow/python/keras/optimizer_v2",
],
shard_count = 32,
tags = [
"no_oss", # TODO(b/130369494): Investigate why it times out on OSS.
# TODO(b/123307453): Add "multi_and_single_gpu",
"@absl_py//absl/testing:parameterized",
],
)
cuda_py_test(
name = "multi_worker_callback_tf1_test",
srcs = ["multi_worker_callback_tf1_test.py"],
additional_deps = [
":distribute",
"//tensorflow/python:platform",
"//tensorflow/python/distribute:collective_all_reduce_strategy",
"//tensorflow/python/distribute:combinations",
"//tensorflow/python/distribute:multi_worker_test_base",
"//tensorflow/python/distribute:distribute_config",
"//tensorflow/python/distribute:distribute_coordinator",
":multi_worker_testing_utils",
"//tensorflow/python/keras",
],
# TODO(b/132384649): Enable for guitar and oss tests.
shard_count = 24,
tags = [
@ -373,12 +362,23 @@ cuda_py_test(
"noguitar",
"notap",
],
deps = [
":distribute",
":multi_worker_testing_utils",
"//tensorflow/python:platform",
"//tensorflow/python/distribute:collective_all_reduce_strategy",
"//tensorflow/python/distribute:combinations",
"//tensorflow/python/distribute:distribute_config",
"//tensorflow/python/distribute:distribute_coordinator",
"//tensorflow/python/distribute:multi_worker_test_base",
"//tensorflow/python/keras",
],
)
cuda_py_test(
name = "multi_worker_callback_tf2_test",
srcs = ["multi_worker_callback_tf2_test.py"],
additional_deps = [
deps = [
"//tensorflow/python/distribute:collective_all_reduce_strategy",
"//tensorflow/python/distribute:combinations",
"//tensorflow/python/distribute:multi_process_runner",
@ -390,42 +390,42 @@ cuda_py_test(
cuda_py_test(
name = "multi_worker_fault_tolerance_test",
srcs = ["multi_worker_fault_tolerance_test.py"],
additional_deps = [
":distribute",
"//tensorflow/python:platform",
"//tensorflow/python/distribute:collective_all_reduce_strategy",
"//tensorflow/python/distribute:combinations",
"//tensorflow/python/distribute:multi_worker_test_base",
"//tensorflow/python/distribute:distribute_config",
"//tensorflow/python/distribute:distribute_coordinator",
":multi_worker_testing_utils",
"//tensorflow/python/keras",
],
shard_count = 14,
# TODO(b/132384649): Enable once fixed.
tags = [
"no_oss",
],
deps = [
":distribute",
":multi_worker_testing_utils",
"//tensorflow/python:platform",
"//tensorflow/python/distribute:collective_all_reduce_strategy",
"//tensorflow/python/distribute:combinations",
"//tensorflow/python/distribute:distribute_config",
"//tensorflow/python/distribute:distribute_coordinator",
"//tensorflow/python/distribute:multi_worker_test_base",
"//tensorflow/python/keras",
],
)
cuda_py_test(
name = "multi_worker_optimizer_comparison_test",
srcs = ["multi_worker_optimizer_comparison_test.py"],
additional_deps = [
"//tensorflow/python/distribute:collective_all_reduce_strategy",
"//tensorflow/python/distribute:combinations",
"//tensorflow/python/keras/distribute:multi_worker_test",
"//tensorflow/python/distribute:multi_worker_test_base",
"//tensorflow/python:platform",
"//tensorflow/python/distribute:distribute_coordinator",
"//tensorflow/python/keras",
],
tags = [
"manual", # b/139844866
"multi_and_single_gpu",
"no_oss",
"notap",
],
deps = [
"//tensorflow/python:platform",
"//tensorflow/python/distribute:collective_all_reduce_strategy",
"//tensorflow/python/distribute:combinations",
"//tensorflow/python/distribute:distribute_coordinator",
"//tensorflow/python/distribute:multi_worker_test_base",
"//tensorflow/python/keras",
"//tensorflow/python/keras/distribute:multi_worker_test",
],
)
py_library(

22
tensorflow/python/keras/mixed_precision/experimental/BUILD
View File

@ -124,7 +124,8 @@ cuda_py_test(
name = "loss_scale_optimizer_test",
size = "small",
srcs = ["loss_scale_optimizer_test.py"],
additional_deps = [
python_version = "PY3",
deps = [
":loss_scale_optimizer",
":test_util",
"//tensorflow/python:client_testlib",
@ -133,7 +134,6 @@ cuda_py_test(
"//tensorflow/python/distribute:one_device_strategy",
"//tensorflow/python/keras",
],
python_version = "PY3",
)
py_library(
@ -150,30 +150,30 @@ cuda_py_test(
name = "keras_test",
size = "medium",
srcs = ["keras_test.py"],
additional_deps = [
python_version = "PY3",
shard_count = 4,
tags = ["no_windows"], # b/139083295: bfloat16 tests fail on Windows
deps = [
":test_util",
"@absl_py//absl/testing:parameterized",
"//tensorflow/python:client_testlib",
"//tensorflow/python/distribute:mirrored_strategy",
"//tensorflow/python/distribute:one_device_strategy",
"//tensorflow/python/keras",
"@absl_py//absl/testing:parameterized",
],
python_version = "PY3",
shard_count = 4,
tags = ["no_windows"], # b/139083295: bfloat16 tests fail on Windows
)
cuda_py_test(
name = "layer_correctness_test",
size = "medium",
srcs = ["layer_correctness_test.py"],
additional_deps = [
"@absl_py//absl/testing:parameterized",
python_version = "PY3",
tags = ["no_rocm"],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python/distribute:mirrored_strategy",
"//tensorflow/python/distribute:one_device_strategy",
"//tensorflow/python/keras",
"@absl_py//absl/testing:parameterized",
],
python_version = "PY3",
tags = ["no_rocm"],
)

94
tensorflow/python/keras/optimizer_v2/BUILD
View File

@ -63,155 +63,140 @@ cuda_py_test(
name = "adagrad_test",
size = "medium",
srcs = ["adagrad_test.py"],
additional_deps = [
shard_count = 4,
deps = [
":optimizer_v2",
"//tensorflow/python:client_testlib",
"//tensorflow/python:embedding_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:framework",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:resource_variable_ops",
"//tensorflow/python:resources",
"//tensorflow/python:variables",
"//tensorflow/python/eager:context",
],
shard_count = 4,
)
cuda_py_test(
name = "adam_test",
size = "medium",
srcs = ["adam_test.py"],
additional_deps = [
shard_count = 4,
deps = [
":optimizer_v2",
"//tensorflow/python:client_testlib",
"//tensorflow/python:embedding_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:framework",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:resource_variable_ops",
"//tensorflow/python:resources",
"//tensorflow/python:variables",
"//tensorflow/python/eager:context",
],
shard_count = 4,
)
cuda_py_test(
name = "adamax_test",
size = "medium",
srcs = ["adamax_test.py"],
additional_deps = [
shard_count = 4,
tags = ["no_rocm"],
deps = [
":optimizer_v2",
"//tensorflow/python:client_testlib",
"//tensorflow/python:embedding_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:framework",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:resource_variable_ops",
"//tensorflow/python:resources",
"//tensorflow/python:variables",
"//tensorflow/python/eager:context",
],
shard_count = 4,
tags = ["no_rocm"],
)
cuda_py_test(
name = "adadelta_test",
size = "medium",
srcs = ["adadelta_test.py"],
additional_deps = [
shard_count = 4,
deps = [
":optimizer_v2",
"//tensorflow/python:client_testlib",
"//tensorflow/python:embedding_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:framework",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:resource_variable_ops",
"//tensorflow/python:resources",
"//tensorflow/python:variables",
"//tensorflow/python/eager:context",
],
shard_count = 4,
)
cuda_py_test(
name = "ftrl_test",
size = "medium",
srcs = ["ftrl_test.py"],
additional_deps = [
shard_count = 4,
deps = [
":optimizer_v2",
"//tensorflow/python:client_testlib",
"//tensorflow/python:embedding_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:framework",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:resource_variable_ops",
"//tensorflow/python:resources",
"//tensorflow/python:variables",
"//tensorflow/python/eager:context",
],
shard_count = 4,
)
cuda_py_test(
name = "gradient_descent_test",
size = "medium",
srcs = ["gradient_descent_test.py"],
additional_deps = [
shard_count = 4,
deps = [
":optimizer_v2",
"//tensorflow/python:client_testlib",
"//tensorflow/python:embedding_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:framework",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:resource_variable_ops",
"//tensorflow/python:resources",
"//tensorflow/python:variables",
"//tensorflow/python/eager:context",
],
shard_count = 4,
)
cuda_py_test(
name = "nadam_test",
size = "medium",
srcs = ["nadam_test.py"],
additional_deps = [
shard_count = 4,
deps = [
":optimizer_v2",
"//tensorflow/python:client_testlib",
"//tensorflow/python:embedding_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:framework",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:resource_variable_ops",
"//tensorflow/python:resources",
"//tensorflow/python:variables",
"//tensorflow/python/eager:context",
],
shard_count = 4,
)
cuda_py_test(
name = "optimizer_v2_test",
size = "medium",
srcs = ["optimizer_v2_test.py"],
additional_deps = [
":optimizer_v2",
"@absl_py//absl/testing:parameterized",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:clip_ops",
"//tensorflow/python:framework",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:gradients",
"//tensorflow/python:resource_variable_ops",
"//tensorflow/python:state_ops",
"//tensorflow/python:variables",
"//tensorflow/python/eager:def_function",
"//tensorflow/python/keras",
],
shard_count = 8,
tags = [
"manual",
@ -220,39 +205,54 @@ cuda_py_test(
"no_windows",
"notap",
],
deps = [
":optimizer_v2",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:clip_ops",
"//tensorflow/python:framework",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:gradients",
"//tensorflow/python:resource_variable_ops",
"//tensorflow/python:state_ops",
"//tensorflow/python:variables",
"//tensorflow/python/eager:def_function",
"//tensorflow/python/keras",
"@absl_py//absl/testing:parameterized",
],
)
cuda_py_test(
name = "learning_rate_schedule_test",
size = "medium",
srcs = ["learning_rate_schedule_test.py"],
additional_deps = [
shard_count = 4,
deps = [
":optimizer_v2",
"@absl_py//absl/testing:parameterized",
"//third_party/py/numpy",
"//tensorflow/python:client_testlib",
"//tensorflow/python/keras",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
shard_count = 4,
)
cuda_py_test(
name = "rmsprop_test",
size = "medium",
srcs = ["rmsprop_test.py"],
additional_deps = [
shard_count = 2,
deps = [
":optimizer_v2",
"@absl_py//absl/testing:parameterized",
"//tensorflow/python/eager:def_function",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:clip_ops",
"//tensorflow/python:framework",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:array_ops",
"//tensorflow/python:clip_ops",
"//tensorflow/python:gradients",
"//tensorflow/python:resource_variable_ops",
"//tensorflow/python:state_ops",
"//tensorflow/python:variables",
"//tensorflow/python/eager:def_function",
"@absl_py//absl/testing:parameterized",
],
shard_count = 2,
)

1010
tensorflow/python/kernel_tests/BUILD
View File

File diff suppressed because it is too large Load Diff

136
tensorflow/python/kernel_tests/distributions/BUILD
View File

@ -11,16 +11,16 @@ cuda_py_test(
name = "bijector_test",
size = "small",
srcs = ["bijector_test.py"],
additional_deps = [
"//tensorflow/python/ops/distributions",
"//third_party/py/numpy",
"@six_archive//:six",
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python/ops/distributions",
"//third_party/py/numpy",
"@six_archive//:six",
],
)
@ -28,9 +28,9 @@ cuda_py_test(
name = "util_test",
size = "medium",
srcs = ["util_test.py"],
additional_deps = [
"//tensorflow/python/ops/distributions",
"//third_party/py/numpy",
shard_count = 3,
xla_enable_strict_auto_jit = False, # TODO(b/144920376)
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework",
@ -39,20 +39,20 @@ cuda_py_test(
"//tensorflow/python:gradients",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python/ops/distributions",
"//third_party/py/numpy",
],
shard_count = 3,
xla_enable_strict_auto_jit = False, # TODO(b/144920376)
)
cuda_py_test(
name = "kullback_leibler_test",
size = "small",
srcs = ["kullback_leibler_test.py"],
additional_deps = [
"//tensorflow/python/ops/distributions",
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:platform_test",
"//tensorflow/python/ops/distributions",
],
)
@ -60,9 +60,7 @@ cuda_py_test(
name = "beta_test",
size = "small",
srcs = ["beta_test.py"],
additional_deps = [
"//tensorflow/python/ops/distributions",
"//third_party/py/numpy",
deps = [
"//tensorflow/python:client",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework",
@ -70,6 +68,8 @@ cuda_py_test(
"//tensorflow/python:math_ops",
"//tensorflow/python:nn_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python/ops/distributions",
"//third_party/py/numpy",
],
)
@ -77,14 +77,14 @@ cuda_py_test(
name = "bernoulli_test",
size = "small",
srcs = ["bernoulli_test.py"],
additional_deps = [
"//tensorflow/python/ops/distributions",
"//third_party/py/numpy",
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python/ops/distributions",
"//third_party/py/numpy",
],
)
@ -92,10 +92,7 @@ cuda_py_test(
name = "categorical_test",
size = "small",
srcs = ["categorical_test.py"],
additional_deps = [
"@absl_py//absl/testing:parameterized",
"//tensorflow/python/ops/distributions",
"//third_party/py/numpy",
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework",
@ -104,6 +101,9 @@ cuda_py_test(
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:random_ops",
"//tensorflow/python/ops/distributions",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)
@ -111,13 +111,13 @@ cuda_py_test(
name = "dirichlet_test",
size = "small",
srcs = ["dirichlet_test.py"],
additional_deps = [
"//tensorflow/python/ops/distributions",
"//third_party/py/numpy",
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:platform_test",
"//tensorflow/python/ops/distributions",
"//third_party/py/numpy",
],
)
@ -125,98 +125,87 @@ cuda_py_test(
name = "dirichlet_multinomial_test",
size = "medium",
srcs = ["dirichlet_multinomial_test.py"],
additional_deps = [
"//tensorflow/python/ops/distributions",
"//third_party/py/numpy",
tags = [
"noguitar", # b/110489471
"notap", # b/110489471
],
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
],
tags = [
"noguitar", # b/110489471
"notap", # b/110489471
"//tensorflow/python/ops/distributions",
"//third_party/py/numpy",
],
)
cuda_py_test(
name = "exponential_test",
srcs = ["exponential_test.py"],
additional_deps = [
"//tensorflow/python/ops/distributions",
"//third_party/py/numpy",
deps = [
"//tensorflow/python:client",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:nn_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python/ops/distributions",
"//third_party/py/numpy",
],
)
cuda_py_test(
name = "gamma_test",
srcs = ["gamma_test.py"],
additional_deps = [
"//tensorflow/python/ops/distributions",
"//third_party/py/numpy",
deps = [
"//tensorflow/python:client",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:nn_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python/ops/distributions",
"//third_party/py/numpy",
],
)
cuda_py_test(
name = "laplace_test",
srcs = ["laplace_test.py"],
additional_deps = [
"//tensorflow/python/ops/distributions",
"//third_party/py/numpy",
deps = [
"//tensorflow/python:client",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:nn_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python/ops/distributions",
"//third_party/py/numpy",
],
)
cuda_py_test(
name = "multinomial_test",
srcs = ["multinomial_test.py"],
additional_deps = [
"//tensorflow/python/ops/distributions",
"//third_party/py/numpy",
tags = ["manual"], # b/69001419
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python/ops/distributions",
"//third_party/py/numpy",
],
tags = ["manual"], # b/69001419
)
cuda_py_test(
name = "student_t_test",
size = "small",
srcs = ["student_t_test.py"],
additional_deps = [
"//tensorflow/python/ops/distributions",
"//third_party/py/numpy",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python:nn_ops",
"//tensorflow/python:platform_test",
],
tags = [
# TODO(b/121223043): Re-enable this test after fixing "mean not defined"
# errors.
@ -225,21 +214,32 @@ cuda_py_test(
# disable to avoid false positives from scipy.
"nomsan",
],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python:nn_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python/ops/distributions",
"//third_party/py/numpy",
],
)
cuda_py_test(
name = "uniform_test",
size = "small",
srcs = ["uniform_test.py"],
additional_deps = [
"//tensorflow/python/ops/distributions",
"//third_party/py/numpy",
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:errors",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python/ops/distributions",
"//third_party/py/numpy",
],
)
@ -247,9 +247,7 @@ cuda_py_test(
name = "normal_test",
size = "medium",
srcs = ["normal_test.py"],
additional_deps = [
"//tensorflow/python/ops/distributions",
"//third_party/py/numpy",
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
@ -258,6 +256,8 @@ cuda_py_test(
"//tensorflow/python:nn_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:variables",
"//tensorflow/python/ops/distributions",
"//third_party/py/numpy",
],
)
@ -265,15 +265,15 @@ cuda_py_test(
name = "special_math_test",
size = "medium",
srcs = ["special_math_test.py"],
additional_deps = [
"//tensorflow/python/ops/distributions",
"//third_party/py/numpy",
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:gradients",
"//tensorflow/python:platform_test",
"//tensorflow/python:variables",
"//tensorflow/python/ops/distributions",
"//third_party/py/numpy",
],
)
@ -281,15 +281,15 @@ cuda_py_test(
name = "identity_bijector_test",
size = "small",
srcs = ["identity_bijector_test.py"],
additional_deps = [
"//tensorflow/python/ops/distributions",
"//third_party/py/numpy",
"@six_archive//:six",
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python/ops/distributions",
"//third_party/py/numpy",
"@six_archive//:six",
],
)

254
tensorflow/python/kernel_tests/linalg/BUILD
View File

@ -11,15 +11,15 @@ cuda_py_test(
name = "linear_operator_test",
size = "small",
srcs = ["linear_operator_test.py"],
additional_deps = [
"//tensorflow/python/ops/linalg",
"//third_party/py/numpy",
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python/ops/linalg",
"//third_party/py/numpy",
],
)
@ -27,15 +27,15 @@ cuda_py_test(
name = "linear_operator_addition_test",
size = "small",
srcs = ["linear_operator_addition_test.py"],
additional_deps = [
"//tensorflow/python/ops/linalg",
"//third_party/py/numpy",
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python/ops/linalg",
"//third_party/py/numpy",
],
)
@ -43,9 +43,12 @@ cuda_py_test(
name = "linear_operator_adjoint_test",
size = "medium",
srcs = ["linear_operator_adjoint_test.py"],
additional_deps = [
"//tensorflow/python/ops/linalg",
"//third_party/py/numpy",
shard_count = 5,
tags = [
"noasan", # times out, b/63678675
"optonly", # times out
],
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework",
@ -53,11 +56,8 @@ cuda_py_test(
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
],
shard_count = 5,
tags = [
"noasan", # times out, b/63678675
"optonly", # times out
"//tensorflow/python/ops/linalg",
"//third_party/py/numpy",
],
)
@ -65,15 +65,15 @@ cuda_py_test(
name = "linear_operator_algebra_test",
size = "small",
srcs = ["linear_operator_algebra_test.py"],
additional_deps = [
"//tensorflow/python/ops/linalg",
"//third_party/py/numpy",
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python/ops/linalg",
"//third_party/py/numpy",
],
)
@ -81,9 +81,12 @@ cuda_py_test(
name = "linear_operator_block_diag_test",
size = "medium",
srcs = ["linear_operator_block_diag_test.py"],
additional_deps = [
"//tensorflow/python/ops/linalg",
"//third_party/py/numpy",
shard_count = 6,
tags = [
"noasan",
"optonly",
],
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework",
@ -91,11 +94,8 @@ cuda_py_test(
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
],
shard_count = 6,
tags = [
"noasan",
"optonly",
"//tensorflow/python/ops/linalg",
"//third_party/py/numpy",
],
)
@ -103,9 +103,12 @@ cuda_py_test(
name = "linear_operator_composition_test",
size = "medium",
srcs = ["linear_operator_composition_test.py"],
additional_deps = [
"//tensorflow/python/ops/linalg",
"//third_party/py/numpy",
shard_count = 5,
tags = [
"noasan", # times out, b/63678675
"optonly", # times out
],
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework",
@ -113,11 +116,8 @@ cuda_py_test(
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
],
shard_count = 5,
tags = [
"noasan", # times out, b/63678675
"optonly", # times out
"//tensorflow/python/ops/linalg",
"//third_party/py/numpy",
],
)
@ -125,8 +125,13 @@ cuda_py_test(
name = "linear_operator_circulant_test",
size = "medium",
srcs = ["linear_operator_circulant_test.py"],
additional_deps = [
"//third_party/py/numpy",
shard_count = 10,
tags = [
"no_rocm", # calls BLAS ops for complex types
"noasan", # times out, b/63678675
"optonly", # times out, b/79171797
],
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework",
@ -136,12 +141,7 @@ cuda_py_test(
"//tensorflow/python:platform_test",
"//tensorflow/python/ops/linalg",
"//tensorflow/python/ops/signal",
],
shard_count = 10,
tags = [
"no_rocm", # calls BLAS ops for complex types
"noasan", # times out, b/63678675
"optonly", # times out, b/79171797
"//third_party/py/numpy",
],
)
@ -149,8 +149,12 @@ cuda_py_test(
name = "linear_operator_diag_test",
size = "medium",
srcs = ["linear_operator_diag_test.py"],
additional_deps = [
"//tensorflow/python/ops/linalg",
shard_count = 5,
tags = [
"noasan",
"optonly",
],
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework",
@ -159,11 +163,7 @@ cuda_py_test(
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:random_ops",
],
shard_count = 5,
tags = [
"noasan",
"optonly",
"//tensorflow/python/ops/linalg",
],
)
@ -171,8 +171,12 @@ cuda_py_test(
name = "linear_operator_householder_test",
size = "medium",
srcs = ["linear_operator_householder_test.py"],
additional_deps = [
"//tensorflow/python/ops/linalg",
shard_count = 5,
tags = [
"noasan",
"optonly",
],
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework",
@ -181,11 +185,7 @@ cuda_py_test(
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:random_ops",
],
shard_count = 5,
tags = [
"noasan",
"optonly",
"//tensorflow/python/ops/linalg",
],
)
@ -193,8 +193,12 @@ cuda_py_test(
name = "linear_operator_identity_test",
size = "medium",
srcs = ["linear_operator_identity_test.py"],
additional_deps = [
"//tensorflow/python/ops/linalg",
shard_count = 5,
tags = [
"noasan",
"optonly",
],
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework",
@ -202,11 +206,7 @@ cuda_py_test(
"//tensorflow/python:linalg_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:random_ops",
],
shard_count = 5,
tags = [
"noasan",
"optonly",
"//tensorflow/python/ops/linalg",
],
)
@ -214,9 +214,12 @@ cuda_py_test(
name = "linear_operator_inversion_test",
size = "medium",
srcs = ["linear_operator_inversion_test.py"],
additional_deps = [
"//tensorflow/python/ops/linalg",
"//third_party/py/numpy",
shard_count = 5,
tags = [
"noasan", # times out, b/63678675
"optonly", # times out
],
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework",
@ -224,11 +227,8 @@ cuda_py_test(
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
],
shard_count = 5,
tags = [
"noasan", # times out, b/63678675
"optonly", # times out
"//tensorflow/python/ops/linalg",
"//third_party/py/numpy",
],
)
@ -236,19 +236,19 @@ cuda_py_test(
name = "linear_operator_full_matrix_test",
size = "medium",
srcs = ["linear_operator_full_matrix_test.py"],
additional_deps = [
"//tensorflow/python/ops/linalg",
shard_count = 5,
tags = [
"noasan",
"optonly",
],
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:platform_test",
],
shard_count = 5,
tags = [
"noasan",
"optonly",
"//tensorflow/python/ops/linalg",
],
)
@ -256,9 +256,13 @@ cuda_py_test(
name = "linear_operator_kronecker_test",
size = "medium",
srcs = ["linear_operator_kronecker_test.py"],
additional_deps = [
"//tensorflow/python/ops/linalg",
"//third_party/py/numpy",
shard_count = 10,
tags = [
"noasan",
"optonly",
],
xla_enable_strict_auto_jit = False, # Fails in XLA b/143610154
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework",
@ -266,31 +270,27 @@ cuda_py_test(
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python/ops/linalg",
"//third_party/py/numpy",
],
shard_count = 10,
tags = [
"noasan",
"optonly",
],
xla_enable_strict_auto_jit = False, # Fails in XLA b/143610154
)
cuda_py_test(
name = "linear_operator_lower_triangular_test",
size = "medium",
srcs = ["linear_operator_lower_triangular_test.py"],
additional_deps = [
"//tensorflow/python/ops/linalg",
tags = [
"noasan",
"optonly",
],
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:platform_test",
],
tags = [
"noasan",
"optonly",
"//tensorflow/python/ops/linalg",
],
)
@ -298,19 +298,19 @@ cuda_py_test(
name = "linear_operator_low_rank_update_test",
size = "medium",
srcs = ["linear_operator_low_rank_update_test.py"],
additional_deps = [
"//tensorflow/python/ops/linalg",
shard_count = 10,
tags = [
"noasan", # times out
"optonly",
],
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:platform_test",
],
shard_count = 10,
tags = [
"noasan", # times out
"optonly",
"//tensorflow/python/ops/linalg",
],
)
@ -318,8 +318,13 @@ cuda_py_test(
name = "linear_operator_permutation_test",
size = "medium",
srcs = ["linear_operator_permutation_test.py"],
additional_deps = [
"//tensorflow/python/ops/linalg",
shard_count = 5,
tags = [
"noasan",
"optonly",
],
xla_enable_strict_auto_jit = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework",
@ -328,32 +333,27 @@ cuda_py_test(
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:random_ops",
"//tensorflow/python/ops/linalg",
],
shard_count = 5,
tags = [
"noasan",
"optonly",
],
xla_enable_strict_auto_jit = True,
)
cuda_py_test(
name = "linear_operator_util_test",
size = "medium",
srcs = ["linear_operator_util_test.py"],
additional_deps = [
"//tensorflow/python/ops/linalg",
shard_count = 5,
tags = [
"noasan",
"optonly",
],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
],
shard_count = 5,
tags = [
"noasan",
"optonly",
"//tensorflow/python/ops/linalg",
],
)
@ -361,8 +361,14 @@ cuda_py_test(
name = "linear_operator_toeplitz_test",
size = "medium",
srcs = ["linear_operator_toeplitz_test.py"],
additional_deps = [
"//third_party/py/numpy",
shard_count = 5,
tags = [
"no_cuda_on_cpu_tap", # flaky, b/135701551
"no_gpu", # flaky, b/135701551
"noasan", # times out, b/63678675
"optonly", # times out, b/79171797
],
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework",
@ -372,13 +378,7 @@ cuda_py_test(
"//tensorflow/python:platform_test",
"//tensorflow/python/ops/linalg",
"//tensorflow/python/ops/signal",
],
shard_count = 5,
tags = [
"no_cuda_on_cpu_tap", # flaky, b/135701551
"no_gpu", # flaky, b/135701551
"noasan", # times out, b/63678675
"optonly", # times out, b/79171797
"//third_party/py/numpy",
],
)
@ -386,8 +386,9 @@ cuda_py_test(
name = "linear_operator_zeros_test",
size = "medium",
srcs = ["linear_operator_zeros_test.py"],
additional_deps = [
"//tensorflow/python/ops/linalg",
shard_count = 5,
tags = ["optonly"], # Test is flaky without optimization.
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework",
@ -395,7 +396,6 @@ cuda_py_test(
"//tensorflow/python:linalg_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python:random_ops",
"//tensorflow/python/ops/linalg",
],
shard_count = 5,
tags = ["optonly"], # Test is flaky without optimization.
)

38
tensorflow/python/kernel_tests/linalg/sparse/BUILD
View File

@ -11,15 +11,15 @@ cuda_py_test(
name = "conjugate_gradient_test",
size = "medium",
srcs = ["conjugate_gradient_test.py"],
additional_deps = [
"//tensorflow/python/ops/linalg",
"//third_party/py/numpy",
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python:platform_test",
"//tensorflow/python/ops/linalg",
"//third_party/py/numpy",
],
)
@ -27,58 +27,58 @@ cuda_py_test(
name = "csr_sparse_matrix_test",
size = "medium",
srcs = ["csr_sparse_matrix_test.py"],
additional_deps = [
"//tensorflow/python/ops/linalg/sparse",
],
main = "csr_sparse_matrix_test.py",
tags = ["no_rocm"],
deps = [
"//tensorflow/python/ops/linalg/sparse",
],
)
cuda_py_test(
name = "csr_sparse_matrix_ops_test",
size = "medium",
srcs = ["csr_sparse_matrix_ops_test.py"],
additional_deps = [
"//tensorflow/python/ops/linalg/sparse",
"//tensorflow/python/ops/linalg/sparse:gen_sparse_csr_matrix_ops",
],
main = "csr_sparse_matrix_ops_test.py",
shard_count = 10,
tags = ["no_rocm"],
deps = [
"//tensorflow/python/ops/linalg/sparse",
"//tensorflow/python/ops/linalg/sparse:gen_sparse_csr_matrix_ops",
],
)
cuda_py_test(
name = "csr_sparse_matrix_grad_test",
size = "medium",
srcs = ["csr_sparse_matrix_grad_test.py"],
additional_deps = [
"//tensorflow/python/ops/linalg/sparse",
],
main = "csr_sparse_matrix_grad_test.py",
shard_count = 50,
tags = ["no_rocm"],
deps = [
"//tensorflow/python/ops/linalg/sparse",
],
)
cuda_py_test(
name = "csr_sparse_matrix_dense_mat_mul_grad_test",
size = "medium",
srcs = ["csr_sparse_matrix_dense_mat_mul_grad_test.py"],
additional_deps = [
"//tensorflow/python/ops/linalg/sparse",
],
main = "csr_sparse_matrix_dense_mat_mul_grad_test.py",
shard_count = 50,
tags = ["no_rocm"],
deps = [
"//tensorflow/python/ops/linalg/sparse",
],
)
cuda_py_test(
name = "csr_sparse_matrix_sparse_mat_mul_grad_test",
size = "medium",
srcs = ["csr_sparse_matrix_sparse_mat_mul_grad_test.py"],
additional_deps = [
"//tensorflow/python/ops/linalg/sparse",
],
main = "csr_sparse_matrix_sparse_mat_mul_grad_test.py",
shard_count = 50,
tags = ["no_rocm"],
deps = [
"//tensorflow/python/ops/linalg/sparse",
],
)

38
tensorflow/python/kernel_tests/random/BUILD
View File

@ -39,8 +39,7 @@ cuda_py_test(
name = "multinomial_op_test",
size = "small",
srcs = ["multinomial_op_test.py"],
additional_deps = [
"//third_party/py/numpy",
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:array_ops",
"//tensorflow/python:client",
@ -50,6 +49,7 @@ cuda_py_test(
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:math_ops",
"//tensorflow/python:random_ops",
"//third_party/py/numpy",
],
)
@ -57,8 +57,8 @@ cuda_py_test(
name = "multinomial_op_big_test",
size = "medium",
srcs = ["multinomial_op_big_test.py"],
additional_deps = [
"//third_party/py/numpy",
shard_count = 3,
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:array_ops",
"//tensorflow/python:client",
@ -68,18 +68,18 @@ cuda_py_test(
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:math_ops",
"//tensorflow/python:random_ops",
"//third_party/py/numpy",
],
shard_count = 3,
)
cuda_py_test(
name = "random_crop_test",
size = "small",
srcs = ["random_crop_test.py"],
additional_deps = [
"//third_party/py/numpy",
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:random_ops",
"//third_party/py/numpy",
],
)
@ -87,12 +87,12 @@ cuda_py_test(
name = "random_ops_test",
size = "medium",
srcs = ["random_ops_test.py"],
additional_deps = [
"//third_party/py/numpy",
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:random_ops",
"//third_party/py/numpy",
],
)
@ -100,14 +100,14 @@ cuda_py_test(
name = "stateless_random_ops_test",
size = "medium",
srcs = ["stateless_random_ops_test.py"],
additional_deps = [
"//third_party/py/numpy",
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework",
"//tensorflow/python:platform",
"//tensorflow/python:random_ops",
"//tensorflow/python:stateless_random_ops",
"//third_party/py/numpy",
],
)
@ -115,26 +115,25 @@ cuda_py_test(
name = "random_gamma_test",
size = "medium",
srcs = ["random_gamma_test.py"],
additional_deps = [
shard_count = 4,
tags = ["nozapfhahn"],
deps = [
":util",
"//third_party/py/numpy",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:platform",
"//tensorflow/python:random_ops",
"//third_party/py/numpy",
],
shard_count = 4,
tags = ["nozapfhahn"],
)
cuda_py_test(
name = "random_grad_test",
size = "small",
srcs = ["random_grad_test.py"],
additional_deps = [
"//third_party/py/numpy",
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework",
@ -143,6 +142,7 @@ cuda_py_test(
"//tensorflow/python:platform",
"//tensorflow/python:random_grad",
"//tensorflow/python:random_ops",
"//third_party/py/numpy",
],
)
@ -169,14 +169,14 @@ cuda_py_test(
name = "random_poisson_test",
size = "medium",
srcs = ["random_poisson_test.py"],
additional_deps = [
deps = [
":util",
"//third_party/py/numpy",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:platform",
"//tensorflow/python:random_ops",
"//third_party/py/numpy",
],
)

40
tensorflow/python/ops/parallel_for/BUILD
View File

@ -107,7 +107,8 @@ py_library(
cuda_py_test(
name = "control_flow_ops_test",
srcs = ["control_flow_ops_test.py"],
additional_deps = [
tags = ["no_rocm"],
deps = [
":control_flow_ops",
":test_util",
"//tensorflow/core:protos_all_py",
@ -120,21 +121,11 @@ cuda_py_test(
"//tensorflow/python:tensor_array_grad",
"//tensorflow/python:util",
],
tags = ["no_rocm"],
)
cuda_py_test(
name = "xla_control_flow_ops_test",
srcs = ["xla_control_flow_ops_test.py"],
additional_deps = [
":control_flow_ops",
":test_util",
"//tensorflow/python:random_ops",
"//tensorflow/compiler/tf2xla/python:xla",
"//tensorflow/python/compiler/xla:xla",
"//tensorflow/python:array_ops",
"//tensorflow/python:math_ops",
],
tags = [
"no_rocm",
# XLA is not enabled by default on Mac or Windows.
@ -142,16 +133,25 @@ cuda_py_test(
"no_windows",
],
xla_enabled = True,
deps = [
":control_flow_ops",
":test_util",
"//tensorflow/compiler/tf2xla/python:xla",
"//tensorflow/python:array_ops",
"//tensorflow/python:math_ops",
"//tensorflow/python:random_ops",
"//tensorflow/python/compiler/xla",
],
)
cuda_py_test(
name = "array_test",
srcs = ["array_test.py"],
additional_deps = [
deps = [
":control_flow_ops",
":test_util",
"//tensorflow/python:client_testlib",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:random_ops",
"//tensorflow/python:util",
"//tensorflow/python/eager:backprop",
@ -161,7 +161,9 @@ cuda_py_test(
cuda_py_test(
name = "math_test",
srcs = ["math_test.py"],
additional_deps = [
shard_count = 5,
tags = ["optonly"], # Too slow in non-opt mode
deps = [
":control_flow_ops",
":test_util",
"//tensorflow/python:client_testlib",
@ -169,8 +171,6 @@ cuda_py_test(
"//tensorflow/python:random_ops",
"//tensorflow/python:util",
],
shard_count = 5,
tags = ["optonly"], # Too slow in non-opt mode
)
py_library(
@ -188,16 +188,16 @@ py_library(
cuda_py_test(
name = "gradients_test",
srcs = ["gradients_test.py"],
additional_deps = [
tags = ["optonly"], # Too slow in non-opt mode
deps = [
":control_flow_ops",
":gradients",
"//third_party/py/numpy",
"//tensorflow/python:layers",
"//tensorflow/python:client_testlib",
"//tensorflow/python:control_flow_ops",
"//tensorflow/python:functional_ops",
"//tensorflow/python:layers",
"//tensorflow/python:random_ops",
"//tensorflow/python/ops/losses",
"//third_party/py/numpy",
],
tags = ["optonly"], # Too slow in non-opt mode
)

50
tensorflow/python/profiler/BUILD
View File

@ -46,17 +46,6 @@ py_library(
cuda_py_test(
name = "model_analyzer_test",
srcs = ["model_analyzer_test.py"],
additional_deps = [
":profile_context",
":model_analyzer",
"//tensorflow/python/profiler/internal:model_analyzer_testlib",
"//tensorflow/python:client",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:distributed_framework_test_lib",
"//tensorflow/python:platform",
"//tensorflow/python:variables",
],
python_version = "PY3",
tags = [
"no_pip",
@ -64,23 +53,34 @@ cuda_py_test(
"oss_serial",
],
xla_enable_strict_auto_jit = False, # Node names are different with autojit
deps = [
":model_analyzer",
":profile_context",
"//tensorflow/python:client",
"//tensorflow/python:client_testlib",
"//tensorflow/python:distributed_framework_test_lib",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:platform",
"//tensorflow/python:variables",
"//tensorflow/python/profiler/internal:model_analyzer_testlib",
],
)
cuda_py_test(
name = "profiler_test",
srcs = ["profiler_test.py"],
additional_deps = [
python_version = "PY3",
tags = ["no_pip"],
xla_enable_strict_auto_jit = False, # Node names are different with autojit
deps = [
":model_analyzer",
"//tensorflow/python:client",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:platform",
"//tensorflow/python/profiler/internal:model_analyzer_testlib",
"//tensorflow/python:variables",
"//tensorflow/python/profiler/internal:model_analyzer_testlib",
],
python_version = "PY3",
tags = ["no_pip"],
xla_enable_strict_auto_jit = False, # Node names are different with autojit
)
py_library(
@ -126,21 +126,21 @@ py_library(
cuda_py_test(
name = "profile_context_test",
srcs = ["profile_context_test.py"],
additional_deps = [
":profile_context",
"//tensorflow/python:client",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:platform",
"//tensorflow/python/profiler/internal:model_analyzer_testlib",
"//tensorflow/python:variables",
],
python_version = "PY3",
tags = [
"no_gpu", # b/136036359
"no_pip",
],
xla_enable_strict_auto_jit = False, # Node names are different with autojit
deps = [
":profile_context",
"//tensorflow/python:client",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:platform",
"//tensorflow/python:variables",
"//tensorflow/python/profiler/internal:model_analyzer_testlib",
],
)
py_library(

18
tensorflow/python/profiler/internal/BUILD
View File

@ -58,21 +58,21 @@ py_test(
cuda_py_test(
name = "run_metadata_test",
srcs = ["run_metadata_test.py"],
additional_deps = [
":model_analyzer_testlib",
"//tensorflow/core/profiler:protos_all_py",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:math_ops",
"//tensorflow/python/profiler:model_analyzer",
"//tensorflow/python:random_ops",
],
python_version = "PY3",
tags = [
"no_gpu", # b/138442728
"no_pip",
],
xla_enable_strict_auto_jit = False, # Node names are different with autojit
deps = [
":model_analyzer_testlib",
"//tensorflow/core/profiler:protos_all_py",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:math_ops",
"//tensorflow/python:random_ops",
"//tensorflow/python/profiler:model_analyzer",
],
)
pybind_extension(

32
tensorflow/python/saved_model/BUILD
View File

@ -386,27 +386,27 @@ py_library(
cuda_py_test(
name = "load_test",
srcs = ["load_test.py"],
additional_deps = [
":load",
":save",
"@absl_py//absl/testing:parameterized",
"//tensorflow/python:constant_op",
"//tensorflow/python:control_flow_ops",
"//tensorflow/python:cond_v2", # b/118513001
"//tensorflow/python:while_v2", # b/118513001
"//tensorflow/python:dtypes",
"//tensorflow/python:lib",
"//tensorflow/python:tensor_spec",
"//tensorflow/python/eager:def_function",
"//tensorflow/python/eager:test",
"//tensorflow/python/module",
"//tensorflow/python/training/tracking:tracking",
],
shard_count = 10,
tags = [
"no_gpu", # TODO(b/136560979): flaky
"no_mac", # TODO(b/124822121): Re-enable this test.
],
deps = [
":load",
":save",
"//tensorflow/python:cond_v2", # b/118513001
"//tensorflow/python:constant_op",
"//tensorflow/python:control_flow_ops",
"//tensorflow/python:dtypes",
"//tensorflow/python:lib",
"//tensorflow/python:tensor_spec",
"//tensorflow/python:while_v2", # b/118513001
"//tensorflow/python/eager:def_function",
"//tensorflow/python/eager:test",
"//tensorflow/python/module",
"//tensorflow/python/training/tracking",
"@absl_py//absl/testing:parameterized",
],
)
tf_py_test(

2
tensorflow/python/training/saving/BUILD
View File

@ -30,7 +30,7 @@ cuda_py_test(
srcs = [
"functional_saver_test.py",
],
additional_deps = [
deps = [
":functional_saver",
":saveable_hook",
"//tensorflow/python/eager:test",