From aab4cde23e160ed6ef80cab53e5eba44708e6396 Mon Sep 17 00:00:00 2001
From: Kibeom Kim <kkb@google.com>
Date: Wed, 23 Sep 2020 11:38:43 -0700
Subject: [PATCH] Enable more TFRT test targets that are newly passing
PiperOrigin-RevId: 333337050
Change-Id: I53a9af35c09f9ddb9ed6dc47bedfce2b8662205b
---
tensorflow/python/autograph/impl/BUILD | 2 +
tensorflow/python/compat/BUILD | 2 +
tensorflow/python/compiler/tensorrt/BUILD | 2 +
tensorflow/python/compiler/xla/BUILD | 1 +
tensorflow/python/data/benchmarks/BUILD | 8 ++++
.../python/data/experimental/benchmarks/BUILD | 13 ++++++
.../data/experimental/kernel_tests/BUILD | 4 ++
.../kernel_tests/serialization/BUILD | 44 +++++++++++++++++++
.../python/data/experimental/service/BUILD | 1 +
tensorflow/python/data/kernel_tests/BUILD | 3 ++
tensorflow/python/debug/BUILD | 6 +++
tensorflow/python/distribute/BUILD | 1 +
tensorflow/python/distribute/client/BUILD | 1 +
.../python/distribute/cluster_resolver/BUILD | 6 +++
.../distribute/cluster_resolver/tpu/BUILD | 1 +
tensorflow/python/keras/BUILD | 8 ++++
tensorflow/python/keras/applications/BUILD | 17 +++++++
tensorflow/python/keras/benchmarks/BUILD | 11 +++++
.../benchmarks/saved_model_benchmarks/BUILD | 8 ++++
tensorflow/python/keras/distribute/BUILD | 3 ++
tensorflow/python/keras/engine/BUILD | 5 +++
tensorflow/python/keras/layers/BUILD | 7 +++
.../python/keras/layers/preprocessing/BUILD | 1 +
.../layers/preprocessing/benchmarks/BUILD | 6 +++
.../python/keras/legacy_tf_layers/BUILD | 5 +++
.../keras/mixed_precision/experimental/BUILD | 3 ++
tensorflow/python/keras/optimizer_v2/BUILD | 8 ++++
tensorflow/python/keras/preprocessing/BUILD | 2 +
tensorflow/python/keras/saving/BUILD | 2 +
tensorflow/python/keras/tests/BUILD | 10 +++++
tensorflow/python/keras/utils/BUILD | 11 +++++
tensorflow/python/keras/wrappers/BUILD | 1 +
tensorflow/python/kernel_tests/BUILD | 2 +
.../python/kernel_tests/array_ops/BUILD | 4 ++
.../python/kernel_tests/boosted_trees/BUILD | 5 +++
.../python/kernel_tests/distributions/BUILD | 7 +++
tensorflow/python/kernel_tests/linalg/BUILD | 2 +
.../python/kernel_tests/linalg/sparse/BUILD | 5 +++
tensorflow/python/kernel_tests/proto/BUILD | 3 ++
tensorflow/python/kernel_tests/random/BUILD | 9 ++++
.../python/kernel_tests/v1_compat_tests/BUILD | 3 ++
tensorflow/python/module/BUILD | 1 +
tensorflow/python/ops/numpy_ops/BUILD | 3 ++
tensorflow/python/ops/v1_compat_tests/BUILD | 1 +
tensorflow/python/profiler/BUILD | 4 ++
tensorflow/python/profiler/internal/BUILD | 1 +
tensorflow/python/saved_model/BUILD | 9 ++++
tensorflow/python/tpu/BUILD | 8 ++++
tensorflow/python/tpu/client/BUILD | 1 +
tensorflow/python/training/BUILD | 2 +
tensorflow/python/training/tracking/BUILD | 4 ++
51 files changed, 277 insertions(+)
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",