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

Enable more TFRT test targets that are newly passing

Browse Source 
PiperOrigin-RevId: 333337050
Change-Id: I53a9af35c09f9ddb9ed6dc47bedfce2b8662205b
This commit is contained in:
Kibeom Kim 2020-09-23 11:38:43 -07:00 committed by TensorFlower Gardener
parent c2926ab744
commit aab4cde23e
51 changed files with 277 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/autograph/impl/BUILD
View File

@ -42,6 +42,7 @@ tf_py_test(
srcs = ["api_test.py"],
python_version = "PY3",
srcs_version = "PY3",
tfrt_enabled = True,
deps = [
":impl",
"//tensorflow/python:client_testlib",
@ -54,6 +55,7 @@ tf_py_test(
tf_py_test(
name = "conversion_test",
srcs = ["conversion_test.py"],
tfrt_enabled = True,
deps = [
":impl",
"//tensorflow/python:client_testlib",

2
tensorflow/python/compat/BUILD
View File

@ -34,6 +34,7 @@ tf_py_test(
size = "small",
srcs = ["compat_test.py"],
tags = ["nofwdcompat"],
tfrt_enabled = True,
deps = [
":compat",
"//tensorflow/python:client_testlib",
@ -44,6 +45,7 @@ tf_py_test(
name = "disable_v2_behavior_test",
size = "small",
srcs = ["disable_v2_behavior_test.py"],
tfrt_enabled = True,
deps = [
":v2_compat",
"//tensorflow/python:client_testlib",

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

@ -97,6 +97,7 @@ cuda_py_test(
"no_windows",
"nomac",
],
tfrt_enabled = True,
xla_enable_strict_auto_jit = False,
deps = [
":trt_convert_py",
@ -176,6 +177,7 @@ cuda_py_test(
"no_windows",
"nomac",
],
tfrt_enabled = True,
xla_enable_strict_auto_jit = False,
deps = [
":tf_trt_integration_test_base",

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

@ -99,6 +99,7 @@ cuda_py_test(
"no_mac",
"no_windows",
],
tfrt_enabled = True,
xla_enabled = True,
deps = [
"//tensorflow/python:client_testlib",

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

@ -10,6 +10,7 @@ exports_files(["LICENSE"])
tf_py_test(
name = "meta_benchmark",
srcs = ["meta_benchmark.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:session",
@ -34,6 +35,7 @@ py_library(
tf_py_test(
name = "batch_benchmark",
srcs = ["batch_benchmark.py"],
tfrt_enabled = True,
deps = [
":benchmark_base",
"//tensorflow/python:sparse_tensor",
@ -45,6 +47,7 @@ tf_py_test(
tf_py_test(
name = "filter_benchmark",
srcs = ["filter_benchmark.py"],
tfrt_enabled = True,
deps = [
":benchmark_base",
"//tensorflow/python/data/ops:dataset_ops",
@ -54,6 +57,7 @@ tf_py_test(
tf_py_test(
name = "from_tensor_slices_benchmark",
srcs = ["from_tensor_slices_benchmark.py"],
tfrt_enabled = True,
deps = [
":benchmark_base",
"//tensorflow/python/data/experimental/ops:get_single_element",
@ -65,6 +69,7 @@ tf_py_test(
tf_py_test(
name = "list_files_benchmark",
srcs = ["list_files_benchmark.py"],
tfrt_enabled = True,
deps = [
":benchmark_base",
"//tensorflow/python:client_testlib",
@ -79,6 +84,7 @@ tf_py_test(
tf_py_test(
name = "map_benchmark",
srcs = ["map_benchmark.py"],
tfrt_enabled = True,
deps = [
":benchmark_base",
"//tensorflow/python/data/ops:dataset_ops",
@ -88,6 +94,7 @@ tf_py_test(
tf_py_test(
name = "prefetch_benchmark",
srcs = ["prefetch_benchmark.py"],
tfrt_enabled = True,
deps = [
":benchmark_base",
"//tensorflow/python/data/ops:dataset_ops",
@ -97,6 +104,7 @@ tf_py_test(
tf_py_test(
name = "range_benchmark",
srcs = ["range_benchmark.py"],
tfrt_enabled = True,
deps = [
":benchmark_base",
"//tensorflow/python/data/ops:dataset_ops",

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

@ -25,6 +25,7 @@ py_binary(
tf_py_test(
name = "autotune_benchmark",
srcs = ["autotune_benchmark.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:math_ops",
@ -37,6 +38,7 @@ tf_py_test(
tf_py_test(
name = "choose_fastest_benchmark",
srcs = ["choose_fastest_benchmark.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_ops",
@ -50,6 +52,7 @@ tf_py_test(
tf_py_test(
name = "choose_fastest_branch_benchmark",
srcs = ["choose_fastest_branch_benchmark.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_ops",
@ -66,6 +69,7 @@ tf_py_test(
name = "csv_dataset_benchmark",
srcs = ["csv_dataset_benchmark.py"],
tags = ["no_pip"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:parsing_ops",
@ -81,6 +85,7 @@ tf_py_test(
tf_py_test(
name = "map_and_batch_benchmark",
srcs = ["map_and_batch_benchmark.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:array_ops",
@ -99,6 +104,7 @@ tf_py_test(
tf_py_test(
name = "map_defun_benchmark",
srcs = ["map_defun_benchmark.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
@ -114,6 +120,7 @@ tf_py_test(
tf_py_test(
name = "map_vectorization_benchmark",
srcs = ["map_vectorization_benchmark.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:array_ops",
@ -133,6 +140,7 @@ tf_py_test(
name = "matching_files_benchmark",
size = "small",
srcs = ["matching_files_benchmark.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
@ -148,6 +156,7 @@ tf_py_test(
tf_py_test(
name = "optimize_benchmark",
srcs = ["optimize_benchmark.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_ops",
@ -161,6 +170,7 @@ tf_py_test(
tf_py_test(
name = "parallel_interleave_benchmark",
srcs = ["parallel_interleave_benchmark.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:math_ops",
@ -176,6 +186,7 @@ tf_py_test(
name = "rejection_resample_benchmark",
srcs = ["rejection_resample_benchmark.py"],
tags = ["no_pip"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python/data/experimental/ops:resampling",
@ -188,6 +199,7 @@ tf_py_test(
tf_py_test(
name = "snapshot_dataset_benchmark",
srcs = ["snapshot_dataset_benchmark.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
@ -206,6 +218,7 @@ tf_py_test(
tf_py_test(
name = "unbatch_benchmark",
srcs = ["unbatch_benchmark.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",

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

@ -92,6 +92,7 @@ cuda_py_test(
size = "small",
srcs = ["copy_to_device_test.py"],
tags = ["no_windows_gpu"],
tfrt_enabled = True,
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:client_testlib",
@ -405,6 +406,7 @@ tf_py_test(
size = "small",
srcs = ["map_defun_op_test.py"],
tags = ["no_pip"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:check_ops",
@ -576,6 +578,7 @@ cuda_py_test(
size = "small",
srcs = ["prefetch_to_device_test.py"],
tags = ["no_windows_gpu"],
tfrt_enabled = True,
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:client_testlib",
@ -742,6 +745,7 @@ tf_py_test(
tf_py_test(
name = "sleep_test",
srcs = ["sleep_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:util",

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

@ -40,6 +40,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -58,6 +59,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:array_ops",
@ -80,6 +82,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:array_ops",
@ -100,6 +103,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -118,6 +122,7 @@ tf_py_test(
"no_windows",
"notsan",
],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
@ -141,6 +146,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -161,6 +167,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -178,6 +185,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -196,6 +204,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -214,6 +223,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -231,6 +241,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -249,6 +260,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -269,6 +281,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -286,6 +299,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -311,6 +325,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -329,6 +344,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -347,6 +363,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:array_ops",
@ -366,6 +383,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -386,6 +404,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -404,6 +423,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -427,6 +447,7 @@ tf_py_test(
tags = [
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -445,6 +466,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -462,6 +484,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -479,6 +502,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:array_ops",
@ -498,6 +522,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -518,6 +543,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -542,6 +568,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -558,6 +585,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -574,6 +602,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -597,6 +626,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -614,6 +644,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -632,6 +663,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -649,6 +681,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_combinations",
@ -669,6 +702,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -686,6 +720,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -703,6 +738,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -722,6 +758,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:array_ops",
@ -741,6 +778,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:array_ops",
@ -761,6 +799,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -780,6 +819,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -798,6 +838,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -815,6 +856,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -833,6 +875,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -850,6 +893,7 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",

1
tensorflow/python/data/experimental/service/BUILD
View File

@ -36,6 +36,7 @@ py_library(
tf_py_test(
name = "server_lib_test",
srcs = ["server_lib_test.py"],
tfrt_enabled = True,
deps = [
":server_lib",
"//tensorflow/core:protos_all_py",

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

@ -114,6 +114,7 @@ tf_py_test(
name = "dataset_spec_test",
size = "small",
srcs = ["dataset_spec_test.py"],
tfrt_enabled = True,
deps = [
":test_base",
"//tensorflow/python:client_testlib",
@ -225,6 +226,7 @@ tf_py_test(
name = "from_sparse_tensor_slices_test",
size = "small",
srcs = ["from_sparse_tensor_slices_test.py"],
tfrt_enabled = True,
deps = [
":test_base",
"//tensorflow/core:protos_all_py",
@ -316,6 +318,7 @@ tf_py_test(
"no_oss", # Test flaky due to port collisions.
"no_windows",
],
tfrt_enabled = True,
deps = [
":test_base",
"//tensorflow/core:protos_all_py",

6
tensorflow/python/debug/BUILD
View File

@ -797,6 +797,7 @@ cuda_py_test(
size = "small",
srcs = ["lib/debug_gradients_test.py"],
python_version = "PY3",
tfrt_enabled = True,
xla_enable_strict_auto_jit = False, # Node names are different with autojit
deps = [
":debug_data",
@ -1041,6 +1042,7 @@ cuda_py_test(
size = "small",
srcs = ["lib/debug_grappler_test.py"],
python_version = "PY3",
tfrt_enabled = True,
xla_enable_strict_auto_jit = False, # Tests TF:Classic implementation.
deps = [
":debug_data",
@ -1059,6 +1061,7 @@ cuda_py_test(
srcs = ["lib/session_debug_file_test.py"],
python_version = "PY3",
tags = ["notsan"],
tfrt_enabled = True,
xla_enable_strict_auto_jit = False, # Node names are different with autojit
deps = [
":debug_data",
@ -1077,6 +1080,7 @@ cuda_py_test(
size = "small",
srcs = ["lib/debug_graph_reconstruction_test.py"],
python_version = "PY3",
tfrt_enabled = True,
xla_enable_strict_auto_jit = False, # Node names are different with autojit
deps = [
":debug_data",
@ -1097,6 +1101,7 @@ cuda_py_test(
srcs = ["lib/session_debug_multi_gpu_test.py"],
python_version = "PY3",
tags = ["no_windows_gpu"],
tfrt_enabled = True,
xla_enable_strict_auto_jit = False, # Node names are different with autojit
deps = [
":debug_data",
@ -1219,6 +1224,7 @@ cuda_py_test(
srcs = ["cli/analyzer_cli_test.py"],
python_version = "PY3",
tags = ["no_windows"], # TODO: needs investigation on Windows
tfrt_enabled = True,
xla_enable_strict_auto_jit = False, # Node names are different with autojit
deps = [
":analyzer_cli",

1
tensorflow/python/distribute/BUILD
View File

@ -997,6 +997,7 @@ cuda_py_test(
name = "cross_device_utils_test",
srcs = ["cross_device_utils_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":combinations",
":cross_device_utils",

1
tensorflow/python/distribute/client/BUILD
View File

@ -40,6 +40,7 @@ tf_py_test(
srcs = ["client_test.py"],
python_version = "PY3",
shard_count = 12,
tfrt_enabled = True,
deps = [
":client",
"//tensorflow/python:client_testlib",

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

@ -98,6 +98,7 @@ tf_py_test(
name = "base_cluster_resolver_py_test",
srcs = ["cluster_resolver_test.py"],
main = "cluster_resolver_test.py",
tfrt_enabled = True,
deps = [
":base_cluster_resolver_py",
"//tensorflow/python:client_testlib",
@ -113,6 +114,7 @@ tf_py_test(
size = "small",
srcs = ["gce_cluster_resolver_test.py"],
main = "gce_cluster_resolver_test.py",
tfrt_enabled = True,
deps = [
":gce_cluster_resolver_py",
"//tensorflow/python:client_testlib",
@ -129,6 +131,7 @@ tf_py_test(
srcs = ["tfconfig_cluster_resolver_test.py"],
grpc_enabled = True,
main = "tfconfig_cluster_resolver_test.py",
tfrt_enabled = True,
deps = [
":tfconfig_cluster_resolver_py",
"//tensorflow/python:client_testlib",
@ -145,6 +148,7 @@ tf_py_test(
srcs = ["sagemaker_cluster_resolver_test.py"],
grpc_enabled = True,
main = "sagemaker_cluster_resolver_test.py",
tfrt_enabled = True,
deps = [
":sagemaker_cluster_resolver_py",
"//tensorflow/python:client_testlib",
@ -161,6 +165,7 @@ tf_py_test(
srcs = ["slurm_cluster_resolver_test.py"],
main = "slurm_cluster_resolver_test.py",
tags = [],
tfrt_enabled = True,
deps = [
":slurm_cluster_resolver_py",
"//tensorflow/python:client_testlib",
@ -176,6 +181,7 @@ tf_py_test(
size = "small",
srcs = ["kubernetes_cluster_resolver_test.py"],
main = "kubernetes_cluster_resolver_test.py",
tfrt_enabled = True,
deps = [
":kubernetes_cluster_resolver_py",
"//tensorflow/python:client_testlib",

1
tensorflow/python/distribute/cluster_resolver/tpu/BUILD
View File

@ -32,6 +32,7 @@ tf_py_test(
grpc_enabled = True,
main = "tpu_cluster_resolver_test.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":tpu_cluster_resolver_py",
"//tensorflow/python:client_testlib",

8
tensorflow/python/keras/BUILD
View File

@ -336,6 +336,7 @@ tf_py_test(
size = "small",
srcs = ["activations_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":activations",
":backend",
@ -355,6 +356,7 @@ tf_py_test(
size = "small",
srcs = ["combinations_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":combinations",
":testing_utils",
@ -371,6 +373,7 @@ tf_py_test(
size = "small",
srcs = ["constraints_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":backend",
":combinations",
@ -385,6 +388,7 @@ tf_py_test(
size = "small",
srcs = ["initializers_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":backend",
":combinations",
@ -419,6 +423,7 @@ tf_py_test(
python_version = "PY3",
shard_count = 8,
tags = ["notsan"],
tfrt_enabled = True,
deps = [
":keras",
"//tensorflow/python:client_testlib",
@ -453,6 +458,7 @@ tf_py_test(
size = "small",
srcs = ["metrics_functional_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":keras",
"//tensorflow/python:client_testlib",
@ -621,6 +627,7 @@ tf_py_test(
size = "medium",
srcs = ["backend_config_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":backend",
":backend_config",
@ -635,6 +642,7 @@ tf_py_test(
srcs = ["keras_parameterized_test.py"],
python_version = "PY3",
tags = ["notsan"],
tfrt_enabled = True,
deps = [
":keras",
"//tensorflow/python:client_testlib",

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

@ -64,6 +64,7 @@ tf_py_test(
"no_rocm",
"notsan", # b/168814536
],
tfrt_enabled = True,
deps = [
":applications",
"//tensorflow/python:client_testlib",
@ -85,6 +86,7 @@ tf_py_test(
"no_pip",
"notsan", # b/168814536
],
tfrt_enabled = True,
deps = [
":applications",
"//tensorflow/python:client_testlib",
@ -101,6 +103,7 @@ tf_py_test(
"no_oss",
"no_pip",
],
tfrt_enabled = True,
deps = [
":applications",
"//tensorflow/python:client_testlib",
@ -117,6 +120,7 @@ tf_py_test(
"no_oss",
"no_pip",
],
tfrt_enabled = True,
deps = [
":applications",
"//tensorflow/python:client_testlib",
@ -133,6 +137,7 @@ tf_py_test(
"no_oss",
"no_pip",
],
tfrt_enabled = True,
deps = [
":applications",
"//tensorflow/python:client_testlib",
@ -149,6 +154,7 @@ tf_py_test(
"no_oss",
"no_pip",
],
tfrt_enabled = True,
deps = [
":applications",
"//tensorflow/python:client_testlib",
@ -165,6 +171,7 @@ tf_py_test(
"no_oss",
"no_pip",
],
tfrt_enabled = True,
deps = [
":applications",
"//tensorflow/python:client_testlib",
@ -181,6 +188,7 @@ tf_py_test(
"no_oss",
"no_pip",
],
tfrt_enabled = True,
deps = [
":applications",
"//tensorflow/python:client_testlib",
@ -197,6 +205,7 @@ tf_py_test(
"no_oss",
"no_pip",
],
tfrt_enabled = True,
deps = [
":applications",
"//tensorflow/python:client_testlib",
@ -213,6 +222,7 @@ tf_py_test(
"no_oss",
"no_pip",
],
tfrt_enabled = True,
deps = [
":applications",
"//tensorflow/python:client_testlib",
@ -229,6 +239,7 @@ tf_py_test(
"no_oss",
"no_pip",
],
tfrt_enabled = True,
deps = [
":applications",
"//tensorflow/python:client_testlib",
@ -245,6 +256,7 @@ tf_py_test(
"no_oss",
"no_pip",
],
tfrt_enabled = True,
deps = [
":applications",
"//tensorflow/python:client_testlib",
@ -263,6 +275,7 @@ tf_py_test(
"no_oss",
"no_pip",
],
tfrt_enabled = True,
deps = [
":applications",
"//tensorflow/python:client_testlib",
@ -281,6 +294,7 @@ tf_py_test(
"no_oss",
"no_pip",
],
tfrt_enabled = True,
deps = [
":applications",
"//tensorflow/python:client_testlib",
@ -297,6 +311,7 @@ tf_py_test(
"no_oss",
"no_pip",
],
tfrt_enabled = True,
deps = [
":applications",
"//tensorflow/python:client_testlib",
@ -313,6 +328,7 @@ tf_py_test(
"no_oss",
"no_pip",
],
tfrt_enabled = True,
deps = [
":applications",
"//tensorflow/python:client_testlib",
@ -325,6 +341,7 @@ tf_py_test(
size = "medium",
srcs = ["imagenet_utils_test.py"],
shard_count = 2,
tfrt_enabled = True,
deps = [
":applications",
"//tensorflow/python:client_testlib",

11
tensorflow/python/keras/benchmarks/BUILD
View File

@ -72,6 +72,7 @@ cuda_py_test(
tags = COMMON_TAGS + [
"no_oss_py38", # TODO(b/162044699)
],
tfrt_enabled = True,
deps = [
":profiler_lib",
"//tensorflow:tensorflow_py",
@ -82,6 +83,7 @@ cuda_py_test(
name = "model_components_benchmarks_test",
srcs = ["model_components_benchmarks_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":profiler_lib",
"//tensorflow:tensorflow_py",
@ -103,6 +105,7 @@ cuda_py_test(
srcs = ["keras_examples_benchmarks/bidirectional_lstm_benchmark_test.py"],
python_version = "PY3",
tags = COMMON_TAGS,
tfrt_enabled = True,
deps = [
":benchmark_util",
":profiler_lib",
@ -115,6 +118,7 @@ cuda_py_test(
srcs = ["keras_examples_benchmarks/text_classification_transformer_benchmark_test.py"],
python_version = "PY3",
tags = COMMON_TAGS,
tfrt_enabled = True,
deps = [
":benchmark_util",
"//tensorflow:tensorflow_py",
@ -126,6 +130,7 @@ cuda_py_test(
srcs = ["keras_examples_benchmarks/antirectifier_benchmark_test.py"],
python_version = "PY3",
tags = COMMON_TAGS,
tfrt_enabled = True,
deps = [
":benchmark_util",
"//tensorflow:tensorflow_py",
@ -137,6 +142,7 @@ cuda_py_test(
srcs = ["keras_examples_benchmarks/mnist_conv_benchmark_test.py"],
python_version = "PY3",
tags = COMMON_TAGS,
tfrt_enabled = True,
deps = [
":benchmark_util",
"//tensorflow:tensorflow_py",
@ -149,6 +155,7 @@ cuda_py_test(
srcs = ["keras_examples_benchmarks/mnist_hierarchical_rnn_benchmark_test.py"],
python_version = "PY3",
tags = COMMON_TAGS,
tfrt_enabled = True,
deps = [
":benchmark_util",
"//tensorflow:tensorflow_py",
@ -160,6 +167,7 @@ cuda_py_test(
srcs = ["keras_examples_benchmarks/mnist_irnn_benchmark_test.py"],
python_version = "PY3",
tags = COMMON_TAGS,
tfrt_enabled = True,
deps = [
":benchmark_util",
"//tensorflow:tensorflow_py",
@ -171,6 +179,7 @@ cuda_py_test(
srcs = ["keras_examples_benchmarks/reuters_mlp_benchmark_test.py"],
python_version = "PY3",
tags = COMMON_TAGS,
tfrt_enabled = True,
deps = [
":benchmark_util",
"//tensorflow:tensorflow_py",
@ -183,6 +192,7 @@ cuda_py_test(
srcs = ["keras_examples_benchmarks/cifar10_cnn_benchmark_test.py"],
python_version = "PY3",
tags = COMMON_TAGS,
tfrt_enabled = True,
deps = [
":benchmark_util",
"//tensorflow:tensorflow_py",
@ -194,6 +204,7 @@ cuda_py_test(
srcs = ["keras_examples_benchmarks/mnist_conv_custom_training_benchmark_test.py"],
python_version = "PY3",
tags = COMMON_TAGS,
tfrt_enabled = True,
deps = [
":distribution_util",
"//tensorflow:tensorflow_py",

8
tensorflow/python/keras/benchmarks/saved_model_benchmarks/BUILD
View File

@ -45,6 +45,7 @@ cuda_py_test(
"no_pip", # b/161253163
"no_windows", # b/160628318
],
tfrt_enabled = True,
deps = [
":saved_model_benchmark_util",
"//tensorflow:tensorflow_py",
@ -59,6 +60,7 @@ cuda_py_test(
"no_pip", # b/161253163
"no_windows", # b/160628318
],
tfrt_enabled = True,
deps = [
":saved_model_benchmark_util",
"//tensorflow:tensorflow_py",
@ -73,6 +75,7 @@ cuda_py_test(
"no_pip", # b/161253163
"no_windows", # b/160628318
],
tfrt_enabled = True,
deps = [
":saved_model_benchmark_util",
"//tensorflow:tensorflow_py",
@ -87,6 +90,7 @@ cuda_py_test(
"no_pip", # b/161253163
"no_windows", # b/160628318
],
tfrt_enabled = True,
deps = [
":saved_model_benchmark_util",
"//tensorflow:tensorflow_py",
@ -101,6 +105,7 @@ cuda_py_test(
"no_pip", # b/161253163
"no_windows", # b/160628318
],
tfrt_enabled = True,
deps = [
":saved_model_benchmark_util",
"//tensorflow:tensorflow_py",
@ -115,6 +120,7 @@ cuda_py_test(
"no_pip", # b/161253163
"no_windows", # b/160628318
],
tfrt_enabled = True,
deps = [
":saved_model_benchmark_util",
"//tensorflow:tensorflow_py",
@ -129,6 +135,7 @@ cuda_py_test(
"no_pip", # b/161253163
"no_windows", # b/160628318
],
tfrt_enabled = True,
deps = [
":saved_model_benchmark_util",
"//tensorflow:tensorflow_py",
@ -143,6 +150,7 @@ cuda_py_test(
"no_pip", # b/161253163
"no_windows", # b/160628318
],
tfrt_enabled = True,
deps = [
":saved_model_benchmark_util",
"//tensorflow:tensorflow_py",

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

@ -138,6 +138,7 @@ cuda_py_test(
srcs = ["worker_training_state_test.py"],
python_version = "PY3",
shard_count = 4,
tfrt_enabled = True,
deps = [
":multi_worker_testing_utils",
":worker_training_state",
@ -598,6 +599,7 @@ cuda_py_test(
"multi_and_single_gpu",
"tf_integration_test",
],
tfrt_enabled = True,
deps = [
":keras_test_lib",
],
@ -643,6 +645,7 @@ cuda_py_test(
"multi_and_single_gpu",
"no_windows_gpu", # TODO(b/130551176)
],
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:training_lib",

5
tensorflow/python/keras/engine/BUILD
View File

@ -299,6 +299,7 @@ cuda_py_test(
tags = [
"nomac", # TODO(mihaimaruseac): b/127695564
],
tfrt_enabled = True,
deps = [
":engine",
"//tensorflow/python:client_testlib",
@ -337,6 +338,7 @@ tf_py_test(
tags = [
"nomac", # TODO(mihaimaruseac): b/127695564
],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python/keras",
@ -352,6 +354,7 @@ tf_py_test(
tags = [
"nomac", # TODO(mihaimaruseac): b/127695564
],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python/keras",
@ -546,6 +549,7 @@ tf_py_test(
"no_oss", # TODO(b/135021748) reenable
"notsan",
],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python/keras",
@ -616,6 +620,7 @@ tf_py_test(
tags = [
"nomac", # TODO(mihaimaruseac): b/127695564
],
tfrt_enabled = True,
deps = [
":base_layer",
":engine",

7
tensorflow/python/keras/layers/BUILD
View File

@ -544,6 +544,7 @@ cuda_py_test(
"no_cuda11",
"no_windows_gpu",
],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python/keras",
@ -591,6 +592,7 @@ tf_py_test(
srcs = ["subclassed_layers_test.py"],
python_version = "PY3",
shard_count = 3,
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python/keras",
@ -805,6 +807,7 @@ cuda_py_test(
size = "medium",
srcs = ["separable_convolutional_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python/keras",
@ -856,6 +859,7 @@ tf_py_test(
size = "small",
srcs = ["serialization_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python/keras",
@ -869,6 +873,7 @@ tf_py_test(
size = "small",
srcs = ["kernelized_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":layers",
"//tensorflow/python:array_ops",
@ -922,6 +927,7 @@ tf_py_test(
tags = [
"notsan",
],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python/keras",
@ -936,6 +942,7 @@ tf_py_test(
size = "small",
srcs = ["layers_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":layers",
"//tensorflow/python:client_testlib",

1
tensorflow/python/keras/layers/preprocessing/BUILD
View File

@ -537,6 +537,7 @@ tf_py_test(
name = "table_utils_test",
srcs = ["table_utils_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":table_utils",
"//tensorflow/python:client_testlib",

6
tensorflow/python/keras/layers/preprocessing/benchmarks/BUILD
View File

@ -20,6 +20,7 @@ tf_py_test(
name = "category_encoding_benchmark",
srcs = ["category_encoding_benchmark.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow:tensorflow_py",
"//tensorflow/python/keras/layers/preprocessing:category_encoding",
@ -30,6 +31,7 @@ tf_py_test(
name = "category_crossing_benchmark",
srcs = ["category_crossing_benchmark.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow:tensorflow_py",
"//tensorflow/python/keras/layers/preprocessing:category_crossing",
@ -40,6 +42,7 @@ tf_py_test(
name = "hashing_benchmark",
srcs = ["hashing_benchmark.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow:tensorflow_py",
"//tensorflow/python/keras/layers/preprocessing:hashing",
@ -50,6 +53,7 @@ tf_py_test(
name = "index_lookup_adapt_benchmark",
srcs = ["index_lookup_adapt_benchmark.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow:tensorflow_py",
"//tensorflow/python/keras/layers/preprocessing:index_lookup",
@ -60,6 +64,7 @@ tf_py_test(
name = "normalization_adapt_benchmark",
srcs = ["normalization_adapt_benchmark.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow:tensorflow_py",
"//tensorflow/python/keras/layers/preprocessing:normalization",
@ -70,6 +75,7 @@ cuda_py_test(
name = "image_preproc_benchmark",
srcs = ["image_preproc_benchmark.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow:tensorflow_py",
"//tensorflow/python/keras/layers/preprocessing:image_preprocessing",

5
tensorflow/python/keras/legacy_tf_layers/BUILD
View File

@ -85,6 +85,7 @@ tf_py_test(
srcs = ["base_test.py"],
main = "base_test.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":core",
":layers_base",
@ -115,6 +116,7 @@ tf_py_test(
srcs = ["core_test.py"],
main = "core_test.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":core",
"//tensorflow/python:array_ops",
@ -142,6 +144,7 @@ tf_py_test(
srcs = ["convolutional_test.py"],
main = "convolutional_test.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":convolutional",
"//tensorflow/python:array_ops",
@ -166,6 +169,7 @@ tf_py_test(
main = "pooling_test.py",
python_version = "PY3",
tags = ["no_rocm"],
tfrt_enabled = True,
deps = [
":pooling",
"//tensorflow/python:array_ops",
@ -183,6 +187,7 @@ cuda_py_test(
main = "normalization_test.py",
python_version = "PY3",
shard_count = 10,
tfrt_enabled = True,
deps = [
":convolutional",
":normalization",

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

@ -93,6 +93,7 @@ cuda_py_test(
name = "device_compatibility_check_test",
srcs = ["device_compatibility_check_test.py"],
srcs_version = "PY2AND3",
tfrt_enabled = True,
deps = [
":device_compatibility_check",
"//tensorflow/python:client_testlib",
@ -204,6 +205,7 @@ cuda_py_test(
name = "loss_scale_benchmark",
size = "medium",
srcs = ["loss_scale_benchmark.py"],
tfrt_enabled = True,
deps = [
":loss_scale_optimizer",
":test_util",
@ -221,6 +223,7 @@ cuda_py_test(
size = "small",
srcs = ["mixed_precision_graph_rewrite_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":loss_scale_optimizer",
":policy",

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

@ -94,6 +94,7 @@ cuda_py_test(
size = "medium",
srcs = ["adagrad_test.py"],
shard_count = 4,
tfrt_enabled = True,
deps = [
":optimizer_v2",
"//tensorflow/python:client_testlib",
@ -135,6 +136,7 @@ cuda_py_test(
size = "medium",
srcs = ["adamax_test.py"],
shard_count = 4,
tfrt_enabled = True,
deps = [
":optimizer_v2",
"//tensorflow/python:client_testlib",
@ -155,6 +157,7 @@ cuda_py_test(
size = "medium",
srcs = ["adadelta_test.py"],
shard_count = 4,
tfrt_enabled = True,
deps = [
":optimizer_v2",
"//tensorflow/python:client_testlib",
@ -175,6 +178,7 @@ cuda_py_test(
size = "medium",
srcs = ["ftrl_test.py"],
shard_count = 4,
tfrt_enabled = True,
deps = [
":optimizer_v2",
"//tensorflow/python:client_testlib",
@ -214,6 +218,7 @@ cuda_py_test(
size = "medium",
srcs = ["nadam_test.py"],
shard_count = 4,
tfrt_enabled = True,
deps = [
":optimizer_v2",
"//tensorflow/python:client_testlib",
@ -260,6 +265,7 @@ cuda_py_test(
size = "medium",
srcs = ["learning_rate_schedule_test.py"],
shard_count = 4,
tfrt_enabled = True,
deps = [
":optimizer_v2",
"//tensorflow/python:client_testlib",
@ -274,6 +280,7 @@ cuda_py_test(
name = "legacy_learning_rate_decay_test",
size = "medium",
srcs = ["legacy_learning_rate_decay_test.py"],
tfrt_enabled = True,
deps = [
":legacy_learning_rate_decay",
"//tensorflow/python:framework_test_lib",
@ -291,6 +298,7 @@ cuda_py_test(
size = "medium",
srcs = ["rmsprop_test.py"],
shard_count = 2,
tfrt_enabled = True,
deps = [
":optimizer_v2",
"//tensorflow/python:array_ops",

2
tensorflow/python/keras/preprocessing/BUILD
View File

@ -115,6 +115,7 @@ tf_py_test(
size = "small",
srcs = ["sequence_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":sequence",
"//tensorflow/python:client_testlib",
@ -127,6 +128,7 @@ tf_py_test(
size = "small",
srcs = ["text_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":text",
"//tensorflow/python:client_testlib",

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

@ -137,6 +137,7 @@ tf_py_test(
"no_oss", # TODO(b/119349471): Re-enable
"no_windows",
],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python/keras",
@ -204,6 +205,7 @@ tf_py_test(
size = "small",
srcs = ["saved_model/json_utils_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//third_party/py/numpy",

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

@ -31,6 +31,7 @@ tf_py_test(
tags = [
"no_pip",
],
tfrt_enabled = True,
deps = [
":get_config_samples",
"//tensorflow/python:client_testlib",
@ -92,6 +93,7 @@ tf_py_test(
srcs = ["convert_to_constants_test.py"],
python_version = "PY3",
shard_count = 4,
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
@ -127,6 +129,7 @@ tf_py_test(
name = "graph_util_test",
srcs = ["graph_util_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:client_testlib",
@ -174,6 +177,7 @@ tf_py_test(
srcs = ["model_architectures_test.py"],
python_version = "PY3",
shard_count = 16,
tfrt_enabled = True,
deps = [
":model_architectures",
"//tensorflow/python/keras",
@ -266,6 +270,7 @@ tf_py_test(
"nomsan", # TODO(b/149948895): Re-enable.
"notsan", # TODO(b/149948895): Re-enable.
],
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
@ -297,6 +302,7 @@ cuda_py_test(
name = "summary_ops_test",
size = "small",
srcs = ["summary_ops_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:client_testlib",
@ -313,6 +319,7 @@ tf_py_test(
name = "saved_model_test",
size = "small",
srcs = ["saved_model_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
@ -357,6 +364,7 @@ tf_py_test(
size = "small",
srcs = ["serialization_util_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
@ -388,6 +396,7 @@ tf_py_test(
"no_windows",
"nomac",
],
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:constant_op",
@ -415,6 +424,7 @@ tf_py_test(
srcs = ["tracking_util_test.py"],
python_version = "PY3",
tags = ["notsan"], # b/74395663
tfrt_enabled = True,
deps = [
"//tensorflow/compiler/tests:xla_test",
"//tensorflow/python:checkpoint_management",

11
tensorflow/python/keras/utils/BUILD
View File

@ -250,6 +250,7 @@ tf_py_test(
size = "small",
srcs = ["generic_utils_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":generic_utils",
"//tensorflow/python:client_testlib",
@ -263,6 +264,7 @@ tf_py_test(
size = "small",
srcs = ["version_utils_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":version_utils",
"//tensorflow/python:client_testlib",
@ -276,6 +278,7 @@ tf_py_test(
size = "small",
srcs = ["tf_utils_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":tf_utils",
"//tensorflow/python:client_testlib",
@ -317,6 +320,7 @@ tf_py_test(
"no_windows", # TODO: needs investigation on Windows
"notsan",
],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python/keras",
@ -330,6 +334,7 @@ tf_py_test(
size = "small",
srcs = ["layer_utils_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":layer_utils",
"//tensorflow/python:client_testlib",
@ -343,6 +348,7 @@ tf_py_test(
size = "small",
srcs = ["np_utils_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python/keras",
@ -356,6 +362,7 @@ tf_py_test(
size = "small",
srcs = ["kernelized_utils_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
@ -372,6 +379,7 @@ cuda_py_test(
"guitar",
"multi_gpu",
],
tfrt_enabled = True,
xla_enable_strict_auto_jit = True,
deps = [
"//tensorflow/python:client_testlib",
@ -386,6 +394,7 @@ tf_py_test(
size = "small",
srcs = ["vis_utils_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python/keras",
@ -399,6 +408,7 @@ tf_py_test(
size = "small",
srcs = ["conv_utils_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python/keras",
@ -412,6 +422,7 @@ tf_py_test(
size = "small",
srcs = ["metrics_utils_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow/python:constant_op",
"//tensorflow/python:framework_ops",

1
tensorflow/python/keras/wrappers/BUILD
View File

@ -38,6 +38,7 @@ tf_py_test(
srcs = ["scikit_learn_test.py"],
python_version = "PY3",
tags = ["notsan"],
tfrt_enabled = True,
deps = [
":wrappers",
"//tensorflow/python:client_testlib",

2
tensorflow/python/kernel_tests/BUILD
View File

@ -697,6 +697,7 @@ cuda_py_test(
name = "logging_ops_test",
size = "small",
srcs = ["logging_ops_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:control_flow_ops",
@ -981,6 +982,7 @@ tf_py_test(
name = "save_restore_ops_test",
size = "small",
srcs = ["save_restore_ops_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:client",

4
tensorflow/python/kernel_tests/array_ops/BUILD
View File

@ -10,6 +10,7 @@ package(
cuda_py_test(
name = "batch_gather_op_test",
srcs = ["batch_gather_op_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
@ -23,6 +24,7 @@ cuda_py_test(
name = "unstack_op_test",
size = "small",
srcs = ["unstack_op_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
@ -37,6 +39,7 @@ cuda_py_test(
name = "slice_op_test",
size = "medium",
srcs = ["slice_op_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
@ -51,6 +54,7 @@ cuda_py_test(
name = "gather_op_test",
size = "medium",
srcs = ["gather_op_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",

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

@ -24,6 +24,7 @@ tf_py_test(
name = "resource_ops_test",
size = "small",
srcs = ["resource_ops_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/core/kernels/boosted_trees:boosted_trees_proto_py",
"//tensorflow/python:boosted_trees_ops",
@ -39,6 +40,7 @@ tf_py_test(
name = "prediction_ops_test",
size = "small",
srcs = ["prediction_ops_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/core/kernels/boosted_trees:boosted_trees_proto_py",
"//tensorflow/python:array_ops",
@ -53,6 +55,7 @@ tf_py_test(
name = "stats_ops_test",
size = "medium",
srcs = ["stats_ops_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:boosted_trees_ops",
@ -69,6 +72,7 @@ tf_py_test(
name = "training_ops_test",
size = "small",
srcs = ["training_ops_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/core/kernels/boosted_trees:boosted_trees_proto_py",
"//tensorflow/python:array_ops",
@ -83,6 +87,7 @@ tf_py_test(
name = "quantile_ops_test",
size = "small",
srcs = ["quantile_ops_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/core/kernels/boosted_trees:boosted_trees_proto_py",
"//tensorflow/python:boosted_trees_ops",

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

@ -11,6 +11,7 @@ cuda_py_test(
name = "bijector_test",
size = "small",
srcs = ["bijector_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
@ -48,6 +49,7 @@ cuda_py_test(
name = "kullback_leibler_test",
size = "small",
srcs = ["kullback_leibler_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
@ -60,6 +62,7 @@ cuda_py_test(
name = "beta_test",
size = "small",
srcs = ["beta_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client",
"//tensorflow/python:client_testlib",
@ -92,6 +95,7 @@ cuda_py_test(
name = "categorical_test",
size = "small",
srcs = ["categorical_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
@ -189,6 +193,7 @@ cuda_py_test(
cuda_py_test(
name = "multinomial_test",
srcs = ["multinomial_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
@ -265,6 +270,7 @@ cuda_py_test(
name = "special_math_test",
size = "medium",
srcs = ["special_math_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
@ -281,6 +287,7 @@ cuda_py_test(
name = "identity_bijector_test",
size = "small",
srcs = ["identity_bijector_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",

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

@ -27,6 +27,7 @@ cuda_py_test(
name = "linear_operator_addition_test",
size = "small",
srcs = ["linear_operator_addition_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
@ -65,6 +66,7 @@ cuda_py_test(
name = "linear_operator_algebra_test",
size = "small",
srcs = ["linear_operator_algebra_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",

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

@ -11,6 +11,7 @@ cuda_py_test(
name = "conjugate_gradient_test",
size = "medium",
srcs = ["conjugate_gradient_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
@ -28,6 +29,7 @@ cuda_py_test(
size = "medium",
srcs = ["csr_sparse_matrix_test.py"],
main = "csr_sparse_matrix_test.py",
tfrt_enabled = True,
deps = [
"//tensorflow/python/ops/linalg/sparse",
],
@ -53,6 +55,7 @@ cuda_py_test(
srcs = ["csr_sparse_matrix_grad_test.py"],
main = "csr_sparse_matrix_grad_test.py",
shard_count = 50,
tfrt_enabled = True,
deps = [
"//tensorflow/python/ops/linalg/sparse",
],
@ -64,6 +67,7 @@ cuda_py_test(
srcs = ["csr_sparse_matrix_dense_mat_mul_grad_test.py"],
main = "csr_sparse_matrix_dense_mat_mul_grad_test.py",
shard_count = 50,
tfrt_enabled = True,
deps = [
"//tensorflow/python/ops/linalg/sparse",
],
@ -75,6 +79,7 @@ cuda_py_test(
srcs = ["csr_sparse_matrix_sparse_mat_mul_grad_test.py"],
main = "csr_sparse_matrix_sparse_mat_mul_grad_test.py",
shard_count = 50,
tfrt_enabled = True,
deps = [
"//tensorflow/python/ops/linalg/sparse",
],

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

@ -28,6 +28,7 @@ tf_py_test(
"no_pip", # TODO(b/78026780)
"no_windows", # TODO(b/78028010)
],
tfrt_enabled = True,
deps = [
":decode_proto_op_test_base",
":py_test_deps",
@ -48,6 +49,7 @@ tf_py_test(
"no_pip", # TODO(b/78026780)
"no_windows", # TODO(b/78028010)
],
tfrt_enabled = True,
deps = [
":encode_proto_op_test_base",
":py_test_deps",
@ -126,6 +128,7 @@ tf_py_test(
tags = [
"no_pip",
],
tfrt_enabled = True,
deps = [
":descriptor_source_test_base",
"//tensorflow/python:client_testlib",

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

@ -24,6 +24,7 @@ cuda_py_test(
name = "parameterized_truncated_normal_op_test",
size = "medium",
srcs = ["parameterized_truncated_normal_op_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:client",
@ -42,6 +43,7 @@ tf_py_test(
name = "random_shuffle_queue_test",
size = "small",
srcs = ["random_shuffle_queue_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:data_flow_ops",
@ -57,6 +59,7 @@ cuda_py_test(
name = "multinomial_op_test",
size = "small",
srcs = ["multinomial_op_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:array_ops",
@ -76,6 +79,7 @@ cuda_py_test(
size = "medium",
srcs = ["multinomial_op_big_test.py"],
shard_count = 3,
tfrt_enabled = True,
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:array_ops",
@ -94,6 +98,7 @@ cuda_py_test(
name = "random_crop_test",
size = "small",
srcs = ["random_crop_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:random_ops",
@ -138,6 +143,7 @@ cuda_py_test(
srcs = ["random_gamma_test.py"],
shard_count = 4,
tags = ["nozapfhahn"],
tfrt_enabled = True,
deps = [
":util",
"//tensorflow/python:array_ops",
@ -154,6 +160,7 @@ cuda_py_test(
name = "random_grad_test",
size = "small",
srcs = ["random_grad_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
@ -174,6 +181,7 @@ tf_py_test(
srcs = ["random_binomial_test.py"],
shard_count = 3,
tags = ["no_oss"],
tfrt_enabled = True,
deps = [
":util",
"//tensorflow/python:array_ops",
@ -190,6 +198,7 @@ cuda_py_test(
name = "random_poisson_test",
size = "medium",
srcs = ["random_poisson_test.py"],
tfrt_enabled = True,
deps = [
":util",
"//tensorflow/python:array_ops",

3
tensorflow/python/kernel_tests/v1_compat_tests/BUILD
View File

@ -12,6 +12,7 @@ tf_py_test(
name = "identity_op_py_test",
size = "small",
srcs = ["identity_op_py_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:array_ops_gen",
@ -23,6 +24,7 @@ cuda_py_test(
name = "scatter_nd_ops_test",
size = "small",
srcs = ["scatter_nd_ops_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:state_ops",
@ -35,6 +37,7 @@ cuda_py_test(
name = "session_ops_test",
size = "small",
srcs = ["session_ops_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",

1
tensorflow/python/module/BUILD
View File

@ -23,6 +23,7 @@ py_library(
tf_py_test(
name = "module_test",
srcs = ["module_test.py"],
tfrt_enabled = True,
deps = [
":module",
"//tensorflow/python:client_testlib",

3
tensorflow/python/ops/numpy_ops/BUILD
View File

@ -43,6 +43,7 @@ py_library(
cuda_py_test(
name = "np_arrays_test",
srcs = ["np_arrays_test.py"],
tfrt_enabled = True,
deps = [
":numpy",
"//tensorflow/python:dtypes",
@ -68,6 +69,7 @@ cuda_py_test(
cuda_py_test(
name = "np_logic_test",
srcs = ["np_logic_test.py"],
tfrt_enabled = True,
deps = [
":numpy",
"//third_party/py/numpy",
@ -100,6 +102,7 @@ cuda_py_test(
cuda_py_test(
name = "np_utils_test",
srcs = ["np_utils_test.py"],
tfrt_enabled = True,
deps = [
":numpy",
"//tensorflow/python:platform",

1
tensorflow/python/ops/v1_compat_tests/BUILD
View File

@ -10,6 +10,7 @@ cuda_py_test(
size = "medium",
srcs = ["gradient_checker_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",

4
tensorflow/python/profiler/BUILD
View File

@ -36,6 +36,7 @@ cuda_py_test(
srcs = ["profiler_client_test.py"],
python_version = "PY3",
tags = ["no_pip"],
tfrt_enabled = True,
deps = [
":profiler_client",
"//tensorflow/python/eager:test",
@ -63,6 +64,7 @@ cuda_py_test(
"no_pip",
"no_rocm",
],
tfrt_enabled = True,
deps = [
":profiler_v2",
"//tensorflow/python:constant_op",
@ -122,6 +124,7 @@ cuda_py_test(
srcs = ["profiler_test.py"],
python_version = "PY3",
tags = ["no_pip"],
tfrt_enabled = True,
xla_enable_strict_auto_jit = False, # Node names are different with autojit
deps = [
":model_analyzer",
@ -182,6 +185,7 @@ cuda_py_test(
"no_gpu", # b/136036359
"no_pip",
],
tfrt_enabled = True,
xla_enable_strict_auto_jit = False, # Node names are different with autojit
deps = [
":profile_context",

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

@ -69,6 +69,7 @@ cuda_py_test(
"no_gpu", # b/138442728
"no_pip",
],
tfrt_enabled = True,
xla_enable_strict_auto_jit = False, # Node names are different with autojit
deps = [
":model_analyzer_testlib",

9
tensorflow/python/saved_model/BUILD
View File

@ -107,6 +107,7 @@ tf_py_test(
name = "loader_test",
size = "small",
srcs = ["loader_test.py"],
tfrt_enabled = True,
deps = [
":builder",
":loader",
@ -165,6 +166,7 @@ tf_py_test(
srcs = ["saved_model_test.py"],
data = ["//tensorflow/cc/saved_model:saved_model_half_plus_two"],
tags = ["no_windows"],
tfrt_enabled = True,
deps = [
":builder",
":constants",
@ -215,6 +217,7 @@ tf_py_test(
name = "utils_test",
size = "small",
srcs = ["utils_test.py"],
tfrt_enabled = True,
deps = [
":utils",
"//tensorflow/core:protos_all_py",
@ -247,6 +250,7 @@ tf_py_test(
name = "signature_def_utils_test",
size = "small",
srcs = ["signature_def_utils_test.py"],
tfrt_enabled = True,
deps = [
":signature_constants",
":signature_def_utils",
@ -262,6 +266,7 @@ tf_py_test(
name = "simple_save_test",
size = "small",
srcs = ["simple_save_test.py"],
tfrt_enabled = True,
deps = [
":loader",
":signature_constants",
@ -306,6 +311,7 @@ tf_py_test(
name = "save_context_test",
srcs = ["save_context_test.py"],
srcs_version = "PY2AND3",
tfrt_enabled = True,
deps = [
":save_context",
":save_options",
@ -525,6 +531,7 @@ py_strict_library(
tf_py_test(
name = "revived_types_test",
srcs = ["revived_types_test.py"],
tfrt_enabled = True,
deps = [
":revived_types",
"//tensorflow/core:protos_all_py",
@ -599,6 +606,7 @@ py_strict_library(
tf_py_test(
name = "nested_structure_coder_test",
srcs = ["nested_structure_coder_test.py"],
tfrt_enabled = True,
deps = [
":nested_structure_coder",
"//tensorflow/core:protos_all_py",
@ -643,6 +651,7 @@ py_strict_library(
tf_py_test(
name = "method_name_updater_test",
srcs = ["method_name_updater_test.py"],
tfrt_enabled = True,
deps = [
":method_name_updater",
"//tensorflow/core:protos_all_py",

8
tensorflow/python/tpu/BUILD
View File

@ -336,6 +336,7 @@ tf_py_test(
"no_oss", # TODO(b/131157871): Reenable in OSS when fixed
"no_windows", # TODO: needs investigation on Windows
],
tfrt_enabled = True,
deps = [
":tpu",
"//tensorflow/python:client_testlib",
@ -349,6 +350,7 @@ tf_py_test(
name = "tpu_sharding_test",
size = "small",
srcs = ["tpu_sharding_test.py"],
tfrt_enabled = True,
deps = [
":tpu",
"//tensorflow/python:client_testlib",
@ -360,6 +362,7 @@ tf_py_test(
name = "bfloat16_test",
size = "small",
srcs = ["bfloat16_test.py"],
tfrt_enabled = True,
deps = [
":tpu",
"//tensorflow/python:client_testlib",
@ -371,6 +374,7 @@ tf_py_test(
name = "tpu_infeed_test",
size = "small",
srcs = ["tpu_infeed_test.py"],
tfrt_enabled = True,
deps = [
":tpu",
"//tensorflow/python:framework",
@ -382,6 +386,7 @@ tf_py_test(
name = "topology_test",
size = "medium",
srcs = ["topology_test.py"],
tfrt_enabled = True,
deps = [
":tpu",
"//tensorflow/python:framework_test_lib",
@ -458,6 +463,7 @@ tf_py_test(
"feature_column_test.py",
],
main = "feature_column_test.py",
tfrt_enabled = True,
deps = [
":feature_column",
"//tensorflow/python:client_testlib",
@ -480,6 +486,7 @@ tf_py_test(
"feature_column_v2_test.py",
],
main = "feature_column_v2_test.py",
tfrt_enabled = True,
deps = [
":feature_column_v2",
"//tensorflow/python:client_testlib",
@ -613,6 +620,7 @@ tf_py_test(
],
python_version = "PY3",
srcs_version = "PY2AND3",
tfrt_enabled = True,
deps = [
":tpu_embedding_v2",
"//tensorflow/python/compat:v2_compat",

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

@ -43,6 +43,7 @@ tf_py_test(
tags = [
"no_oss_py2",
],
tfrt_enabled = True,
deps = [
":client",
"//tensorflow/python:client_testlib",

2
tensorflow/python/training/BUILD
View File

@ -840,6 +840,7 @@ cuda_py_test(
"checkpoint_management_test.py",
],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":checkpoint_management",
":saver",
@ -907,6 +908,7 @@ cuda_py_test(
],
python_version = "PY3",
tags = ["multi_gpu"],
tfrt_enabled = True,
deps = [
":adam",
":checkpoint_management",

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

@ -57,6 +57,7 @@ py_library(
tf_py_test(
name = "tracking_test",
srcs = ["tracking_test.py"],
tfrt_enabled = True,
deps = [
":base",
":tracking",
@ -158,6 +159,7 @@ tf_py_test(
name = "util_test",
srcs = ["util_test.py"],
tags = ["notsan"], # b/74395663
tfrt_enabled = True,
deps = [
":base",
":graph_view",
@ -198,6 +200,7 @@ tf_py_test(
tags = [
"notsan", # b/74395663
],
tfrt_enabled = True,
deps = [
":tracking",
":util",
@ -240,6 +243,7 @@ tf_py_test(
tf_py_test(
name = "benchmarks_test",
srcs = ["benchmarks_test.py"],
tfrt_enabled = True,
deps = [
":util",
"//tensorflow/python:framework_ops",