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

Move additional_deps to deps for tf_py_test.

Browse Source 
PiperOrigin-RevId: 285307780
Change-Id: Id556a146ba01afa1138454153ae6568e073ace41
This commit is contained in:
Brian Atkinson 2019-12-12 17:34:41 -08:00 committed by TensorFlower Gardener
parent 2d9c178f4e
commit 4ca6bb5c34
26 changed files with 1510 additions and 1510 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

58
tensorflow/examples/speech_commands/BUILD
View File

@ -69,13 +69,13 @@ tf_py_test(
name = "models_test",
size = "small",
srcs = ["models_test.py"],
additional_deps = [
":models",
"//tensorflow/python:client_testlib",
],
tags = [
"no_pip", # b/131330719
],
deps = [
":models",
"//tensorflow/python:client_testlib",
],
)
py_library(
@ -96,15 +96,15 @@ tf_py_test(
name = "input_data_test",
size = "small",
srcs = ["input_data_test.py"],
additional_deps = [
":input_data",
":models",
"//tensorflow/python:client_testlib",
],
tags = [
"no_pip", # b/131330719
"v1only", # uses contrib
],
deps = [
":input_data",
":models",
"//tensorflow/python:client_testlib",
],
)
py_binary(
@ -134,14 +134,14 @@ tf_py_test(
name = "train_test",
size = "small",
srcs = ["train_test.py"],
additional_deps = [
":train_main_lib",
"//tensorflow/python:client_testlib",
],
tags = [
"no_pip", # b/131330719
"v1only", # uses contrib
],
deps = [
":train_main_lib",
"//tensorflow/python:client_testlib",
],
)
py_binary(
@ -181,14 +181,14 @@ tf_py_test(
name = "freeze_test",
size = "small",
srcs = ["freeze_test.py"],
additional_deps = [
":freeze_main_lib",
"//tensorflow/python:client_testlib",
],
tags = [
"no_pip", # b/131330719
"v1only", # uses contrib
],
deps = [
":freeze_main_lib",
"//tensorflow/python:client_testlib",
],
)
py_binary(
@ -225,14 +225,14 @@ tf_py_test(
name = "wav_to_features_test",
size = "small",
srcs = ["wav_to_features_test.py"],
additional_deps = [
":wav_to_features_main_lib",
"//tensorflow/python:client_testlib",
],
tags = [
"no_pip", # b/131330719
"v1only", # uses contrib
],
deps = [
":wav_to_features_main_lib",
"//tensorflow/python:client_testlib",
],
)
py_binary(
@ -269,14 +269,14 @@ tf_py_test(
name = "generate_streaming_test_wav_test",
size = "small",
srcs = ["generate_streaming_test_wav_test.py"],
additional_deps = [
":generate_streaming_test_wav_main_lib",
"//tensorflow/python:client_testlib",
],
tags = [
"no_pip", # b/131330719
"v1only", # uses contrib
],
deps = [
":generate_streaming_test_wav_main_lib",
"//tensorflow/python:client_testlib",
],
)
tf_cc_binary(
@ -324,14 +324,14 @@ tf_py_test(
name = "label_wav_test",
size = "medium",
srcs = ["label_wav_test.py"],
additional_deps = [
":label_wav_main_lib",
"//tensorflow/python:client_testlib",
],
tags = [
"no_pip", # b/131330719
"v1only", # uses contrib
],
deps = [
":label_wav_main_lib",
"//tensorflow/python:client_testlib",
],
)
cc_library(

18
tensorflow/examples/tutorials/mnist/BUILD
View File

@ -98,17 +98,17 @@ tf_py_test(
srcs = [
"fully_connected_feed.py",
],
additional_deps = [
":input_data",
":mnist",
"//tensorflow:tensorflow_py",
],
args = [
"--fake_data",
"--max_steps=10",
],
main = "fully_connected_feed.py",
tags = ["no_pip"],
deps = [
":input_data",
":mnist",
"//tensorflow:tensorflow_py",
],
)
tf_py_test(
@ -117,10 +117,6 @@ tf_py_test(
srcs = [
"mnist_with_summaries.py",
],
additional_deps = [
":input_data",
"//tensorflow:tensorflow_py",
],
args = [
"--fake_data",
"--max_steps=10",
@ -132,4 +128,8 @@ tf_py_test(
"noasan", # http://b/146080738
"notsan", # http://b/29184009
],
deps = [
":input_data",
"//tensorflow:tensorflow_py",
],
)

4
tensorflow/lite/experimental/microfrontend/BUILD
View File

@ -92,9 +92,9 @@ tf_py_test(
name = "audio_microfrontend_op_test",
size = "small",
srcs = ["python/kernel_tests/audio_microfrontend_op_test.py"],
additional_deps = [
tags = ["no_pip"],
deps = [
":audio_microfrontend_py",
"//tensorflow:tensorflow_py",
],
tags = ["no_pip"],
)

10
tensorflow/lite/toco/python/BUILD
View File

@ -89,14 +89,14 @@ py_binary(
tf_py_test(
name = "toco_from_protos_test",
srcs = ["toco_from_protos_test.py"],
additional_deps = [
"//tensorflow:tensorflow_py",
"//tensorflow/lite/toco:model_flags_proto_py",
"//tensorflow/lite/toco:toco_flags_proto_py",
],
python_version = "PY3",
tags = [
"no_oss",
"no_pip",
],
deps = [
"//tensorflow:tensorflow_py",
"//tensorflow/lite/toco:model_flags_proto_py",
"//tensorflow/lite/toco:toco_flags_proto_py",
],
)

8
tensorflow/python/compat/BUILD
View File

@ -32,20 +32,20 @@ tf_py_test(
name = "compat_test",
size = "small",
srcs = ["compat_test.py"],
additional_deps = [
tags = ["nofwdcompat"],
deps = [
":compat",
"//tensorflow/python:client_testlib",
],
tags = ["nofwdcompat"],
)
tf_py_test(
name = "disable_v2_behavior_test",
size = "small",
srcs = ["disable_v2_behavior_test.py"],
additional_deps = [
deps = [
":v2_compat",
"//tensorflow/python:framework",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework",
],
)

22
tensorflow/python/data/benchmarks/BUILD
View File

@ -10,12 +10,12 @@ exports_files(["LICENSE"])
tf_py_test(
name = "meta_benchmark",
srcs = ["meta_benchmark.py"],
additional_deps = [
"//third_party/py/numpy",
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:session",
"//tensorflow/python/data/experimental/ops:testing",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
],
)
@ -34,18 +34,18 @@ py_library(
tf_py_test(
name = "batch_benchmark",
srcs = ["batch_benchmark.py"],
additional_deps = [
deps = [
":benchmark_base",
"//third_party/py/numpy",
"//tensorflow/python:sparse_tensor",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
],
)
tf_py_test(
name = "filter_benchmark",
srcs = ["filter_benchmark.py"],
additional_deps = [
deps = [
":benchmark_base",
"//tensorflow/python/data/ops:dataset_ops",
],
@ -54,31 +54,31 @@ tf_py_test(
tf_py_test(
name = "from_tensor_slices_benchmark",
srcs = ["from_tensor_slices_benchmark.py"],
additional_deps = [
deps = [
":benchmark_base",
"//third_party/py/numpy",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
],
)
tf_py_test(
name = "list_files_benchmark",
srcs = ["list_files_benchmark.py"],
additional_deps = [
deps = [
":benchmark_base",
"//third_party/py/numpy",
"//tensorflow/python:client_testlib",
"//tensorflow/python:errors",
"//tensorflow/python:framework_ops",
"//tensorflow/python:session",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
],
)
tf_py_test(
name = "map_benchmark",
srcs = ["map_benchmark.py"],
additional_deps = [
deps = [
":benchmark_base",
"//tensorflow/python/data/ops:dataset_ops",
],
@ -87,7 +87,7 @@ tf_py_test(
tf_py_test(
name = "range_benchmark",
srcs = ["range_benchmark.py"],
additional_deps = [
deps = [
":benchmark_base",
"//tensorflow/python/data/ops:dataset_ops",
],

56
tensorflow/python/data/experimental/benchmarks/BUILD
View File

@ -15,33 +15,32 @@ exports_files(
tf_py_test(
name = "autotune_benchmark",
srcs = ["autotune_benchmark.py"],
additional_deps = [
"//third_party/py/numpy",
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:math_ops",
"//tensorflow/python:session",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
],
)
tf_py_test(
name = "choose_fastest_benchmark",
srcs = ["choose_fastest_benchmark.py"],
additional_deps = [
"//third_party/py/numpy",
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_ops",
"//tensorflow/python:math_ops",
"//tensorflow/python:session",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
],
)
tf_py_test(
name = "choose_fastest_branch_benchmark",
srcs = ["choose_fastest_branch_benchmark.py"],
additional_deps = [
"//third_party/py/numpy",
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_ops",
"//tensorflow/python:math_ops",
@ -49,14 +48,15 @@ tf_py_test(
"//tensorflow/python/data/benchmarks:benchmark_base",
"//tensorflow/python/data/experimental/ops:testing",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
],
)
tf_py_test(
name = "csv_dataset_benchmark",
srcs = ["csv_dataset_benchmark.py"],
additional_deps = [
"//third_party/py/numpy",
tags = ["no_pip"],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:parsing_ops",
"//tensorflow/python:platform",
@ -64,15 +64,14 @@ tf_py_test(
"//tensorflow/python:session",
"//tensorflow/python/data/experimental/ops:readers",
"//tensorflow/python/data/ops:readers",
"//third_party/py/numpy",
],
tags = ["no_pip"],
)
tf_py_test(
name = "map_and_batch_benchmark",
srcs = ["map_and_batch_benchmark.py"],
additional_deps = [
"//third_party/py/numpy",
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
@ -83,13 +82,14 @@ tf_py_test(
"//tensorflow/python:session",
"//tensorflow/python/data/experimental/ops:batching",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
],
)
tf_py_test(
name = "map_defun_benchmark",
srcs = ["map_defun_benchmark.py"],
additional_deps = [
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
@ -104,8 +104,7 @@ tf_py_test(
tf_py_test(
name = "map_vectorization_benchmark",
srcs = ["map_vectorization_benchmark.py"],
additional_deps = [
"//third_party/py/numpy",
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
@ -116,6 +115,7 @@ tf_py_test(
"//tensorflow/python:session",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/util:nest",
"//third_party/py/numpy",
],
)
@ -123,8 +123,7 @@ tf_py_test(
name = "matching_files_benchmark",
size = "small",
srcs = ["matching_files_benchmark.py"],
additional_deps = [
"//third_party/py/numpy",
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
@ -132,54 +131,54 @@ tf_py_test(
"//tensorflow/python:util",
"//tensorflow/python/data/experimental/ops:matching_files",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
],
)
tf_py_test(
name = "optimize_benchmark",
srcs = ["optimize_benchmark.py"],
additional_deps = [
"//third_party/py/numpy",
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_ops",
"//tensorflow/python:math_ops",
"//tensorflow/python:session",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
],
)
tf_py_test(
name = "parallel_interleave_benchmark",
srcs = ["parallel_interleave_benchmark.py"],
additional_deps = [
"//third_party/py/numpy",
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:math_ops",
"//tensorflow/python:session",
"//tensorflow/python/data/experimental/ops:interleave_ops",
"//tensorflow/python/data/experimental/ops:testing",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
],
)
tf_py_test(
name = "rejection_resample_benchmark",
srcs = ["rejection_resample_benchmark.py"],
additional_deps = [
"//third_party/py/numpy",
"@six_archive//:six",
tags = ["no_pip"],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python/data/experimental/ops:resampling",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
"@six_archive//:six",
],
tags = ["no_pip"],
)
tf_py_test(
name = "snapshot_dataset_benchmark",
srcs = ["snapshot_dataset_benchmark.py"],
additional_deps = [
"//third_party/py/numpy",
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:errors",
@ -190,14 +189,14 @@ tf_py_test(
"//tensorflow/python/data/experimental/ops:snapshot",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
],
)
tf_py_test(
name = "unbatch_benchmark",
srcs = ["unbatch_benchmark.py"],
additional_deps = [
"//third_party/py/numpy",
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
@ -205,5 +204,6 @@ tf_py_test(
"//tensorflow/python:session",
"//tensorflow/python/data/experimental/ops:batching",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
],
)

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

@ -12,27 +12,29 @@ tf_py_test(
name = "assert_next_test",
size = "small",
srcs = ["assert_next_test.py"],
additional_deps = [
tags = [
"no_oss",
"no_pip",
"no_windows",
],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:errors",
"//tensorflow/python/data/experimental/ops:testing",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
],
tags = [
"no_oss",
"no_pip",
"no_windows",
],
)
tf_py_test(
name = "auto_shard_dataset_test",
size = "medium",
srcs = ["auto_shard_dataset_test.py"],
additional_deps = [
tags = [
"no_pip",
],
deps = [
":reader_dataset_ops_test_base",
"@absl_py//absl/testing:parameterized",
"//tensorflow/python:client_testlib",
"//tensorflow/python/data/experimental/ops:distribute",
"//tensorflow/python/data/experimental/ops:readers",
@ -40,9 +42,7 @@ tf_py_test(
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/util:nest",
],
tags = [
"no_pip",
"@absl_py//absl/testing:parameterized",
],
)
@ -50,9 +50,7 @@ tf_py_test(
name = "bucket_by_sequence_length_test",
size = "medium",
srcs = ["bucket_by_sequence_length_test.py"],
additional_deps = [
"@absl_py//absl/testing:parameterized",
"//third_party/py/numpy",
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
@ -62,17 +60,19 @@ tf_py_test(
"//tensorflow/python/data/experimental/ops:grouping",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)
tf_py_test(
name = "cardinality_test",
srcs = ["cardinality_test.py"],
additional_deps = [
"@absl_py//absl/testing:parameterized",
deps = [
"//tensorflow/python/data/experimental/ops:cardinality",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
"@absl_py//absl/testing:parameterized",
],
)
@ -100,7 +100,7 @@ tf_py_test(
name = "counter_test",
size = "small",
srcs = ["counter_test.py"],
additional_deps = [
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
"//tensorflow/python/data/experimental/ops:counter",
@ -112,7 +112,8 @@ tf_py_test(
name = "csv_dataset_test",
size = "medium",
srcs = ["csv_dataset_test.py"],
additional_deps = [
tags = ["no_pip"],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
"//tensorflow/python:dtypes",
@ -125,14 +126,12 @@ tf_py_test(
"//tensorflow/python/data/ops:readers",
"//tensorflow/python/eager:context",
],
tags = ["no_pip"],
)
tf_py_test(
name = "dense_to_sparse_batch_test",
srcs = ["dense_to_sparse_batch_test.py"],
additional_deps = [
"//third_party/py/numpy",
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
@ -140,20 +139,21 @@ tf_py_test(
"//tensorflow/python/data/experimental/ops:batching",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
],
)
tf_py_test(
name = "dense_to_ragged_batch_test",
srcs = ["dense_to_ragged_batch_test.py"],
additional_deps = [
"//third_party/py/numpy",
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
"//tensorflow/python:errors",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
],
)
@ -161,14 +161,14 @@ tf_py_test(
name = "directed_interleave_dataset_test",
size = "medium",
srcs = ["directed_interleave_dataset_test.py"],
additional_deps = [
"//third_party/py/numpy",
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:errors",
"//tensorflow/python:random_seed",
"//tensorflow/python/data/experimental/ops:interleave_ops",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
],
)
@ -176,8 +176,7 @@ tf_py_test(
name = "get_single_element_test",
size = "small",
srcs = ["get_single_element_test.py"],
additional_deps = [
"@absl_py//absl/testing:parameterized",
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
@ -188,6 +187,7 @@ tf_py_test(
"//tensorflow/python/data/experimental/ops:grouping",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
"@absl_py//absl/testing:parameterized",
],
)
@ -195,8 +195,7 @@ tf_py_test(
name = "group_by_reducer_test",
size = "medium",
srcs = ["group_by_reducer_test.py"],
additional_deps = [
"//third_party/py/numpy",
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
@ -208,6 +207,7 @@ tf_py_test(
"//tensorflow/python/data/experimental/ops:grouping",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
],
)
@ -215,8 +215,7 @@ tf_py_test(
name = "group_by_window_test",
size = "medium",
srcs = ["group_by_window_test.py"],
additional_deps = [
"//third_party/py/numpy",
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
@ -229,14 +228,14 @@ tf_py_test(
"//tensorflow/python/data/experimental/ops:grouping",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
],
)
tf_py_test(
name = "ignore_errors_test",
srcs = ["ignore_errors_test.py"],
additional_deps = [
"//third_party/py/numpy",
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:errors",
@ -245,6 +244,7 @@ tf_py_test(
"//tensorflow/python/data/experimental/ops:error_ops",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
],
)
@ -252,9 +252,9 @@ tf_py_test(
name = "make_batched_features_dataset_test",
size = "medium",
srcs = ["make_batched_features_dataset_test.py"],
additional_deps = [
tags = ["no_pip"],
deps = [
":reader_dataset_ops_test_base",
"//third_party/py/numpy",
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
"//tensorflow/python:errors",
@ -264,16 +264,16 @@ tf_py_test(
"//tensorflow/python/data/experimental/ops:readers",
"//tensorflow/python/data/ops:readers",
"//tensorflow/python/data/util:nest",
"//third_party/py/numpy",
],
tags = ["no_pip"],
)
tf_py_test(
name = "make_csv_dataset_test",
size = "medium",
srcs = ["make_csv_dataset_test.py"],
additional_deps = [
"//third_party/py/numpy",
tags = ["no_pip"],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
"//tensorflow/python:dtypes",
@ -282,15 +282,16 @@ tf_py_test(
"//tensorflow/python/data/experimental/ops:readers",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/util:nest",
"//third_party/py/numpy",
],
tags = ["no_pip"],
)
tf_py_test(
name = "make_tf_record_dataset_test",
size = "medium",
srcs = ["make_tf_record_dataset_test.py"],
additional_deps = [
tags = ["no_pip"],
deps = [
":reader_dataset_ops_test_base",
"//tensorflow/python:client_testlib",
"//tensorflow/python:errors",
@ -299,16 +300,13 @@ tf_py_test(
"//tensorflow/python/data/experimental/ops:readers",
"//tensorflow/python/data/util:nest",
],
tags = ["no_pip"],
)
tf_py_test(
name = "map_and_batch_test",
size = "medium",
srcs = ["map_and_batch_test.py"],
additional_deps = [
"@absl_py//absl/testing:parameterized",
"//third_party/py/numpy",
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:cond_v2",
@ -321,6 +319,8 @@ tf_py_test(
"//tensorflow/python/data/experimental/ops:batching",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)
@ -328,7 +328,8 @@ tf_py_test(
name = "map_defun_op_test",
size = "small",
srcs = ["map_defun_op_test.py"],
additional_deps = [
tags = ["no_pip"],
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:check_ops",
"//tensorflow/python:client_testlib",
@ -345,15 +346,14 @@ tf_py_test(
"//tensorflow/python/data/experimental/ops:map_defun",
"//tensorflow/python/data/kernel_tests:test_base",
],
tags = ["no_pip"],
)
tf_py_test(
name = "matching_files_test",
size = "small",
srcs = ["matching_files_test.py"],
additional_deps = [
"//third_party/py/numpy",
tags = ["no_pip"],
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
@ -362,26 +362,26 @@ tf_py_test(
"//tensorflow/python/data/experimental/ops:matching_files",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
],
tags = ["no_pip"],
)
tf_py_test(
name = "model_dataset_test",
size = "small",
srcs = ["model_dataset_test.py"],
additional_deps = [
"@absl_py//absl/testing:parameterized",
tags = [
"no_oss",
"no_pip",
"no_windows",
],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:errors",
"//tensorflow/python/data/experimental/ops:testing",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
],
tags = [
"no_oss",
"no_pip",
"no_windows",
"@absl_py//absl/testing:parameterized",
],
)
@ -389,27 +389,30 @@ tf_py_test(
name = "non_serializable_test",
size = "small",
srcs = ["non_serializable_test.py"],
additional_deps = [
tags = [
"no_oss",
"no_pip",
"no_windows",
],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:errors",
"//tensorflow/python/data/experimental/ops:testing",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
],
tags = [
"no_oss",
"no_pip",
"no_windows",
],
)
tf_py_test(
name = "optimize_dataset_test",
size = "medium",
srcs = ["optimize_dataset_test.py"],
additional_deps = [
"@absl_py//absl/testing:parameterized",
"//third_party/py/numpy",
tags = [
"no_oss",
"no_pip",
"no_windows",
],
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
@ -424,11 +427,8 @@ tf_py_test(
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/eager:context",
],
tags = [
"no_oss",
"no_pip",
"no_windows",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)
@ -436,9 +436,8 @@ tf_py_test(
name = "override_threadpool_test",
size = "small",
srcs = ["override_threadpool_test.py"],
additional_deps = [
"@absl_py//absl/testing:parameterized",
"//third_party/py/numpy",
tags = ["no_pip"],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
"//tensorflow/python:errors",
@ -447,16 +446,17 @@ tf_py_test(
"//tensorflow/python/data/experimental/ops:unique",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
tags = ["no_pip"],
)
tf_py_test(
name = "parallel_interleave_test",
size = "medium",
srcs = ["parallel_interleave_test.py"],
additional_deps = [
"@six_archive//:six",
tags = ["no_pip"],
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
@ -468,16 +468,15 @@ tf_py_test(
"//tensorflow/python/data/experimental/ops:interleave_ops",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
"@six_archive//:six",
],
tags = ["no_pip"],
)
tf_py_test(
name = "parse_example_dataset_test",
size = "small",
srcs = ["parse_example_dataset_test.py"],
additional_deps = [
"//third_party/py/numpy",
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
@ -490,6 +489,7 @@ tf_py_test(
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/util:nest",
"//third_party/py/numpy",
],
)
@ -514,8 +514,7 @@ tf_py_test(
name = "prefetch_with_slack_test",
size = "small",
srcs = ["prefetch_with_slack_test.py"],
additional_deps = [
"@absl_py//absl/testing:parameterized",
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:client_testlib",
"//tensorflow/python:errors",
@ -524,6 +523,7 @@ tf_py_test(
"//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",
],
)
@ -553,8 +553,7 @@ tf_py_test(
name = "rebatch_dataset_test",
size = "small",
srcs = ["rebatch_dataset_test.py"],
additional_deps = [
"@absl_py//absl/testing:parameterized",
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:client_testlib",
"//tensorflow/python:parsing_ops",
@ -563,6 +562,7 @@ tf_py_test(
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/util:nest",
"//tensorflow/python/ops/ragged:ragged_tensor",
"@absl_py//absl/testing:parameterized",
],
)
@ -570,10 +570,12 @@ tf_py_test(
name = "rejection_resample_test",
size = "medium",
srcs = ["rejection_resample_test.py"],
additional_deps = [
"@absl_py//absl/testing:parameterized",
"//third_party/py/numpy",
"@six_archive//:six",
shard_count = 5,
tags = [
"noasan",
"optonly",
],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
"//tensorflow/python:errors",
@ -584,40 +586,38 @@ tf_py_test(
"//tensorflow/python/data/experimental/ops:resampling",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
],
shard_count = 5,
tags = [
"noasan",
"optonly",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
"@six_archive//:six",
],
)
tf_py_test(
name = "replicate_test",
srcs = ["replicate_test.py"],
additional_deps = [
grpc_enabled = True,
tags = ["no_oss"],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_combinations",
"//tensorflow/python/data/experimental/ops:distribute",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
],
grpc_enabled = True,
tags = ["no_oss"],
)
tf_py_test(
name = "replicate_cluster_test",
srcs = ["replicate_cluster_test.py"],
additional_deps = [
grpc_enabled = True,
tags = ["no_oss"],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_combinations",
"//tensorflow/python/data/experimental/ops:distribute",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
],
grpc_enabled = True,
tags = ["no_oss"],
)
cuda_py_test(
@ -649,25 +649,25 @@ tf_py_test(
name = "shuffle_and_repeat_test",
size = "medium",
srcs = ["shuffle_and_repeat_test.py"],
additional_deps = [
"//third_party/py/numpy",
tags = [
"no_pip",
"optonly",
],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:errors",
"//tensorflow/python:framework_ops",
"//tensorflow/python/data/experimental/ops:shuffle_ops",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
],
tags = [
"no_pip",
"optonly",
"//third_party/py/numpy",
],
)
tf_py_test(
name = "sleep_test",
srcs = ["sleep_test.py"],
additional_deps = [
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:util",
"//tensorflow/python/data/experimental/ops:testing",
@ -698,41 +698,42 @@ tf_py_test(
name = "sql_dataset_test",
size = "medium",
srcs = ["sql_dataset_test.py"],
additional_deps = [
tags = ["no_pip"],
deps = [
":sql_dataset_test_base",
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
"//tensorflow/python:errors",
],
tags = ["no_pip"],
)
tf_py_test(
name = "snapshot_test",
size = "medium",
srcs = ["snapshot_test.py"],
additional_deps = [
shard_count = 10,
deps = [
":reader_dataset_ops_test_base",
"@absl_py//absl/testing:parameterized",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:string_ops",
"//tensorflow/python/data/experimental/ops:snapshot",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/ops:readers",
"@absl_py//absl/testing:parameterized",
],
shard_count = 10,
)
tf_py_test(
name = "stats_dataset_ops_test",
size = "large",
srcs = ["stats_dataset_ops_test.py"],
additional_deps = [
tags = [
"no_pip",
],
deps = [
":reader_dataset_ops_test_base",
":stats_dataset_test_base",
"@absl_py//absl/testing:parameterized",
"//third_party/py/numpy",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:errors",
@ -743,9 +744,8 @@ tf_py_test(
"//tensorflow/python/data/experimental/ops:stats_ops",
"//tensorflow/python/data/experimental/ops:stats_options",
"//tensorflow/python/data/ops:dataset_ops",
],
tags = [
"no_pip",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)
@ -764,9 +764,7 @@ tf_py_test(
name = "take_while_test",
size = "small",
srcs = ["take_while_test.py"],
additional_deps = [
"@absl_py//absl/testing:parameterized",
"//third_party/py/numpy",
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
@ -778,6 +776,8 @@ tf_py_test(
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/eager:context",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)
@ -785,7 +785,7 @@ tf_py_test(
name = "tf_record_writer_test",
size = "small",
srcs = ["tf_record_writer_test.py"],
additional_deps = [
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
@ -802,7 +802,8 @@ tf_py_test(
name = "unique_test",
size = "small",
srcs = ["unique_test.py"],
additional_deps = [
tags = ["no_pip"],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
"//tensorflow/python:errors",
@ -811,13 +812,12 @@ tf_py_test(
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
],
tags = ["no_pip"],
)
tf_py_test(
name = "variant_test",
srcs = ["variant_test.py"],
additional_deps = [
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:util",
"//tensorflow/python/data/kernel_tests:test_base",

180
tensorflow/python/data/experimental/kernel_tests/optimization/BUILD
View File

@ -11,18 +11,18 @@ tf_py_test(
name = "choose_fastest_dataset_test",
size = "small",
srcs = ["choose_fastest_dataset_test.py"],
additional_deps = [
"@absl_py//absl/testing:parameterized",
tags = [
"no_oss",
"no_pip",
"no_windows",
],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:errors",
"//tensorflow/python/data/experimental/ops:optimization",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
],
tags = [
"no_oss",
"no_pip",
"no_windows",
"@absl_py//absl/testing:parameterized",
],
)
@ -30,8 +30,12 @@ tf_py_test(
name = "filter_fusion_test",
size = "medium",
srcs = ["filter_fusion_test.py"],
additional_deps = [
"@absl_py//absl/testing:parameterized",
tags = [
"no_oss",
"no_pip",
"no_windows",
],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
"//tensorflow/python:dtypes",
@ -41,11 +45,7 @@ tf_py_test(
"//tensorflow/python/data/experimental/ops:testing",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
],
tags = [
"no_oss",
"no_pip",
"no_windows",
"@absl_py//absl/testing:parameterized",
],
)
@ -53,8 +53,14 @@ tf_py_test(
name = "filter_with_random_uniform_fusion_test",
size = "medium",
srcs = ["filter_with_random_uniform_fusion_test.py"],
additional_deps = [
"@absl_py//absl/testing:parameterized",
tags = [
"manual",
"no_oss",
"no_pip",
"no_windows",
"notap", # TODO(b/131229793)
],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
"//tensorflow/python:dtypes",
@ -64,13 +70,7 @@ tf_py_test(
"//tensorflow/python/data/experimental/ops:testing",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
],
tags = [
"manual",
"no_oss",
"no_pip",
"no_windows",
"notap", # TODO(b/131229793)
"@absl_py//absl/testing:parameterized",
],
)
@ -78,8 +78,12 @@ tf_py_test(
name = "hoist_random_uniform_test",
size = "small",
srcs = ["hoist_random_uniform_test.py"],
additional_deps = [
"@absl_py//absl/testing:parameterized",
tags = [
"no_oss",
"no_pip",
"no_windows",
],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
"//tensorflow/python:control_flow_ops",
@ -92,11 +96,7 @@ tf_py_test(
"//tensorflow/python/data/experimental/ops:testing",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
],
tags = [
"no_oss",
"no_pip",
"no_windows",
"@absl_py//absl/testing:parameterized",
],
)
@ -104,25 +104,30 @@ tf_py_test(
name = "inject_prefetch_test",
size = "small",
srcs = ["inject_prefetch_test.py"],
additional_deps = [
tags = [
"no_oss",
"no_pip",
"no_windows",
],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:errors",
"//tensorflow/python/data/experimental/ops:testing",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
],
tags = [
"no_oss",
"no_pip",
"no_windows",
],
)
tf_py_test(
name = "latency_all_edges_test",
size = "small",
srcs = ["latency_all_edges_test.py"],
additional_deps = [
tags = [
"no_oss",
"no_pip",
"no_windows",
],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:errors",
"//tensorflow/python/data/experimental/kernel_tests:stats_dataset_test_base",
@ -131,17 +136,17 @@ tf_py_test(
"//tensorflow/python/data/experimental/ops:testing",
"//tensorflow/python/data/ops:dataset_ops",
],
tags = [
"no_oss",
"no_pip",
"no_windows",
],
)
tf_py_test(
name = "map_and_batch_fusion_test",
srcs = ["map_and_batch_fusion_test.py"],
additional_deps = [
tags = [
"no_oss",
"no_pip",
"no_windows",
],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:errors",
"//tensorflow/python/data/experimental/ops:optimization_options",
@ -149,18 +154,17 @@ tf_py_test(
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
],
tags = [
"no_oss",
"no_pip",
"no_windows",
],
)
tf_py_test(
name = "map_and_filter_fusion_test",
srcs = ["map_and_filter_fusion_test.py"],
additional_deps = [
"@absl_py//absl/testing:parameterized",
tags = [
"no_oss",
"no_pip",
"no_windows",
],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
"//tensorflow/python:dtypes",
@ -170,30 +174,26 @@ tf_py_test(
"//tensorflow/python/data/experimental/ops:testing",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
],
tags = [
"no_oss",
"no_pip",
"no_windows",
"@absl_py//absl/testing:parameterized",
],
)
tf_py_test(
name = "map_fusion_test",
srcs = ["map_fusion_test.py"],
additional_deps = [
"@absl_py//absl/testing:parameterized",
tags = [
"no_oss",
"no_pip",
"no_windows",
],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:errors",
"//tensorflow/python/data/experimental/ops:optimization_options",
"//tensorflow/python/data/experimental/ops:testing",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
],
tags = [
"no_oss",
"no_pip",
"no_windows",
"@absl_py//absl/testing:parameterized",
],
)
@ -201,8 +201,12 @@ tf_py_test(
name = "map_parallelization_test",
size = "small",
srcs = ["map_parallelization_test.py"],
additional_deps = [
"@absl_py//absl/testing:parameterized",
tags = [
"no_oss",
"no_pip",
"no_windows",
],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
"//tensorflow/python:control_flow_ops",
@ -215,11 +219,7 @@ tf_py_test(
"//tensorflow/python/data/experimental/ops:testing",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
],
tags = [
"no_oss",
"no_pip",
"no_windows",
"@absl_py//absl/testing:parameterized",
],
)
@ -227,9 +227,13 @@ tf_py_test(
name = "map_vectorization_test",
size = "small",
srcs = ["map_vectorization_test.py"],
additional_deps = [
"@absl_py//absl/testing:parameterized",
"//third_party/py/numpy",
shard_count = 8,
tags = [
"no_oss",
"no_pip",
"no_windows",
],
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:array_ops",
"//tensorflow/python:bitwise_ops",
@ -251,12 +255,8 @@ tf_py_test(
"//tensorflow/python/data/experimental/ops:testing",
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
],
shard_count = 8,
tags = [
"no_oss",
"no_pip",
"no_windows",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)
@ -264,7 +264,12 @@ tf_py_test(
name = "noop_elimination_test",
size = "small",
srcs = ["noop_elimination_test.py"],
additional_deps = [
tags = [
"no_oss",
"no_pip",
"no_windows",
],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
"//tensorflow/python:dtypes",
@ -274,17 +279,17 @@ tf_py_test(
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
],
tags = [
"no_oss",
"no_pip",
"no_windows",
],
)
tf_py_test(
name = "shuffle_and_repeat_fusion_test",
srcs = ["shuffle_and_repeat_fusion_test.py"],
additional_deps = [
tags = [
"no_oss",
"no_pip",
"no_windows",
],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:errors",
"//tensorflow/python/data/experimental/ops:optimization_options",
@ -292,9 +297,4 @@ tf_py_test(
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
],
tags = [
"no_oss",
"no_pip",
"no_windows",
],
)

548
tensorflow/python/data/experimental/kernel_tests/serialization/BUILD
View File

File diff suppressed because it is too large Load Diff

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

@ -12,18 +12,18 @@ tf_py_test(
name = "batch_test",
size = "small",
srcs = ["batch_test.py"],
additional_deps = [
deps = [
":test_base",
"@absl_py//absl/testing:parameterized",
"//third_party/py/numpy",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
"//tensorflow/python:errors",
"//tensorflow/python:math_ops",
"//tensorflow/python:sparse_tensor",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/ops/ragged",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)
@ -31,11 +31,8 @@ tf_py_test(
name = "cache_test",
size = "small",
srcs = ["cache_test.py"],
additional_deps = [
deps = [
":test_base",
"//third_party/py/numpy",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/ops:iterator_ops",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
@ -43,6 +40,9 @@ tf_py_test(
"//tensorflow/python:errors",
"//tensorflow/python:framework_ops",
"//tensorflow/python:variables",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/ops:iterator_ops",
"//third_party/py/numpy",
],
)
@ -50,32 +50,32 @@ tf_py_test(
name = "checkpoint_test",
size = "medium",
srcs = ["checkpoint_test.py"],
additional_deps = [
grpc_enabled = True,
deps = [
":test_base",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/eager:context",
"//tensorflow/python/training/tracking:util",
"//tensorflow/python:checkpoint_management",
"//tensorflow/python:client_testlib",
"//tensorflow/python:errors",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/eager:context",
"//tensorflow/python/training/tracking:util",
],
grpc_enabled = True,
)
tf_py_test(
name = "concatenate_test",
size = "small",
srcs = ["concatenate_test.py"],
additional_deps = [
deps = [
":test_base",
"//third_party/py/numpy",
"//tensorflow/python:client_testlib",
"//tensorflow/python:errors",
"//tensorflow/python:tensor_shape",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/util:nest",
"//third_party/py/numpy",
],
)
@ -83,22 +83,22 @@ tf_py_test(
name = "dataset_test",
size = "small",
srcs = ["dataset_test.py"],
additional_deps = [
tags = [
"no_rocm",
],
deps = [
":test_base",
"@absl_py//absl/testing:parameterized",
"//third_party/py/numpy",
"//tensorflow/core:protos_all_py",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/ops:readers",
"//tensorflow/python/data/util:nest",
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
"//tensorflow/python:errors",
"//tensorflow/python:script_ops",
"//tensorflow/python:sparse_tensor",
],
tags = [
"no_rocm",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/ops:readers",
"//tensorflow/python/data/util:nest",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)
@ -106,7 +106,7 @@ tf_py_test(
name = "enumerate_test",
size = "small",
srcs = ["enumerate_test.py"],
additional_deps = [
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
"//tensorflow/python:dtypes",
@ -121,9 +121,8 @@ tf_py_test(
name = "filter_test",
size = "small",
srcs = ["filter_test.py"],
additional_deps = [
deps = [
":test_base",
"//third_party/py/numpy",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
@ -132,6 +131,7 @@ tf_py_test(
"//tensorflow/python:math_ops",
"//tensorflow/python:sparse_tensor",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
],
)
@ -139,7 +139,7 @@ tf_py_test(
name = "fixed_length_record_dataset_test",
size = "small",
srcs = ["fixed_length_record_dataset_test.py"],
additional_deps = [
deps = [
":test_base",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
@ -160,12 +160,10 @@ tf_py_test(
name = "flat_map_test",
size = "medium",
srcs = ["flat_map_test.py"],
additional_deps = [
grpc_enabled = True,
deps = [
":test_base",
"//third_party/py/numpy",
"//tensorflow/core:protos_all_py",
"//tensorflow/python/data/ops:readers",
"//tensorflow/python/data/util:nest",
"//tensorflow/python:client_testlib",
"//tensorflow/python:errors",
"//tensorflow/python:session",
@ -174,26 +172,28 @@ tf_py_test(
"//tensorflow/python:tensor_array_ops",
"//tensorflow/python:training",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/ops:readers",
"//tensorflow/python/data/util:nest",
"//tensorflow/python/ops/ragged",
"//third_party/py/numpy",
],
grpc_enabled = True,
)
tf_py_test(
name = "from_generator_test",
size = "medium",
srcs = ["from_generator_test.py"],
additional_deps = [
deps = [
":test_base",
"//third_party/py/numpy",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
"//tensorflow/python:dtypes",
"//tensorflow/python:errors",
"//tensorflow/python:script_ops",
"//tensorflow/python:tensor_array_ops",
"//tensorflow/python:session",
"//tensorflow/python:tensor_array_ops",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
],
)
@ -201,9 +201,8 @@ tf_py_test(
name = "from_sparse_tensor_slices_test",
size = "small",
srcs = ["from_sparse_tensor_slices_test.py"],
additional_deps = [
deps = [
":test_base",
"//third_party/py/numpy",
"//tensorflow/core:protos_all_py",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
@ -217,6 +216,7 @@ tf_py_test(
"//tensorflow/python:tensor_shape",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/util:nest",
"//third_party/py/numpy",
],
)
@ -224,9 +224,8 @@ tf_py_test(
name = "from_tensors_test",
size = "small",
srcs = ["from_tensors_test.py"],
additional_deps = [
deps = [
":test_base",
"//third_party/py/numpy",
"//tensorflow/core:protos_all_py",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
@ -240,6 +239,7 @@ tf_py_test(
"//tensorflow/python:tensor_shape",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/util:nest",
"//third_party/py/numpy",
],
)
@ -247,11 +247,8 @@ tf_py_test(
name = "from_tensor_slices_test",
size = "small",
srcs = ["from_tensor_slices_test.py"],
additional_deps = [
deps = [
":test_base",
"//third_party/py/numpy",
"//tensorflow/python/ops/ragged:ragged_factory_ops",
"//tensorflow/python/ops/ragged:ragged_tensor",
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
"//tensorflow/python:errors",
@ -260,6 +257,9 @@ tf_py_test(
"//tensorflow/python:tensor_shape",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/ops/ragged",
"//tensorflow/python/ops/ragged:ragged_factory_ops",
"//tensorflow/python/ops/ragged:ragged_tensor",
"//third_party/py/numpy",
],
)
@ -267,18 +267,18 @@ tf_py_test(
name = "interleave_test",
size = "medium",
srcs = ["interleave_test.py"],
additional_deps = [
deps = [
":test_base",
"@absl_py//absl/testing:parameterized",
"//third_party/py/numpy",
"//tensorflow/core:protos_all_py",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:errors",
"//tensorflow/python:script_ops",
"//tensorflow/python:sparse_ops",
"//tensorflow/python:sparse_tensor",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)
@ -286,11 +286,14 @@ tf_py_test(
name = "iterator_cluster_test",
size = "small",
srcs = ["iterator_cluster_test.py"],
additional_deps = [
grpc_enabled = True,
tags = [
"no_oss", # Test flaky due to port collisions.
"no_windows",
],
deps = [
":test_base",
"//tensorflow/core:protos_all_py",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/ops:iterator_ops",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
@ -304,11 +307,8 @@ tf_py_test(
"//tensorflow/python:math_ops",
"//tensorflow/python:session",
"//tensorflow/python:string_ops",
],
grpc_enabled = True,
tags = [
"no_oss", # Test flaky due to port collisions.
"no_windows",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/ops:iterator_ops",
],
)
@ -358,7 +358,7 @@ tf_py_test(
name = "list_files_test",
size = "small",
srcs = ["list_files_test.py"],
additional_deps = [
deps = [
":test_base",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
@ -373,12 +373,10 @@ tf_py_test(
name = "map_test",
size = "medium",
srcs = ["map_test.py"],
additional_deps = [
shard_count = 4,
deps = [
":test_base",
"@absl_py//absl/testing:parameterized",
"//third_party/py/numpy",
"//tensorflow/core:protos_all_py",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
@ -399,9 +397,11 @@ tf_py_test(
"//tensorflow/python:tensor_array_ops",
"//tensorflow/python:tensor_util",
"//tensorflow/python:variable_scope",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/ops/ragged",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
shard_count = 4,
)
cuda_py_test(
@ -450,16 +450,16 @@ tf_py_test(
name = "as_numpy_iterator_test",
size = "small",
srcs = ["as_numpy_iterator_test.py"],
additional_deps = [
deps = [
":test_base",
"@absl_py//absl/testing:parameterized",
"//third_party/py/numpy",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/ops/ragged:ragged_tensor_value",
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
"//tensorflow/python:framework_combinations",
"//tensorflow/python:sparse_tensor",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/ops/ragged:ragged_tensor_value",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)
@ -488,12 +488,12 @@ tf_py_test(
name = "options_test",
size = "small",
srcs = ["options_test.py"],
additional_deps = [
deps = [
":test_base",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python/data/experimental/ops:optimization_options",
"//tensorflow/python/data/experimental/ops:threading_options",
"//tensorflow/python:client_testlib",
"//tensorflow/python/data/ops:dataset_ops",
],
)
@ -501,11 +501,8 @@ tf_py_test(
name = "padded_batch_test",
size = "small",
srcs = ["padded_batch_test.py"],
additional_deps = [
deps = [
":test_base",
"@absl_py//absl/testing:parameterized",
"//third_party/py/numpy",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
@ -515,6 +512,9 @@ tf_py_test(
"//tensorflow/python:string_ops",
"//tensorflow/python:tensor_shape",
"//tensorflow/python:util",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)
@ -522,14 +522,14 @@ tf_py_test(
name = "prefetch_test",
size = "small",
srcs = ["prefetch_test.py"],
additional_deps = [
deps = [
":test_base",
"@absl_py//absl/testing:parameterized",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:dataset_ops_gen",
"//tensorflow/python:dtypes",
"//tensorflow/python:errors",
"@absl_py//absl/testing:parameterized",
],
)
@ -537,12 +537,12 @@ tf_py_test(
name = "range_test",
size = "small",
srcs = ["range_test.py"],
additional_deps = [
deps = [
":test_base",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:errors",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python/data/ops:dataset_ops",
],
)
@ -568,14 +568,14 @@ tf_py_test(
name = "repeat_test",
size = "small",
srcs = ["repeat_test.py"],
additional_deps = [
deps = [
":test_base",
"//third_party/py/numpy",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
"//tensorflow/python:errors",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
],
)
@ -583,7 +583,7 @@ tf_py_test(
name = "shard_test",
size = "small",
srcs = ["shard_test.py"],
additional_deps = [
deps = [
":test_base",
"//tensorflow/python:client_testlib",
"//tensorflow/python:errors",
@ -595,10 +595,8 @@ tf_py_test(
name = "shuffle_test",
size = "small",
srcs = ["shuffle_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:constant_op",
@ -608,6 +606,8 @@ tf_py_test(
"//tensorflow/python:random_seed",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/ops:iterator_ops",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)
@ -615,14 +615,14 @@ tf_py_test(
name = "skip_test",
size = "small",
srcs = ["skip_test.py"],
additional_deps = [
deps = [
":test_base",
"//third_party/py/numpy",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
"//tensorflow/python:errors",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
],
)
@ -630,14 +630,14 @@ tf_py_test(
name = "take_test",
size = "small",
srcs = ["take_test.py"],
additional_deps = [
deps = [
":test_base",
"//third_party/py/numpy",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
"//tensorflow/python:errors",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
],
)
@ -645,16 +645,16 @@ tf_py_test(
name = "text_line_dataset_test",
size = "small",
srcs = ["text_line_dataset_test.py"],
additional_deps = [
deps = [
":test_base",
"//tensorflow/python/data/ops:iterator_ops",
"//tensorflow/python/data/ops:readers",
"//tensorflow/python/eager:context",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
"//tensorflow/python:errors",
"//tensorflow/python:util",
"//tensorflow/python/data/ops:iterator_ops",
"//tensorflow/python/data/ops:readers",
"//tensorflow/python/eager:context",
],
)
@ -662,11 +662,8 @@ tf_py_test(
name = "tf_record_dataset_test",
size = "small",
srcs = ["tf_record_dataset_test.py"],
additional_deps = [
deps = [
":test_base",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/ops:iterator_ops",
"//tensorflow/python/data/ops:readers",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
@ -674,6 +671,9 @@ tf_py_test(
"//tensorflow/python:errors",
"//tensorflow/python:lib",
"//tensorflow/python:util",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/data/ops:iterator_ops",
"//tensorflow/python/data/ops:readers",
],
)
@ -699,9 +699,7 @@ tf_py_test(
name = "unbatch_test",
size = "medium",
srcs = ["unbatch_test.py"],
additional_deps = [
"@absl_py//absl/testing:parameterized",
"//third_party/py/numpy",
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
@ -717,6 +715,8 @@ tf_py_test(
"//tensorflow/python/data/kernel_tests:test_base",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/ops/ragged",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)
@ -724,10 +724,8 @@ tf_py_test(
name = "window_test",
size = "medium",
srcs = ["window_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",
@ -735,6 +733,8 @@ tf_py_test(
"//tensorflow/python:math_ops",
"//tensorflow/python:sparse_tensor",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)
@ -742,13 +742,13 @@ tf_py_test(
name = "zip_test",
size = "small",
srcs = ["zip_test.py"],
additional_deps = [
deps = [
":test_base",
"//third_party/py/numpy",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
"//tensorflow/python:errors",
"//tensorflow/python/data/ops:dataset_ops",
"//third_party/py/numpy",
],
)

16
tensorflow/python/distribute/BUILD
View File

@ -41,20 +41,20 @@ py_library(
tf_py_test(
name = "all_reduce_test",
srcs = ["all_reduce_test.py"],
additional_deps = [
deps = [
":all_reduce",
"//third_party/py/numpy",
"//tensorflow/core:protos_all_py",
"//tensorflow/python:array_ops",
"//tensorflow/python:client",
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
"//tensorflow/python:framework_ops",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:math_ops",
"//tensorflow/python:constant_op",
"//tensorflow/python:client_testlib",
"//tensorflow/python:platform",
"//tensorflow/python:platform_test",
"//tensorflow/python:state_ops",
"//third_party/py/numpy",
],
)
@ -379,10 +379,11 @@ py_library(
tf_py_test(
name = "mirrored_function_strategy_test",
srcs = ["mirrored_function_strategy_test.py"],
additional_deps = [
tags = ["no_pip"],
deps = [
":distribute_lib",
":strategy_combinations",
":mirrored_function_strategy",
":strategy_combinations",
":values",
"//tensorflow/python:constant_op",
"//tensorflow/python:framework_ops",
@ -390,7 +391,6 @@ tf_py_test(
"//tensorflow/python/eager:def_function",
"//tensorflow/python/eager:test",
],
tags = ["no_pip"],
)
py_library(
@ -830,7 +830,7 @@ tf_py_test(
name = "sharded_variable_test",
size = "small",
srcs = ["sharded_variable_test.py"],
additional_deps = [
deps = [
":sharded_variable",
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",

42
tensorflow/python/distribute/cluster_resolver/BUILD
View File

@ -94,7 +94,8 @@ py_library(
tf_py_test(
name = "base_cluster_resolver_py_test",
srcs = ["cluster_resolver_test.py"],
additional_deps = [
main = "cluster_resolver_test.py",
deps = [
":base_cluster_resolver_py",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
@ -102,14 +103,14 @@ tf_py_test(
"//tensorflow/python:platform_test",
"//tensorflow/python:training_server_lib",
],
main = "cluster_resolver_test.py",
)
tf_py_test(
name = "gce_cluster_resolver_py_test",
size = "small",
srcs = ["gce_cluster_resolver_test.py"],
additional_deps = [
main = "gce_cluster_resolver_test.py",
deps = [
":gce_cluster_resolver_py",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
@ -117,14 +118,15 @@ tf_py_test(
"//tensorflow/python:platform_test",
"//tensorflow/python:training_server_lib",
],
main = "gce_cluster_resolver_test.py",
)
tf_py_test(
name = "tfconfig_cluster_resolver_py_test",
size = "small",
srcs = ["tfconfig_cluster_resolver_test.py"],
additional_deps = [
grpc_enabled = True,
main = "tfconfig_cluster_resolver_test.py",
deps = [
":tfconfig_cluster_resolver_py",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
@ -132,33 +134,34 @@ tf_py_test(
"//tensorflow/python:platform_test",
"//tensorflow/python:training_server_lib",
],
grpc_enabled = True,
main = "tfconfig_cluster_resolver_test.py",
)
tf_py_test(
name = "tpu_cluster_resolver_py_test",
size = "small",
srcs = ["tpu_cluster_resolver_test.py"],
additional_deps = [
grpc_enabled = True,
main = "tpu_cluster_resolver_test.py",
python_version = "PY3",
deps = [
":tpu_cluster_resolver_py",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:platform_test",
"//tensorflow/python:training_server_lib",
"//tensorflow/python/tpu/client:client",
"//tensorflow/python/tpu/client",
],
grpc_enabled = True,
main = "tpu_cluster_resolver_test.py",
python_version = "PY3",
)
tf_py_test(
name = "tpu_cluster_resolver_py2_test",
size = "small",
srcs = ["tpu_cluster_resolver_test.py"],
additional_deps = [
grpc_enabled = True,
main = "tpu_cluster_resolver_test.py",
python_version = "PY3",
deps = [
":tpu_cluster_resolver_py",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
@ -166,16 +169,15 @@ tf_py_test(
"//tensorflow/python:platform_test",
"//tensorflow/python:training_server_lib",
],
grpc_enabled = True,
main = "tpu_cluster_resolver_test.py",
python_version = "PY3",
)
tf_py_test(
name = "slurm_cluster_resolver_py_test",
size = "small",
srcs = ["slurm_cluster_resolver_test.py"],
additional_deps = [
main = "slurm_cluster_resolver_test.py",
tags = [],
deps = [
":slurm_cluster_resolver_py",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
@ -183,15 +185,14 @@ tf_py_test(
"//tensorflow/python:platform_test",
"//tensorflow/python:training_server_lib",
],
main = "slurm_cluster_resolver_test.py",
tags = [],
)
tf_py_test(
name = "kubernetes_cluster_resolver_py_test",
size = "small",
srcs = ["kubernetes_cluster_resolver_test.py"],
additional_deps = [
main = "kubernetes_cluster_resolver_test.py",
deps = [
":kubernetes_cluster_resolver_py",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
@ -199,5 +200,4 @@ tf_py_test(
"//tensorflow/python:platform_test",
"//tensorflow/python:training_server_lib",
],
main = "kubernetes_cluster_resolver_test.py",
)

26
tensorflow/python/feature_column/BUILD
View File

@ -107,23 +107,23 @@ filegroup(
tf_py_test(
name = "feature_column_test",
srcs = ["feature_column_test.py"],
additional_deps = [
":feature_column_test_main_lib",
],
tags = [
"no_cuda_on_cpu_tap",
"no_pip",
"no_windows",
],
deps = [
":feature_column_test_main_lib",
],
)
tf_py_test(
name = "dense_features_test",
srcs = ["dense_features_test.py"],
additional_deps = [
tags = ["no_pip"],
deps = [
":feature_column_test_main_lib",
],
tags = ["no_pip"],
)
py_library(
@ -158,22 +158,22 @@ py_library(
tf_py_test(
name = "feature_column_v2_test",
srcs = ["feature_column_v2_test.py"],
additional_deps = [":feature_column_v2_test_main_lib"],
shard_count = 5,
tags = [
"no_cuda_on_cpu_tap",
"no_pip",
"no_windows",
],
deps = [":feature_column_v2_test_main_lib"],
)
tf_py_test(
name = "dense_features_v2_test",
srcs = ["dense_features_v2_test.py"],
additional_deps = [
tags = ["no_pip"],
deps = [
":feature_column_v2_test_main_lib",
],
tags = ["no_pip"],
)
py_library(
@ -221,10 +221,8 @@ py_library(
tf_py_test(
name = "sequence_feature_column_test",
srcs = ["sequence_feature_column_test.py"],
additional_deps = [
deps = [
":feature_column_v2",
"@absl_py//absl/testing:parameterized",
"//third_party/py/numpy",
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
"//tensorflow/python:errors",
@ -234,6 +232,8 @@ tf_py_test(
"//tensorflow/python:session",
"//tensorflow/python:sparse_tensor",
"//tensorflow/python:training",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)
@ -257,10 +257,10 @@ py_test(
tf_py_test(
name = "serialization_test",
srcs = ["serialization_test.py"],
additional_deps = [
deps = [
":feature_column_v2",
"@absl_py//absl/testing:parameterized",
"//tensorflow/python:client_testlib",
"//tensorflow/python:util",
"@absl_py//absl/testing:parameterized",
],
)

910
tensorflow/python/keras/BUILD
View File

File diff suppressed because it is too large Load Diff

10
tensorflow/python/keras/applications/BUILD
View File

@ -41,12 +41,12 @@ tf_py_test(
name = "applications_test",
size = "medium",
srcs = ["applications_test.py"],
additional_deps = [
":applications",
"@absl_py//absl/testing:parameterized",
"//tensorflow/python:client_testlib",
],
shard_count = 32,
deps = [
":applications",
"//tensorflow/python:client_testlib",
"@absl_py//absl/testing:parameterized",
],
)
tf_py_test(

456
tensorflow/python/kernel_tests/BUILD
View File

File diff suppressed because it is too large Load Diff

14
tensorflow/python/kernel_tests/boosted_trees/BUILD
View File

@ -24,7 +24,7 @@ tf_py_test(
name = "resource_ops_test",
size = "small",
srcs = ["resource_ops_test.py"],
additional_deps = [
deps = [
"//tensorflow/core/kernels/boosted_trees:boosted_trees_proto_py",
"//tensorflow/python:boosted_trees_ops",
"//tensorflow/python:framework_ops",
@ -39,12 +39,12 @@ tf_py_test(
name = "prediction_ops_test",
size = "small",
srcs = ["prediction_ops_test.py"],
additional_deps = [
deps = [
"//tensorflow/core/kernels/boosted_trees:boosted_trees_proto_py",
"//tensorflow/python:array_ops",
"//tensorflow/python:boosted_trees_ops",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:constant_op",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:resources",
],
)
@ -53,8 +53,7 @@ tf_py_test(
name = "stats_ops_test",
size = "medium",
srcs = ["stats_ops_test.py"],
additional_deps = [
"//third_party/py/numpy",
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:boosted_trees_ops",
"//tensorflow/python:constant_op",
@ -62,6 +61,7 @@ tf_py_test(
"//tensorflow/python:framework",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:platform_test",
"//third_party/py/numpy",
],
)
@ -69,7 +69,7 @@ tf_py_test(
name = "training_ops_test",
size = "small",
srcs = ["training_ops_test.py"],
additional_deps = [
deps = [
"//tensorflow/core/kernels/boosted_trees:boosted_trees_proto_py",
"//tensorflow/python:array_ops",
"//tensorflow/python:boosted_trees_ops",
@ -83,7 +83,7 @@ tf_py_test(
name = "quantile_ops_test",
size = "small",
srcs = ["quantile_ops_test.py"],
additional_deps = [
deps = [
"//tensorflow/core/kernels/boosted_trees:boosted_trees_proto_py",
"//tensorflow/python:boosted_trees_ops",
"//tensorflow/python:constant_op",

30
tensorflow/python/kernel_tests/proto/BUILD
View File

@ -19,11 +19,6 @@ tf_py_test(
name = "decode_proto_op_test",
size = "small",
srcs = ["decode_proto_op_test.py"],
additional_deps = [
":decode_proto_op_test_base",
":py_test_deps",
"//tensorflow/python:proto_ops",
],
data = if_static(
[],
otherwise = [":libtestexample.so"],
@ -33,17 +28,17 @@ tf_py_test(
"no_pip", # TODO(b/78026780)
"no_windows", # TODO(b/78028010)
],
deps = [
":decode_proto_op_test_base",
":py_test_deps",
"//tensorflow/python:proto_ops",
],
)
tf_py_test(
name = "encode_proto_op_test",
size = "small",
srcs = ["encode_proto_op_test.py"],
additional_deps = [
":encode_proto_op_test_base",
":py_test_deps",
"//tensorflow/python:proto_ops",
],
data = if_static(
[],
otherwise = [":libtestexample.so"],
@ -53,6 +48,11 @@ tf_py_test(
"no_pip", # TODO(b/78026780)
"no_windows", # TODO(b/78028010)
],
deps = [
":encode_proto_op_test_base",
":py_test_deps",
"//tensorflow/python:proto_ops",
],
)
py_library(
@ -122,13 +122,13 @@ tf_py_test(
name = "descriptor_source_test",
size = "small",
srcs = ["descriptor_source_test.py"],
additional_deps = [
":descriptor_source_test_base",
"//tensorflow/python:proto_ops",
"//tensorflow/python:client_testlib",
],
python_version = "PY3",
tags = [
"no_pip",
],
deps = [
":descriptor_source_test_base",
"//tensorflow/python:client_testlib",
"//tensorflow/python:proto_ops",
],
)

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

@ -24,14 +24,14 @@ tf_py_test(
name = "random_shuffle_queue_test",
size = "small",
srcs = ["random_shuffle_queue_test.py"],
additional_deps = [
"//third_party/py/numpy",
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:data_flow_ops",
"//tensorflow/python:errors",
"//tensorflow/python:framework",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:platform",
"//third_party/py/numpy",
],
)
@ -151,18 +151,18 @@ tf_py_test(
name = "random_binomial_test",
size = "medium",
srcs = ["random_binomial_test.py"],
additional_deps = [
shard_count = 3,
tags = ["no_oss"],
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:stateful_random_ops",
"//third_party/py/numpy",
],
shard_count = 3,
tags = ["no_oss"],
)
cuda_py_test(

4
tensorflow/python/profiler/BUILD
View File

@ -100,7 +100,8 @@ tf_py_test(
name = "tfprof_logger_test",
size = "small",
srcs = ["tfprof_logger_test.py"],
additional_deps = [
python_version = "PY3",
deps = [
":tfprof_logger",
"//tensorflow/core:protos_all_py",
"//tensorflow/core/profiler:protos_all_py",
@ -110,7 +111,6 @@ tf_py_test(
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:math_ops",
],
python_version = "PY3",
)
py_library(

34
tensorflow/python/saved_model/BUILD
View File

@ -100,12 +100,11 @@ tf_py_test(
name = "loader_test",
size = "small",
srcs = ["loader_test.py"],
additional_deps = [
deps = [
":builder",
":loader",
":signature_def_utils",
":utils",
"@absl_py//absl/testing:parameterized",
"//tensorflow/python:client",
"//tensorflow/python:client_testlib",
"//tensorflow/python:control_flow_ops",
@ -115,6 +114,7 @@ tf_py_test(
"//tensorflow/python:state_ops",
"//tensorflow/python:training",
"//tensorflow/python:variables",
"@absl_py//absl/testing:parameterized",
],
)
@ -155,7 +155,9 @@ tf_py_test(
name = "saved_model_test",
size = "small",
srcs = ["saved_model_test.py"],
additional_deps = [
data = ["//tensorflow/cc/saved_model:saved_model_half_plus_two"],
tags = ["no_windows"],
deps = [
":builder",
":constants",
":loader",
@ -178,8 +180,6 @@ tf_py_test(
"//tensorflow/python:util",
"//tensorflow/python:variables",
],
data = ["//tensorflow/cc/saved_model:saved_model_half_plus_two"],
tags = ["no_windows"],
)
py_library(
@ -204,7 +204,7 @@ tf_py_test(
name = "utils_test",
size = "small",
srcs = ["utils_test.py"],
additional_deps = [
deps = [
":utils",
"//tensorflow/core:protos_all_py",
"//tensorflow/python:array_ops",
@ -234,7 +234,7 @@ tf_py_test(
name = "signature_def_utils_test",
size = "small",
srcs = ["signature_def_utils_test.py"],
additional_deps = [
deps = [
":signature_constants",
":signature_def_utils",
":utils",
@ -249,7 +249,7 @@ tf_py_test(
name = "simple_save_test",
size = "small",
srcs = ["simple_save_test.py"],
additional_deps = [
deps = [
":loader",
":signature_constants",
":simple_save",
@ -320,19 +320,19 @@ py_library(
tf_py_test(
name = "save_test",
srcs = ["save_test.py"],
additional_deps = [
tags = ["no_rocm"],
deps = [
":loader",
":save",
":save_options",
":signature_constants",
":tag_constants",
"@absl_py//absl/testing:parameterized",
"//tensorflow/core:protos_all_py",
"//tensorflow/python:error_interpolation",
"//tensorflow/python/eager:def_function",
"//tensorflow/python/eager:test",
"@absl_py//absl/testing:parameterized",
],
tags = ["no_rocm"],
)
py_library(
@ -412,14 +412,13 @@ cuda_py_test(
tf_py_test(
name = "load_v1_in_v2_test",
srcs = ["load_v1_in_v2_test.py"],
additional_deps = [
deps = [
":builder",
":load",
":save",
":signature_def_utils",
":simple_save",
":utils",
"@absl_py//absl/testing:parameterized",
"//tensorflow/python:array_ops",
"//tensorflow/python:constant_op",
"//tensorflow/python:dtypes",
@ -428,10 +427,11 @@ tf_py_test(
"//tensorflow/python:resource_variable_ops",
"//tensorflow/python:session",
"//tensorflow/python:tensor_spec",
"//tensorflow/python:variables",
"//tensorflow/python/eager:def_function",
"//tensorflow/python/eager:test",
"//tensorflow/python/training/tracking:tracking",
"//tensorflow/python:variables",
"//tensorflow/python/training/tracking",
"@absl_py//absl/testing:parameterized",
],
)
@ -449,7 +449,7 @@ py_library(
tf_py_test(
name = "revived_types_test",
srcs = ["revived_types_test.py"],
additional_deps = [
deps = [
":revived_types",
"//tensorflow/core:protos_all_py",
"//tensorflow/python:client_testlib",
@ -501,7 +501,7 @@ py_library(
tf_py_test(
name = "nested_structure_coder_test",
srcs = ["nested_structure_coder_test.py"],
additional_deps = [
deps = [
":nested_structure_coder",
"//tensorflow/core:protos_all_py",
"//tensorflow/python:framework",

38
tensorflow/python/tpu/BUILD
View File

@ -257,37 +257,37 @@ tf_py_test(
name = "datasets_test",
size = "medium",
srcs = ["datasets_test.py"],
additional_deps = [
"//tensorflow/python:client_testlib",
":datasets",
],
grpc_enabled = True,
shard_count = 4,
tags = ["no_oss"],
deps = [
":datasets",
"//tensorflow/python:client_testlib",
],
)
tf_py_test(
name = "tpu_test",
size = "small",
srcs = ["tpu_test.py"],
additional_deps = [
tags = [
"no_oss", # TODO(b/131157871): Reenable in OSS when fixed
"no_windows", # TODO: needs investigation on Windows
],
deps = [
":tpu",
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
"//tensorflow/python:framework",
"//tensorflow/python:layers",
],
tags = [
"no_oss", # TODO(b/131157871): Reenable in OSS when fixed
"no_windows", # TODO: needs investigation on Windows
],
)
tf_py_test(
name = "tpu_sharding_test",
size = "small",
srcs = ["tpu_sharding_test.py"],
additional_deps = [
deps = [
":tpu",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework",
@ -298,7 +298,7 @@ tf_py_test(
name = "bfloat16_test",
size = "small",
srcs = ["bfloat16_test.py"],
additional_deps = [
deps = [
":tpu",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework",
@ -309,7 +309,7 @@ tf_py_test(
name = "tpu_infeed_test",
size = "small",
srcs = ["tpu_infeed_test.py"],
additional_deps = [
deps = [
":tpu",
"//tensorflow/python:framework",
"//tensorflow/python:framework_test_lib",
@ -320,7 +320,7 @@ tf_py_test(
name = "topology_test",
size = "medium",
srcs = ["topology_test.py"],
additional_deps = [
deps = [
":tpu",
"//tensorflow/python:framework_test_lib",
],
@ -395,9 +395,9 @@ tf_py_test(
srcs = [
"feature_column_test.py",
],
additional_deps = [
main = "feature_column_test.py",
deps = [
":feature_column",
"//third_party/py/numpy",
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
"//tensorflow/python:framework_ops",
@ -408,8 +408,8 @@ tf_py_test(
"//tensorflow/python:variables",
"//tensorflow/python/feature_column",
"//tensorflow/python/feature_column:feature_column_py",
"//third_party/py/numpy",
],
main = "feature_column_test.py",
)
tf_py_test(
@ -417,9 +417,9 @@ tf_py_test(
srcs = [
"feature_column_v2_test.py",
],
additional_deps = [
main = "feature_column_v2_test.py",
deps = [
":feature_column_v2",
"//third_party/py/numpy",
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
"//tensorflow/python:framework_ops",
@ -430,8 +430,8 @@ tf_py_test(
"//tensorflow/python:variables",
"//tensorflow/python/feature_column",
"//tensorflow/python/feature_column:feature_column_py",
"//third_party/py/numpy",
],
main = "feature_column_v2_test.py",
)
tf_proto_library(

16
tensorflow/python/tpu/client/BUILD
View File

@ -37,7 +37,10 @@ tf_py_test(
name = "client_py_test",
size = "small",
srcs = ["client_test.py"],
additional_deps = [
grpc_enabled = True,
main = "client_test.py",
python_version = "PY3",
deps = [
":client",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
@ -45,16 +48,16 @@ tf_py_test(
"//tensorflow/python:platform_test",
"//tensorflow/python:training_server_lib",
],
grpc_enabled = True,
main = "client_test.py",
python_version = "PY3",
)
tf_py_test(
name = "client_py2_test",
size = "small",
srcs = ["client_test.py"],
additional_deps = [
grpc_enabled = True,
main = "client_test.py",
python_version = "PY2",
deps = [
":client",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
@ -62,7 +65,4 @@ tf_py_test(
"//tensorflow/python:platform_test",
"//tensorflow/python:training_server_lib",
],
grpc_enabled = True,
main = "client_test.py",
python_version = "PY2",
)

46
tensorflow/python/training/tracking/BUILD
View File

@ -39,7 +39,7 @@ py_library(
tf_py_test(
name = "base_test",
srcs = ["base_test.py"],
additional_deps = [
deps = [
":base",
"//tensorflow/python:client_testlib",
],
@ -58,7 +58,7 @@ py_library(
tf_py_test(
name = "tracking_test",
srcs = ["tracking_test.py"],
additional_deps = [
deps = [
":base",
":tracking",
"//tensorflow/python:client_testlib",
@ -86,7 +86,11 @@ py_library(
tf_py_test(
name = "data_structures_test",
srcs = ["data_structures_test.py"],
additional_deps = [
tags = [
"no_windows",
"nomac",
],
deps = [
":data_structures",
"//tensorflow/python:array_ops",
"//tensorflow/python:framework_test_lib",
@ -97,10 +101,6 @@ tf_py_test(
"//tensorflow/python/keras:engine",
"//tensorflow/python/keras:layers",
],
tags = [
"no_windows",
"nomac",
],
)
py_library(
@ -159,14 +159,12 @@ py_library(
tf_py_test(
name = "util_test",
srcs = ["util_test.py"],
additional_deps = [
tags = ["notsan"], # b/74395663
deps = [
":base",
":graph_view",
":tracking",
":util",
"@absl_py//absl/testing:parameterized",
"@six_archive//:six",
"//tensorflow/python/keras/optimizer_v2",
"//tensorflow/python:checkpoint_management",
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
@ -184,6 +182,7 @@ tf_py_test(
"//tensorflow/python:template",
"//tensorflow/python:training_util",
"//tensorflow/python:variable_scope",
"//tensorflow/python:variables",
"//tensorflow/python/eager:backprop",
"//tensorflow/python/eager:context",
"//tensorflow/python/eager:def_function",
@ -191,9 +190,10 @@ tf_py_test(
"//tensorflow/python/keras:backend",
"//tensorflow/python/keras:engine",
"//tensorflow/python/keras:layers",
"//tensorflow/python:variables",
"//tensorflow/python/keras/optimizer_v2",
"@absl_py//absl/testing:parameterized",
"@six_archive//:six",
],
tags = ["notsan"], # b/74395663
)
tf_xla_py_test(
@ -224,13 +224,14 @@ tf_xla_py_test(
tf_py_test(
name = "util_with_v1_optimizers_test",
srcs = ["util_with_v1_optimizers_test.py"],
additional_deps = [
tags = [
"notsan", # b/74395663
],
deps = [
":base",
":graph_view",
":tracking",
":util",
"@absl_py//absl/testing:parameterized",
"@six_archive//:six",
"//tensorflow/python:checkpoint_management",
"//tensorflow/python:constant_op",
"//tensorflow/python:control_flow_ops",
@ -254,9 +255,8 @@ tf_py_test(
"//tensorflow/python/eager:test",
"//tensorflow/python/keras:engine",
"//tensorflow/python/keras:layers",
],
tags = [
"notsan", # b/74395663
"@absl_py//absl/testing:parameterized",
"@six_archive//:six",
],
)
@ -272,22 +272,22 @@ py_library(
tf_py_test(
name = "python_state_test",
srcs = ["python_state_test.py"],
additional_deps = [
deps = [
":base",
":util",
"//tensorflow/python/module",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python/module",
],
)
tf_py_test(
name = "benchmarks_test",
srcs = ["benchmarks_test.py"],
additional_deps = [
deps = [
":util",
"//tensorflow/python:platform_test",
"//tensorflow/python:framework_ops",
"//tensorflow/python:platform_test",
],
)

6
tensorflow/tools/graph_transforms/BUILD
View File

@ -339,7 +339,9 @@ tf_py_test(
name = "transform_graph_py_test",
size = "small",
srcs = ["python/transform_graph_test.py"],
additional_deps = [
main = "python/transform_graph_test.py",
tags = ["v1only"],
deps = [
":transform_graph_py",
"//tensorflow/core:protos_all_py",
"//tensorflow/python:client_testlib",
@ -347,6 +349,4 @@ tf_py_test(
"//tensorflow/python:math_ops",
"//tensorflow/python:variables",
],
main = "python/transform_graph_test.py",
tags = ["v1only"],
)