Move additional_deps to deps.
PiperOrigin-RevId: 285469994 Change-Id: I4fc2308752d000859a5ebb486917a96967e43d65
This commit is contained in:
parent
913e50445b
commit
b1bc83537f
@ -33,7 +33,7 @@ def distribute_py_test(
|
||||
srcs = srcs,
|
||||
data = data,
|
||||
main = main,
|
||||
additional_deps = deps,
|
||||
deps = deps,
|
||||
shard_count = shard_count,
|
||||
tags = tags,
|
||||
args = args,
|
||||
|
@ -134,11 +134,6 @@ cuda_py_tests(
|
||||
"test/vgg_block_nchw_test.py",
|
||||
"test/vgg_block_test.py",
|
||||
],
|
||||
additional_deps = [
|
||||
":tf_trt_integration_test_base",
|
||||
"//tensorflow/python:client_testlib",
|
||||
"//tensorflow/python:framework_test_lib",
|
||||
],
|
||||
python_version = "PY3",
|
||||
tags = [
|
||||
"no_cuda_on_cpu_tap",
|
||||
@ -146,6 +141,11 @@ cuda_py_tests(
|
||||
"no_windows",
|
||||
"nomac",
|
||||
],
|
||||
deps = [
|
||||
":tf_trt_integration_test_base",
|
||||
"//tensorflow/python:client_testlib",
|
||||
"//tensorflow/python:framework_test_lib",
|
||||
],
|
||||
)
|
||||
|
||||
cuda_py_tests(
|
||||
@ -154,11 +154,6 @@ cuda_py_tests(
|
||||
"test/biasadd_matmul_test.py",
|
||||
"test/concatenation_test.py",
|
||||
],
|
||||
additional_deps = [
|
||||
":tf_trt_integration_test_base",
|
||||
"//tensorflow/python:client_testlib",
|
||||
"//tensorflow/python:framework_test_lib",
|
||||
],
|
||||
python_version = "PY3",
|
||||
tags = [
|
||||
"no_rocm",
|
||||
@ -166,6 +161,11 @@ cuda_py_tests(
|
||||
"nomac",
|
||||
"notap", # b/140261407
|
||||
],
|
||||
deps = [
|
||||
":tf_trt_integration_test_base",
|
||||
"//tensorflow/python:client_testlib",
|
||||
"//tensorflow/python:framework_test_lib",
|
||||
],
|
||||
)
|
||||
|
||||
cuda_py_test(
|
||||
|
@ -1532,11 +1532,11 @@ tf_py_test(
|
||||
name = "compile_utils_test",
|
||||
size = "medium",
|
||||
srcs = ["engine/compile_utils_test.py"],
|
||||
additional_deps = [
|
||||
deps = [
|
||||
":keras",
|
||||
"@absl_py//absl/testing:parameterized",
|
||||
"//third_party/py/numpy",
|
||||
"//tensorflow/python:client_testlib",
|
||||
"//third_party/py/numpy",
|
||||
"@absl_py//absl/testing:parameterized",
|
||||
],
|
||||
)
|
||||
|
||||
|
@ -53,10 +53,10 @@ tf_py_test(
|
||||
name = "imagenet_utils_test",
|
||||
size = "medium",
|
||||
srcs = ["imagenet_utils_test.py"],
|
||||
additional_deps = [
|
||||
":applications",
|
||||
"@absl_py//absl/testing:parameterized",
|
||||
"//tensorflow/python:client_testlib",
|
||||
],
|
||||
shard_count = 2,
|
||||
deps = [
|
||||
":applications",
|
||||
"//tensorflow/python:client_testlib",
|
||||
"@absl_py//absl/testing:parameterized",
|
||||
],
|
||||
)
|
||||
|
@ -3539,13 +3539,13 @@ sycl_py_test(
|
||||
name = "basic_gpu_test",
|
||||
size = "small",
|
||||
srcs = ["basic_gpu_test.py"],
|
||||
additional_deps = [
|
||||
"//third_party/py/numpy",
|
||||
deps = [
|
||||
"//tensorflow/python:array_ops_gen",
|
||||
"//tensorflow/python:client_testlib",
|
||||
"//tensorflow/python:framework_for_generated_wrappers",
|
||||
"//tensorflow/python:math_ops",
|
||||
"//tensorflow/python:math_ops_gen",
|
||||
"//third_party/py/numpy",
|
||||
],
|
||||
)
|
||||
|
||||
|
@ -24,142 +24,142 @@ py_library(
|
||||
cuda_py_tests(
|
||||
name = "dct_ops_test",
|
||||
srcs = ["dct_ops_test.py"],
|
||||
additional_deps = [
|
||||
"@absl_py//absl/testing:parameterized",
|
||||
"//third_party/py/numpy",
|
||||
python_version = "PY3",
|
||||
tags = ["no_rocm"],
|
||||
deps = [
|
||||
"//tensorflow/python:client_testlib",
|
||||
"//tensorflow/python:framework_for_generated_wrappers",
|
||||
"//tensorflow/python/ops/signal",
|
||||
"//third_party/py/numpy",
|
||||
"@absl_py//absl/testing:parameterized",
|
||||
],
|
||||
python_version = "PY3",
|
||||
tags = ["no_rocm"],
|
||||
)
|
||||
|
||||
cuda_py_tests(
|
||||
name = "fft_ops_test",
|
||||
size = "medium",
|
||||
srcs = ["fft_ops_test.py"],
|
||||
additional_deps = [
|
||||
"//third_party/py/numpy",
|
||||
"//tensorflow/python:client_testlib",
|
||||
"//tensorflow/python:framework_for_generated_wrappers",
|
||||
"//tensorflow/python:math_ops",
|
||||
"//tensorflow/python/ops/signal",
|
||||
],
|
||||
python_version = "PY3",
|
||||
shard_count = 8,
|
||||
tags = [
|
||||
"no_rocm",
|
||||
"optonly",
|
||||
],
|
||||
deps = [
|
||||
"//tensorflow/python:client_testlib",
|
||||
"//tensorflow/python:framework_for_generated_wrappers",
|
||||
"//tensorflow/python:math_ops",
|
||||
"//tensorflow/python/ops/signal",
|
||||
"//third_party/py/numpy",
|
||||
],
|
||||
)
|
||||
|
||||
cuda_py_tests(
|
||||
name = "mel_ops_test",
|
||||
srcs = ["mel_ops_test.py"],
|
||||
additional_deps = [
|
||||
python_version = "PY3",
|
||||
deps = [
|
||||
":test_util",
|
||||
"//third_party/py/numpy",
|
||||
"//tensorflow/python:client_testlib",
|
||||
"//tensorflow/python/ops/signal",
|
||||
"//third_party/py/numpy",
|
||||
],
|
||||
python_version = "PY3",
|
||||
)
|
||||
|
||||
cuda_py_tests(
|
||||
name = "mfcc_ops_test",
|
||||
srcs = ["mfcc_ops_test.py"],
|
||||
additional_deps = [
|
||||
"@absl_py//absl/testing:parameterized",
|
||||
"//third_party/py/numpy",
|
||||
python_version = "PY3",
|
||||
tags = ["no_rocm"],
|
||||
deps = [
|
||||
"//tensorflow/python:client_testlib",
|
||||
"//tensorflow/python:framework",
|
||||
"//tensorflow/python:framework_for_generated_wrappers",
|
||||
"//tensorflow/python:framework_test_lib",
|
||||
"//tensorflow/python/ops/signal",
|
||||
"//third_party/py/numpy",
|
||||
"@absl_py//absl/testing:parameterized",
|
||||
],
|
||||
python_version = "PY3",
|
||||
tags = ["no_rocm"],
|
||||
)
|
||||
|
||||
cuda_py_tests(
|
||||
name = "reconstruction_ops_test",
|
||||
srcs = ["reconstruction_ops_test.py"],
|
||||
additional_deps = [
|
||||
"@absl_py//absl/testing:parameterized",
|
||||
"//third_party/py/numpy",
|
||||
python_version = "PY3",
|
||||
deps = [
|
||||
"//tensorflow/python:array_ops",
|
||||
"//tensorflow/python:gradients",
|
||||
"//tensorflow/python:math_ops",
|
||||
"//tensorflow/python:client_testlib",
|
||||
"//tensorflow/python:framework",
|
||||
"//tensorflow/python:framework_for_generated_wrappers",
|
||||
"//tensorflow/python:framework_test_lib",
|
||||
"//tensorflow/python/ops/signal",
|
||||
"//tensorflow/python:gradients",
|
||||
"//tensorflow/python:math_ops",
|
||||
"//tensorflow/python:platform_test",
|
||||
"//tensorflow/python/ops/signal",
|
||||
"//third_party/py/numpy",
|
||||
"@absl_py//absl/testing:parameterized",
|
||||
],
|
||||
python_version = "PY3",
|
||||
)
|
||||
|
||||
cuda_py_tests(
|
||||
name = "shape_ops_test",
|
||||
srcs = ["shape_ops_test.py"],
|
||||
additional_deps = [
|
||||
python_version = "PY3",
|
||||
deps = [
|
||||
":test_util",
|
||||
"//third_party/py/numpy",
|
||||
"//tensorflow/python:array_ops",
|
||||
"//tensorflow/python:math_ops",
|
||||
"//tensorflow/python:client_testlib",
|
||||
"//tensorflow/python:framework",
|
||||
"//tensorflow/python:framework_for_generated_wrappers",
|
||||
"//tensorflow/python:framework_test_lib",
|
||||
"//tensorflow/python/ops/signal",
|
||||
"//tensorflow/python:math_ops",
|
||||
"//tensorflow/python:platform_test",
|
||||
"//tensorflow/python/ops/signal",
|
||||
"//third_party/py/numpy",
|
||||
],
|
||||
python_version = "PY3",
|
||||
)
|
||||
|
||||
cuda_py_tests(
|
||||
name = "spectral_ops_test",
|
||||
size = "large",
|
||||
srcs = ["spectral_ops_test.py"],
|
||||
additional_deps = [
|
||||
"@absl_py//absl/testing:parameterized",
|
||||
"//third_party/py/numpy",
|
||||
"//tensorflow/python:array_ops",
|
||||
"//tensorflow/python:gradients",
|
||||
"//tensorflow/python:math_ops",
|
||||
"//tensorflow/python:random_ops",
|
||||
"//tensorflow/python:client_testlib",
|
||||
"//tensorflow/python:framework",
|
||||
"//tensorflow/python:framework_for_generated_wrappers",
|
||||
"//tensorflow/python:framework_test_lib",
|
||||
"//tensorflow/python:platform_test",
|
||||
"//tensorflow/python/ops/signal",
|
||||
],
|
||||
python_version = "PY3",
|
||||
tags = [
|
||||
"no_rocm",
|
||||
"nomac",
|
||||
],
|
||||
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:gradients",
|
||||
"//tensorflow/python:math_ops",
|
||||
"//tensorflow/python:platform_test",
|
||||
"//tensorflow/python:random_ops",
|
||||
"//tensorflow/python/ops/signal",
|
||||
"//third_party/py/numpy",
|
||||
"@absl_py//absl/testing:parameterized",
|
||||
],
|
||||
)
|
||||
|
||||
cuda_py_tests(
|
||||
name = "window_ops_test",
|
||||
srcs = ["window_ops_test.py"],
|
||||
additional_deps = [
|
||||
":test_util",
|
||||
"//third_party/py/numpy",
|
||||
"//tensorflow/python:client_testlib",
|
||||
"//tensorflow/python:framework",
|
||||
"//tensorflow/python:framework_for_generated_wrappers",
|
||||
"//tensorflow/python:framework_test_lib",
|
||||
"//tensorflow/python/ops/signal",
|
||||
"//tensorflow/python:platform_test",
|
||||
],
|
||||
python_version = "PY3",
|
||||
shard_count = 4,
|
||||
tags = [
|
||||
"no_windows_gpu",
|
||||
],
|
||||
deps = [
|
||||
":test_util",
|
||||
"//tensorflow/python:client_testlib",
|
||||
"//tensorflow/python:framework",
|
||||
"//tensorflow/python:framework_for_generated_wrappers",
|
||||
"//tensorflow/python:framework_test_lib",
|
||||
"//tensorflow/python:platform_test",
|
||||
"//tensorflow/python/ops/signal",
|
||||
"//third_party/py/numpy",
|
||||
],
|
||||
)
|
||||
|
@ -87,7 +87,7 @@ py_library(
|
||||
#cuda_py_test(
|
||||
# name = "run_and_gather_logs_test",
|
||||
# srcs = ["run_and_gather_logs.py"],
|
||||
# additional_deps = [
|
||||
# deps = [
|
||||
# ":run_and_gather_logs",
|
||||
# ],
|
||||
# args = [
|
||||
|
@ -39,7 +39,7 @@ def tf_cc_logged_benchmark(
|
||||
target,
|
||||
],
|
||||
main = "run_and_gather_logs.py",
|
||||
additional_deps = [
|
||||
deps = [
|
||||
"//tensorflow/tools/test:run_and_gather_logs",
|
||||
],
|
||||
**kwargs
|
||||
|
Loading…
Reference in New Issue
Block a user