diff --git a/tensorflow/python/autograph/impl/BUILD b/tensorflow/python/autograph/impl/BUILD index 866314f5b76..30ee67d41bc 100644 --- a/tensorflow/python/autograph/impl/BUILD +++ b/tensorflow/python/autograph/impl/BUILD @@ -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", diff --git a/tensorflow/python/compat/BUILD b/tensorflow/python/compat/BUILD index 70f492175cb..1c650ea5499 100644 --- a/tensorflow/python/compat/BUILD +++ b/tensorflow/python/compat/BUILD @@ -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", diff --git a/tensorflow/python/compiler/tensorrt/BUILD b/tensorflow/python/compiler/tensorrt/BUILD index b33eaa847e4..77c98bc61d6 100644 --- a/tensorflow/python/compiler/tensorrt/BUILD +++ b/tensorflow/python/compiler/tensorrt/BUILD @@ -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", diff --git a/tensorflow/python/compiler/xla/BUILD b/tensorflow/python/compiler/xla/BUILD index 5f4e27b47cb..5d92e43c61a 100644 --- a/tensorflow/python/compiler/xla/BUILD +++ b/tensorflow/python/compiler/xla/BUILD @@ -99,6 +99,7 @@ cuda_py_test( "no_mac", "no_windows", ], + tfrt_enabled = True, xla_enabled = True, deps = [ "//tensorflow/python:client_testlib", diff --git a/tensorflow/python/data/benchmarks/BUILD b/tensorflow/python/data/benchmarks/BUILD index 3f0faf5364a..94c189c43f1 100644 --- a/tensorflow/python/data/benchmarks/BUILD +++ b/tensorflow/python/data/benchmarks/BUILD @@ -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", diff --git a/tensorflow/python/data/experimental/benchmarks/BUILD b/tensorflow/python/data/experimental/benchmarks/BUILD index e3ca2d52ab5..a3ceb9ed37f 100644 --- a/tensorflow/python/data/experimental/benchmarks/BUILD +++ b/tensorflow/python/data/experimental/benchmarks/BUILD @@ -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", diff --git a/tensorflow/python/data/experimental/kernel_tests/BUILD b/tensorflow/python/data/experimental/kernel_tests/BUILD index 053771255f4..84e21a7b536 100644 --- a/tensorflow/python/data/experimental/kernel_tests/BUILD +++ b/tensorflow/python/data/experimental/kernel_tests/BUILD @@ -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", diff --git a/tensorflow/python/data/experimental/kernel_tests/serialization/BUILD b/tensorflow/python/data/experimental/kernel_tests/serialization/BUILD index e2c7d3ced2c..e0be365d414 100644 --- a/tensorflow/python/data/experimental/kernel_tests/serialization/BUILD +++ b/tensorflow/python/data/experimental/kernel_tests/serialization/BUILD @@ -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", diff --git a/tensorflow/python/data/experimental/service/BUILD b/tensorflow/python/data/experimental/service/BUILD index 020e5a91e29..b6096b4b0bb 100644 --- a/tensorflow/python/data/experimental/service/BUILD +++ b/tensorflow/python/data/experimental/service/BUILD @@ -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", diff --git a/tensorflow/python/data/kernel_tests/BUILD b/tensorflow/python/data/kernel_tests/BUILD index 44937b41889..15ee2c6d500 100644 --- a/tensorflow/python/data/kernel_tests/BUILD +++ b/tensorflow/python/data/kernel_tests/BUILD @@ -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", diff --git a/tensorflow/python/debug/BUILD b/tensorflow/python/debug/BUILD index 36fdb20aeae..176d2546525 100644 --- a/tensorflow/python/debug/BUILD +++ b/tensorflow/python/debug/BUILD @@ -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", diff --git a/tensorflow/python/distribute/BUILD b/tensorflow/python/distribute/BUILD index 18d66d687f2..8503119b305 100644 --- a/tensorflow/python/distribute/BUILD +++ b/tensorflow/python/distribute/BUILD @@ -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", diff --git a/tensorflow/python/distribute/client/BUILD b/tensorflow/python/distribute/client/BUILD index 4827df56214..157657fcf30 100644 --- a/tensorflow/python/distribute/client/BUILD +++ b/tensorflow/python/distribute/client/BUILD @@ -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", diff --git a/tensorflow/python/distribute/cluster_resolver/BUILD b/tensorflow/python/distribute/cluster_resolver/BUILD index e925104ffa3..6fda3f5311d 100644 --- a/tensorflow/python/distribute/cluster_resolver/BUILD +++ b/tensorflow/python/distribute/cluster_resolver/BUILD @@ -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", diff --git a/tensorflow/python/distribute/cluster_resolver/tpu/BUILD b/tensorflow/python/distribute/cluster_resolver/tpu/BUILD index dff2b6937f7..01b21f73dee 100644 --- a/tensorflow/python/distribute/cluster_resolver/tpu/BUILD +++ b/tensorflow/python/distribute/cluster_resolver/tpu/BUILD @@ -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", diff --git a/tensorflow/python/keras/BUILD b/tensorflow/python/keras/BUILD index 0f9ab84a270..53ac4c23d01 100755 --- a/tensorflow/python/keras/BUILD +++ b/tensorflow/python/keras/BUILD @@ -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", diff --git a/tensorflow/python/keras/applications/BUILD b/tensorflow/python/keras/applications/BUILD index 4e9fa04fcf1..aba936c2a8f 100644 --- a/tensorflow/python/keras/applications/BUILD +++ b/tensorflow/python/keras/applications/BUILD @@ -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", diff --git a/tensorflow/python/keras/benchmarks/BUILD b/tensorflow/python/keras/benchmarks/BUILD index f12a1a86f29..a2209513b62 100644 --- a/tensorflow/python/keras/benchmarks/BUILD +++ b/tensorflow/python/keras/benchmarks/BUILD @@ -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", diff --git a/tensorflow/python/keras/benchmarks/saved_model_benchmarks/BUILD b/tensorflow/python/keras/benchmarks/saved_model_benchmarks/BUILD index a1bd5f32fa6..4debe1d0eac 100644 --- a/tensorflow/python/keras/benchmarks/saved_model_benchmarks/BUILD +++ b/tensorflow/python/keras/benchmarks/saved_model_benchmarks/BUILD @@ -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", diff --git a/tensorflow/python/keras/distribute/BUILD b/tensorflow/python/keras/distribute/BUILD index ef6c098a535..be6fd742f4f 100644 --- a/tensorflow/python/keras/distribute/BUILD +++ b/tensorflow/python/keras/distribute/BUILD @@ -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", diff --git a/tensorflow/python/keras/engine/BUILD b/tensorflow/python/keras/engine/BUILD index 258dc2f1290..4eef969bedf 100644 --- a/tensorflow/python/keras/engine/BUILD +++ b/tensorflow/python/keras/engine/BUILD @@ -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", diff --git a/tensorflow/python/keras/layers/BUILD b/tensorflow/python/keras/layers/BUILD index 6806babe5ba..677f96c3d30 100644 --- a/tensorflow/python/keras/layers/BUILD +++ b/tensorflow/python/keras/layers/BUILD @@ -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", diff --git a/tensorflow/python/keras/layers/preprocessing/BUILD b/tensorflow/python/keras/layers/preprocessing/BUILD index 7ca29cdb1a3..34a644b70d2 100644 --- a/tensorflow/python/keras/layers/preprocessing/BUILD +++ b/tensorflow/python/keras/layers/preprocessing/BUILD @@ -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", diff --git a/tensorflow/python/keras/layers/preprocessing/benchmarks/BUILD b/tensorflow/python/keras/layers/preprocessing/benchmarks/BUILD index ab2dd49ce1b..7c8d904acb1 100644 --- a/tensorflow/python/keras/layers/preprocessing/benchmarks/BUILD +++ b/tensorflow/python/keras/layers/preprocessing/benchmarks/BUILD @@ -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", diff --git a/tensorflow/python/keras/legacy_tf_layers/BUILD b/tensorflow/python/keras/legacy_tf_layers/BUILD index be93336b833..df24b618efa 100644 --- a/tensorflow/python/keras/legacy_tf_layers/BUILD +++ b/tensorflow/python/keras/legacy_tf_layers/BUILD @@ -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", diff --git a/tensorflow/python/keras/mixed_precision/experimental/BUILD b/tensorflow/python/keras/mixed_precision/experimental/BUILD index 1d20febae46..d41b3686512 100644 --- a/tensorflow/python/keras/mixed_precision/experimental/BUILD +++ b/tensorflow/python/keras/mixed_precision/experimental/BUILD @@ -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", diff --git a/tensorflow/python/keras/optimizer_v2/BUILD b/tensorflow/python/keras/optimizer_v2/BUILD index 566c051b727..b208e2e1e1e 100644 --- a/tensorflow/python/keras/optimizer_v2/BUILD +++ b/tensorflow/python/keras/optimizer_v2/BUILD @@ -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", diff --git a/tensorflow/python/keras/preprocessing/BUILD b/tensorflow/python/keras/preprocessing/BUILD index 1b32e11f8c9..80bc1d5459d 100644 --- a/tensorflow/python/keras/preprocessing/BUILD +++ b/tensorflow/python/keras/preprocessing/BUILD @@ -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", diff --git a/tensorflow/python/keras/saving/BUILD b/tensorflow/python/keras/saving/BUILD index c384ce33993..18aa8990dc0 100644 --- a/tensorflow/python/keras/saving/BUILD +++ b/tensorflow/python/keras/saving/BUILD @@ -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", diff --git a/tensorflow/python/keras/tests/BUILD b/tensorflow/python/keras/tests/BUILD index d6f07484160..f6db1b7ac32 100644 --- a/tensorflow/python/keras/tests/BUILD +++ b/tensorflow/python/keras/tests/BUILD @@ -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", diff --git a/tensorflow/python/keras/utils/BUILD b/tensorflow/python/keras/utils/BUILD index 79c99cc1870..58be8566db3 100644 --- a/tensorflow/python/keras/utils/BUILD +++ b/tensorflow/python/keras/utils/BUILD @@ -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", diff --git a/tensorflow/python/keras/wrappers/BUILD b/tensorflow/python/keras/wrappers/BUILD index d00ce9ceab8..1122bcbdac9 100644 --- a/tensorflow/python/keras/wrappers/BUILD +++ b/tensorflow/python/keras/wrappers/BUILD @@ -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", diff --git a/tensorflow/python/kernel_tests/BUILD b/tensorflow/python/kernel_tests/BUILD index 59baf0a26fb..0b9fbeb0211 100644 --- a/tensorflow/python/kernel_tests/BUILD +++ b/tensorflow/python/kernel_tests/BUILD @@ -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", diff --git a/tensorflow/python/kernel_tests/array_ops/BUILD b/tensorflow/python/kernel_tests/array_ops/BUILD index 42fa2fef81e..bd06627a707 100644 --- a/tensorflow/python/kernel_tests/array_ops/BUILD +++ b/tensorflow/python/kernel_tests/array_ops/BUILD @@ -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", diff --git a/tensorflow/python/kernel_tests/boosted_trees/BUILD b/tensorflow/python/kernel_tests/boosted_trees/BUILD index 5b318324d4c..e7a34382355 100644 --- a/tensorflow/python/kernel_tests/boosted_trees/BUILD +++ b/tensorflow/python/kernel_tests/boosted_trees/BUILD @@ -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", diff --git a/tensorflow/python/kernel_tests/distributions/BUILD b/tensorflow/python/kernel_tests/distributions/BUILD index fc1c9af6946..ff530da88a6 100644 --- a/tensorflow/python/kernel_tests/distributions/BUILD +++ b/tensorflow/python/kernel_tests/distributions/BUILD @@ -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", diff --git a/tensorflow/python/kernel_tests/linalg/BUILD b/tensorflow/python/kernel_tests/linalg/BUILD index 097183d1025..67d1adba3a9 100644 --- a/tensorflow/python/kernel_tests/linalg/BUILD +++ b/tensorflow/python/kernel_tests/linalg/BUILD @@ -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", diff --git a/tensorflow/python/kernel_tests/linalg/sparse/BUILD b/tensorflow/python/kernel_tests/linalg/sparse/BUILD index 96ebc38ce5a..560ba7b2fd4 100644 --- a/tensorflow/python/kernel_tests/linalg/sparse/BUILD +++ b/tensorflow/python/kernel_tests/linalg/sparse/BUILD @@ -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", ], diff --git a/tensorflow/python/kernel_tests/proto/BUILD b/tensorflow/python/kernel_tests/proto/BUILD index 0e935dfe8c4..cc6147a8fd2 100644 --- a/tensorflow/python/kernel_tests/proto/BUILD +++ b/tensorflow/python/kernel_tests/proto/BUILD @@ -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", diff --git a/tensorflow/python/kernel_tests/random/BUILD b/tensorflow/python/kernel_tests/random/BUILD index 06360fc2095..6add4128371 100644 --- a/tensorflow/python/kernel_tests/random/BUILD +++ b/tensorflow/python/kernel_tests/random/BUILD @@ -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", diff --git a/tensorflow/python/kernel_tests/v1_compat_tests/BUILD b/tensorflow/python/kernel_tests/v1_compat_tests/BUILD index bd9c02d8101..ac04803ba3b 100644 --- a/tensorflow/python/kernel_tests/v1_compat_tests/BUILD +++ b/tensorflow/python/kernel_tests/v1_compat_tests/BUILD @@ -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", diff --git a/tensorflow/python/module/BUILD b/tensorflow/python/module/BUILD index fea6fe123ad..1b93d8cf8b6 100644 --- a/tensorflow/python/module/BUILD +++ b/tensorflow/python/module/BUILD @@ -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", diff --git a/tensorflow/python/ops/numpy_ops/BUILD b/tensorflow/python/ops/numpy_ops/BUILD index 96eb5509de3..d22c96e50c8 100644 --- a/tensorflow/python/ops/numpy_ops/BUILD +++ b/tensorflow/python/ops/numpy_ops/BUILD @@ -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", diff --git a/tensorflow/python/ops/v1_compat_tests/BUILD b/tensorflow/python/ops/v1_compat_tests/BUILD index 37bff01d429..3f44e5208a2 100644 --- a/tensorflow/python/ops/v1_compat_tests/BUILD +++ b/tensorflow/python/ops/v1_compat_tests/BUILD @@ -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", diff --git a/tensorflow/python/profiler/BUILD b/tensorflow/python/profiler/BUILD index 9996f5c9894..622e14616ab 100644 --- a/tensorflow/python/profiler/BUILD +++ b/tensorflow/python/profiler/BUILD @@ -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", diff --git a/tensorflow/python/profiler/internal/BUILD b/tensorflow/python/profiler/internal/BUILD index e471a4b97f6..8cdf8faaaf5 100644 --- a/tensorflow/python/profiler/internal/BUILD +++ b/tensorflow/python/profiler/internal/BUILD @@ -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", diff --git a/tensorflow/python/saved_model/BUILD b/tensorflow/python/saved_model/BUILD index 9c1cc090d7d..b18c7c1e738 100644 --- a/tensorflow/python/saved_model/BUILD +++ b/tensorflow/python/saved_model/BUILD @@ -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", diff --git a/tensorflow/python/tpu/BUILD b/tensorflow/python/tpu/BUILD index b2a60087a3e..1d569211415 100644 --- a/tensorflow/python/tpu/BUILD +++ b/tensorflow/python/tpu/BUILD @@ -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", diff --git a/tensorflow/python/tpu/client/BUILD b/tensorflow/python/tpu/client/BUILD index dc94bffb64e..a6973d4ec22 100644 --- a/tensorflow/python/tpu/client/BUILD +++ b/tensorflow/python/tpu/client/BUILD @@ -43,6 +43,7 @@ tf_py_test( tags = [ "no_oss_py2", ], + tfrt_enabled = True, deps = [ ":client", "//tensorflow/python:client_testlib", diff --git a/tensorflow/python/training/BUILD b/tensorflow/python/training/BUILD index 0e864b176d6..53e35b7e3e8 100644 --- a/tensorflow/python/training/BUILD +++ b/tensorflow/python/training/BUILD @@ -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", diff --git a/tensorflow/python/training/tracking/BUILD b/tensorflow/python/training/tracking/BUILD index 370b78c84f5..6001dc2cbbe 100644 --- a/tensorflow/python/training/tracking/BUILD +++ b/tensorflow/python/training/tracking/BUILD @@ -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",