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

Remove deprecated tfrt_enabled test target flag.

Browse Source 
PiperOrigin-RevId: 338530097
Change-Id: I0bd2ad366210330ece06f99a4fdb16de395ece05
This commit is contained in:
Kibeom Kim 2020-10-22 12:47:42 -07:00 committed by TensorFlower Gardener
parent 652de04b91
commit 0c67638ac2
53 changed files with 1 additions and 544 deletions
tensorflow/python
BUILD
autograph/impl
BUILD
compat
BUILD
compiler
tensorrt
BUILD
xla
BUILD
data
benchmarks
BUILD
experimental
benchmarks
BUILD
kernel_tests
BUILD
serialization
BUILD
service
BUILD
kernel_tests
BUILD
debug
BUILD
distribute
BUILD
cluster_resolver
BUILD
tpu
BUILD
eager
BUILD
benchmarks/resnet50
BUILD
framework/experimental
BUILD
keras
BUILD
applications
BUILD
benchmarks
BUILD
saved_model_benchmarks
BUILD
distribute
BUILD
engine
BUILD
layers
BUILD
preprocessing
BUILD
benchmarks
BUILD
legacy_tf_layers
BUILD
optimizer_v2
BUILD
preprocessing
BUILD
saving
BUILD
tests
BUILD
utils
BUILD
kernel_tests
BUILD
array_ops
BUILD
boosted_trees
BUILD
distributions
BUILD
linalg
BUILD
sparse
BUILD
proto
BUILD
random
BUILD
v1_compat_tests
BUILD
module
BUILD
ops
numpy_ops
BUILD
v1_compat_tests
BUILD
profiler
BUILD
internal
BUILD
saved_model
BUILD
tpu
BUILD
client
BUILD
training
BUILD
experimental
BUILD
tracking
BUILD

104
tensorflow/python/BUILD
View File

@ -358,7 +358,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":platform",
":platform_test",
@ -377,7 +376,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":platform",
":platform_test",
@ -389,7 +387,6 @@ tf_py_test(
size = "small",
srcs = ["platform/flags_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client_testlib",
":platform",
@ -405,7 +402,6 @@ tf_py_test(
"no_windows",
"nomac",
],
tfrt_enabled = True,
deps = [
":client_testlib",
":platform",
@ -1191,7 +1187,6 @@ tf_py_test(
name = "decorator_utils_test",
srcs = ["util/decorator_utils_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client_testlib",
":platform",
@ -1203,7 +1198,6 @@ tf_py_test(
name = "deprecation_test",
srcs = ["util/deprecation_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client_testlib",
":platform",
@ -1215,7 +1209,6 @@ tf_py_test(
name = "dispatch_test",
srcs = ["util/dispatch_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client_testlib",
":platform",
@ -1227,7 +1220,6 @@ tf_py_test(
name = "keyword_args_test",
srcs = ["util/keyword_args_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client_testlib",
":util",
@ -1566,7 +1558,6 @@ tf_py_test(
srcs = ["framework/function_def_to_graph_test.py"],
python_version = "PY3",
tags = ["no_pip"],
tfrt_enabled = True,
deps = [
":array_ops",
":client_testlib",
@ -1695,7 +1686,6 @@ tf_py_test(
srcs = ["framework/py_context_manager_test.py"],
python_version = "PY3",
tags = ["no_pip"],
tfrt_enabled = True,
deps = [
":_py_context_manager",
],
@ -1750,7 +1740,6 @@ tf_py_test(
srcs = ["framework/op_def_util_test.py"],
python_version = "PY3",
tags = ["no_pip"],
tfrt_enabled = True,
)
cc_library(
@ -2003,7 +1992,6 @@ tf_py_test(
size = "small",
srcs = ["framework/smart_cond_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client_testlib",
":constant_op",
@ -2076,7 +2064,6 @@ tf_py_test(
srcs = ["framework/composite_tensor_utils_test.py"],
main = "framework/composite_tensor_utils_test.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":array_ops",
":composite_tensor",
@ -2339,7 +2326,6 @@ tf_py_test(
srcs = ["framework/constant_op_test.py"],
main = "framework/constant_op_test.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":constant_op",
],
@ -2351,7 +2337,6 @@ tf_py_test(
srcs = ["framework/registry_test.py"],
main = "framework/registry_test.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client_testlib",
":framework_for_generated_wrappers",
@ -2365,7 +2350,6 @@ tf_py_test(
srcs = ["framework/errors_test.py"],
main = "framework/errors_test.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client_testlib",
":errors",
@ -2379,7 +2363,6 @@ tf_py_test(
srcs = ["framework/error_interpolation_test.py"],
main = "framework/error_interpolation_test.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client_testlib",
":constant_op",
@ -2394,7 +2377,6 @@ tf_py_test(
srcs = ["framework/subscribe_test.py"],
main = "framework/subscribe_test.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":framework",
":framework_for_generated_wrappers",
@ -2437,7 +2419,6 @@ tf_py_test(
tags = [
"no_pip",
],
tfrt_enabled = True,
deps = [
":client_testlib",
":platform",
@ -2450,7 +2431,6 @@ tf_py_test(
srcs = ["framework/proto_test.py"],
main = "framework/proto_test.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client_testlib",
":framework_for_generated_wrappers",
@ -2539,7 +2519,6 @@ tf_py_test(
srcs = ["framework/versions_test.py"],
main = "framework/versions_test.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client_testlib",
":framework_for_generated_wrappers",
@ -2588,7 +2567,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":array_ops",
":client_testlib",
@ -2612,7 +2590,6 @@ tf_py_test(
srcs = ["framework/traceable_stack_test.py"],
main = "framework/traceable_stack_test.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":framework_test_lib",
":platform_test",
@ -2667,7 +2644,6 @@ tf_py_test(
srcs = ["framework/common_shapes_test.py"],
main = "framework/common_shapes_test.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":framework",
":framework_for_generated_wrappers",
@ -2684,7 +2660,6 @@ tf_py_test(
main = "framework/ops_test.py",
python_version = "PY3",
tags = ["no_pip"], # test_ops_2 is not available in pip.
tfrt_enabled = True,
deps = [
":cond_v2",
":control_flow_ops",
@ -2714,7 +2689,6 @@ tf_py_test(
srcs = ["framework/ops_enable_eager_test.py"],
main = "framework/ops_enable_eager_test.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":framework",
":platform_test",
@ -2728,7 +2702,6 @@ tf_py_test(
srcs = ["framework/tensor_shape_test.py"],
main = "framework/tensor_shape_test.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":framework_for_generated_wrappers",
":framework_test_lib",
@ -2744,7 +2717,6 @@ tf_py_test(
srcs = ["framework/type_spec_test.py"],
main = "framework/type_spec_test.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":framework_for_generated_wrappers",
":framework_test_lib",
@ -2760,7 +2732,6 @@ tf_py_test(
srcs = ["framework/tensor_spec_test.py"],
main = "framework/tensor_spec_test.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":framework_for_generated_wrappers",
":framework_test_lib",
@ -2793,7 +2764,6 @@ tf_py_test(
srcs = ["framework/device_spec_test.py"],
main = "framework/device_spec_test.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":framework_for_generated_wrappers",
":framework_test_lib",
@ -2808,7 +2778,6 @@ tf_py_test(
srcs = ["framework/device_test.py"],
main = "framework/device_test.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":framework_for_generated_wrappers",
":framework_test_lib",
@ -2823,7 +2792,6 @@ tf_py_test(
srcs = ["framework/random_seed_test.py"],
main = "framework/random_seed_test.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client_testlib",
":framework",
@ -2836,7 +2804,6 @@ tf_py_test(
srcs = ["framework/tensor_shape_div_test.py"],
main = "framework/tensor_shape_div_test.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":framework_for_generated_wrappers",
":framework_test_lib",
@ -2853,7 +2820,6 @@ tf_py_test(
main = "framework/tensor_util_test.py",
python_version = "PY3",
tags = ["no_windows"],
tfrt_enabled = True,
deps = [
":array_ops",
":client_testlib",
@ -2908,7 +2874,6 @@ tf_py_test(
"nomsan", # TODO(b/149948895): Re-enable.
"notsan", # TODO(b/149948895): Re-enable.
],
tfrt_enabled = True,
deps = [
":framework_test_lib",
# TODO(kkb): Find more appropriate place to add `memory_checker` as deps
@ -2934,7 +2899,6 @@ tf_py_test(
srcs = ["framework/dtypes_test.py"],
main = "framework/dtypes_test.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":framework_for_generated_wrappers",
":framework_test_lib",
@ -2950,7 +2914,6 @@ tf_py_test(
size = "small",
srcs = ["framework/op_def_library_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":framework_for_generated_wrappers",
":framework_test_lib",
@ -2964,7 +2927,6 @@ tf_py_test(
srcs = ["framework/kernels_test.py"],
main = "framework/kernels_test.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":framework_test_lib",
":kernels",
@ -3506,7 +3468,6 @@ tf_py_test(
size = "small",
srcs = ["ops/clip_ops_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client_testlib",
":clip_ops",
@ -3532,7 +3493,6 @@ tf_py_test(
size = "medium",
srcs = ["ops/clustering_ops_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client_testlib",
":clustering_ops",
@ -3577,7 +3537,6 @@ tf_py_test(
"no_windows",
"nomac",
],
tfrt_enabled = True,
xla_enable_strict_auto_jit = True,
deps = [
":client_testlib",
@ -3701,7 +3660,6 @@ tf_py_test(
size = "small",
srcs = ["ops/control_flow_v2_toggles_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client_testlib",
":control_flow_util_v2",
@ -3715,7 +3673,6 @@ tf_py_test(
size = "small",
srcs = ["ops/control_flow_v2_enable_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client_testlib",
":control_flow_util",
@ -3737,7 +3694,6 @@ tf_py_test(
"no_oss",
"no_pip",
],
tfrt_enabled = True,
deps = [
":client_testlib",
":control_flow_util",
@ -3820,7 +3776,6 @@ tf_py_test(
size = "small",
srcs = ["ops/bincount_ops_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":bincount_ops",
":platform_test",
@ -4699,7 +4654,6 @@ tf_py_test(
name = "sort_ops_test",
srcs = ["ops/sort_ops_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":array_ops",
":client_testlib",
@ -4799,7 +4753,6 @@ cuda_py_test(
name = "rnn_grad_test",
srcs = ["ops/rnn_grad_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":array_ops",
":client_testlib",
@ -5082,7 +5035,6 @@ cuda_py_test(
srcs = ["ops/bitwise_ops_test.py"],
python_version = "PY3",
tags = ["no_windows"],
tfrt_enabled = True,
deps = [
":bitwise_ops",
":constant_op",
@ -5140,7 +5092,6 @@ cuda_py_test(
size = "medium",
srcs = ["ops/gradient_checker_v2_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":array_ops",
":client_testlib",
@ -5195,7 +5146,6 @@ cuda_py_test(
size = "small",
srcs = ["ops/histogram_ops_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":array_ops",
":client_testlib",
@ -5213,7 +5163,6 @@ cuda_py_test(
srcs = ["ops/image_grad_deterministic_test.py"],
python_version = "PY3",
shard_count = 5,
tfrt_enabled = True,
deps = [
":image_grad_test_base",
],
@ -5225,7 +5174,6 @@ cuda_py_test(
srcs = ["ops/image_grad_test.py"],
python_version = "PY3",
shard_count = 5,
tfrt_enabled = True,
deps = [
":image_grad_test_base",
],
@ -5275,7 +5223,6 @@ cuda_py_test(
size = "small",
srcs = ["ops/init_ops_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client_testlib",
":framework_ops",
@ -5291,7 +5238,6 @@ cuda_py_test(
size = "medium",
srcs = ["ops/init_ops_v2_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":array_ops",
":client_testlib",
@ -5309,7 +5255,6 @@ cuda_py_test(
srcs = ["ops/math_grad_test.py"],
python_version = "PY3",
tags = ["no_windows_gpu"],
tfrt_enabled = True,
deps = [
":array_ops",
":client_testlib",
@ -5349,7 +5294,6 @@ cuda_py_test(
tags = [
"no_windows_gpu",
],
tfrt_enabled = True,
deps = [
":framework_for_generated_wrappers",
":framework_test_lib",
@ -5366,7 +5310,6 @@ cuda_py_test(
python_version = "PY3",
shard_count = 4,
tags = ["no_windows"],
tfrt_enabled = True,
deps = [
":array_ops",
":client_testlib",
@ -5471,7 +5414,6 @@ cuda_py_test(
"no_oss", # TODO(b/149565560)
"no_windows_gpu",
],
tfrt_enabled = True,
deps = [
":framework_for_generated_wrappers",
":framework_test_lib",
@ -5509,7 +5451,6 @@ tf_py_test(
size = "small",
srcs = ["ops/variable_spec_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":framework_for_generated_wrappers",
":framework_test_lib",
@ -5556,7 +5497,6 @@ tf_py_test(
name = "tf_export_test",
srcs = ["util/tf_export_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client_testlib",
":platform",
@ -5617,7 +5557,6 @@ tf_py_test(
name = "tf_stack_test",
srcs = ["util/tf_stack_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client_testlib",
":tf_export",
@ -5680,7 +5619,6 @@ tf_py_test(
"no_pip", # b/168621686
"no_windows", # b/169275019
],
tfrt_enabled = True,
deps = [
":_function_parameter_canonicalizer_binding_for_test",
":client_testlib",
@ -5740,7 +5678,6 @@ tf_py_test(
size = "small",
srcs = ["util/object_identity_test.py"],
python_version = "PY3",
tfrt_enabled = True,
)
# Placeholder for intenal nest_test comments.
@ -5750,7 +5687,6 @@ tf_py_test(
srcs = ["util/nest_test.py"],
main = "util/nest_test.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [":util_nest_test_main_lib"],
)
@ -5776,7 +5712,6 @@ tf_py_test(
srcs = ["util/serialization_test.py"],
main = "util/serialization_test.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client_testlib",
":util",
@ -5787,7 +5722,6 @@ tf_py_test(
name = "function_utils_test",
srcs = ["util/function_utils_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client_testlib",
":util",
@ -5799,7 +5733,6 @@ tf_py_test(
size = "small",
srcs = ["util/tf_contextlib_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client_testlib",
":util",
@ -5811,7 +5744,6 @@ tf_py_test(
size = "small",
srcs = ["util/tf_decorator_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client_testlib",
":util",
@ -5835,7 +5767,6 @@ tf_py_test(
size = "small",
srcs = ["util/tf_should_use_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client_testlib",
":tf_should_use",
@ -5847,7 +5778,6 @@ tf_py_test(
size = "small",
srcs = ["util/tf_inspect_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client_testlib",
":util",
@ -5872,7 +5802,6 @@ tf_py_test(
srcs = ["util/lock_util_test.py"],
main = "util/lock_util_test.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client_testlib",
":util",
@ -5885,7 +5814,6 @@ tf_py_test(
size = "small",
srcs = ["util/module_wrapper_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client_testlib",
":util",
@ -5929,7 +5857,6 @@ tf_py_test(
main = "util/protobuf/compare_test.py",
python_version = "PY3",
tags = ["no_pip"], # compare_test_pb2 proto is not available in pip.
tfrt_enabled = True,
deps = [
":compare_test_proto_py",
":platform_test",
@ -5944,7 +5871,6 @@ tf_py_test(
srcs = ["util/example_parser_configuration_test.py"],
main = "util/example_parser_configuration_test.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":array_ops",
":client",
@ -5960,7 +5886,6 @@ tf_py_test(
size = "small",
srcs = ["client/events_writer_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":errors",
":framework_test_lib",
@ -6481,7 +6406,6 @@ tf_py_test(
"no_pip_gpu",
"notsan", # data race due to b/62910646
],
tfrt_enabled = True,
deps = [
":client",
":framework",
@ -6501,7 +6425,6 @@ tf_py_test(
"no_gpu",
"no_windows",
],
tfrt_enabled = True,
deps = [
":array_ops",
":client",
@ -6526,7 +6449,6 @@ cuda_py_test(
"gpu_cupti",
"no_gpu", # b/154742661
],
tfrt_enabled = True,
xla_enable_strict_auto_jit = False, # Graph structure is different with autojit
deps = [
":client",
@ -6546,7 +6468,6 @@ cuda_py_test(
"no_gpu", # b/127386241
"no_windows_gpu",
],
tfrt_enabled = True,
deps = [
":client",
":client_testlib",
@ -6561,7 +6482,6 @@ tf_py_test(
size = "small",
srcs = ["framework/c_api_util_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":c_api_util",
":framework_test_lib",
@ -6574,7 +6494,6 @@ tf_py_test(
size = "small",
srcs = ["framework/graph_util_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client",
":client_testlib",
@ -6609,7 +6528,6 @@ tf_py_test(
size = "small",
srcs = ["lib/core/bfloat16_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client_testlib",
":lib",
@ -6626,7 +6544,6 @@ tf_py_test(
"no_rocm",
"no_windows",
],
tfrt_enabled = True,
deps = [
":client_testlib",
":errors",
@ -6639,7 +6556,6 @@ tf_py_test(
size = "small",
srcs = ["lib/io/tf_record_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client_testlib",
":errors",
@ -6844,7 +6760,6 @@ cuda_py_test(
main = "ops/accumulate_n_benchmark.py",
python_version = "PY3",
shard_count = 6,
tfrt_enabled = True,
deps = [
":array_ops",
":client",
@ -6864,7 +6779,6 @@ cuda_py_test(
srcs = ["ops/batch_norm_benchmark.py"],
main = "ops/batch_norm_benchmark.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":array_ops",
":client",
@ -6886,7 +6800,6 @@ cuda_py_test(
srcs = ["ops/collective_ops_benchmark.py"],
main = "ops/collective_ops_benchmark.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":array_ops",
":client",
@ -6904,7 +6817,6 @@ cuda_py_test(
srcs = ["ops/concat_benchmark.py"],
main = "ops/concat_benchmark.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":array_ops",
":client",
@ -6923,7 +6835,6 @@ cuda_py_test(
srcs = ["ops/control_flow_ops_benchmark.py"],
main = "ops/control_flow_ops_benchmark.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client_testlib",
":constant_op",
@ -6939,7 +6850,6 @@ cuda_py_test(
srcs = ["ops/conv2d_benchmark.py"],
main = "ops/conv2d_benchmark.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client",
":client_testlib",
@ -6960,7 +6870,6 @@ cuda_py_test(
srcs = ["ops/split_benchmark.py"],
main = "ops/split_benchmark.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":array_ops",
":client",
@ -6981,7 +6890,6 @@ cuda_py_test(
srcs = ["ops/transpose_benchmark.py"],
main = "ops/transpose_benchmark.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":array_ops",
":client",
@ -7002,7 +6910,6 @@ cuda_py_test(
srcs = ["ops/matmul_benchmark.py"],
main = "ops/matmul_benchmark.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [":matmul_benchmark_main_lib"],
)
@ -7032,7 +6939,6 @@ cuda_py_test(
grpc_enabled = True,
main = "client/session_benchmark.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":array_ops",
":client",
@ -7051,7 +6957,6 @@ cuda_py_test(
srcs = ["framework/graph_building_benchmark.py"],
main = "framework/graph_building_benchmark.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":array_ops",
":client_testlib",
@ -7067,7 +6972,6 @@ cuda_py_test(
size = "medium",
srcs = ["ops/nn_grad_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client_testlib",
":framework_for_generated_wrappers",
@ -7122,7 +7026,6 @@ tf_py_test(
"grappler",
"no_pip", # tf_optimizer is not available in pip.
],
tfrt_enabled = True,
deps = [
":client_testlib",
":framework_for_generated_wrappers",
@ -7143,7 +7046,6 @@ tf_py_test(
"grappler",
"no_pip", # tf_optimizer is not available in pip.
],
tfrt_enabled = True,
deps = [
":array_ops",
":client_testlib",
@ -7262,7 +7164,6 @@ tf_py_test(
"grappler",
"no_pip", # tf_optimizer is not available in pip.
],
tfrt_enabled = True,
deps = [
":client_testlib",
":framework_for_generated_wrappers",
@ -7284,7 +7185,6 @@ tf_py_test(
tags = [
"grappler",
],
tfrt_enabled = True,
deps = [
":client_testlib",
":framework_for_generated_wrappers",
@ -7420,7 +7320,6 @@ tf_py_test(
"no_pip",
"no_windows", # TODO(b/151942037)
],
tfrt_enabled = True,
deps = [
":array_ops",
":client_testlib",
@ -7455,7 +7354,6 @@ tf_py_test(
"grappler",
"no_pip",
],
tfrt_enabled = True,
deps = [
":array_ops",
":client_testlib",
@ -7476,7 +7374,6 @@ cuda_py_test(
],
python_version = "PY3",
tags = ["grappler"],
tfrt_enabled = True,
# This test analyzes the graph, but XLA changes the names of nodes.
xla_enable_strict_auto_jit = False,
deps = [
@ -7776,7 +7673,6 @@ cuda_py_test(
name = "raw_ops_test",
srcs = ["ops/raw_ops_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":client_testlib",
],

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

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

2
tensorflow/python/compat/BUILD
View File

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

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

@ -101,7 +101,6 @@ cuda_py_test(
"no_windows",
"nomac",
],
tfrt_enabled = True,
xla_enable_strict_auto_jit = False,
deps = [
":trt_convert_py",
@ -183,7 +182,6 @@ cuda_py_test(
"no_windows",
"nomac",
],
tfrt_enabled = True,
xla_enable_strict_auto_jit = False,
deps = [
":tf_trt_integration_test_base",

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

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

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

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

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

@ -25,7 +25,6 @@ py_binary(
tf_py_test(
name = "autotune_benchmark",
srcs = ["autotune_benchmark.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:math_ops",
@ -38,7 +37,6 @@ 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",
@ -52,7 +50,6 @@ 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",
@ -69,7 +66,6 @@ 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",
@ -85,7 +81,6 @@ 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",
@ -104,7 +99,6 @@ 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",
@ -120,7 +114,6 @@ 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",
@ -140,7 +133,6 @@ 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",
@ -156,7 +148,6 @@ 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",
@ -170,7 +161,6 @@ 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",
@ -186,7 +176,6 @@ 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",
@ -199,7 +188,6 @@ 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",
@ -218,7 +206,6 @@ 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",

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

@ -402,7 +402,6 @@ 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",
@ -740,7 +739,6 @@ tf_py_test(
tf_py_test(
name = "sleep_test",
srcs = ["sleep_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:util",

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

@ -40,7 +40,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -59,7 +58,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:array_ops",
@ -82,7 +80,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:array_ops",
@ -103,7 +100,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -122,7 +118,6 @@ tf_py_test(
"no_windows",
"notsan",
],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:constant_op",
@ -166,7 +161,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -184,7 +178,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -203,7 +196,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -222,7 +214,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -240,7 +231,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -259,7 +249,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -280,7 +269,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -298,7 +286,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -324,7 +311,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -343,7 +329,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -363,7 +348,6 @@ tf_py_test(
"no_windows",
"noasan", # TODO(b/337374867) fails with -fsanitize=null
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:array_ops",
@ -383,7 +367,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -404,7 +387,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -423,7 +405,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -447,7 +428,6 @@ tf_py_test(
tags = [
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -466,7 +446,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -484,7 +463,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -502,7 +480,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:array_ops",
@ -522,7 +499,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -543,7 +519,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -568,7 +543,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -585,7 +559,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -602,7 +575,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -626,7 +598,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -644,7 +615,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -663,7 +633,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -681,7 +650,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_combinations",
@ -702,7 +670,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -720,7 +687,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -738,7 +704,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -758,7 +723,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:array_ops",
@ -778,7 +742,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:array_ops",
@ -799,7 +762,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -819,7 +781,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -838,7 +799,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -856,7 +816,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -875,7 +834,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",
@ -893,7 +851,6 @@ tf_py_test(
"no_pip",
"no_windows",
],
tfrt_enabled = True,
deps = [
":dataset_serialization_test_base",
"//tensorflow/python:client_testlib",

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

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

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

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

6
tensorflow/python/debug/BUILD
View File

@ -797,7 +797,6 @@ 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",
@ -1042,7 +1041,6 @@ 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",
@ -1061,7 +1059,6 @@ 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",
@ -1080,7 +1077,6 @@ 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",
@ -1101,7 +1097,6 @@ 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",
@ -1224,7 +1219,6 @@ cuda_py_test(
srcs = ["cli/analyzer_cli_test.py"],
python_version = "PY3",
tags = ["no_windows"], # TODO: needs investigation on Windows
tfrt_enabled = True,
xla_enable_strict_auto_jit = False, # Node names are different with autojit
deps = [
":analyzer_cli",

1
tensorflow/python/distribute/BUILD
View File

@ -1047,7 +1047,6 @@ 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",

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

@ -98,7 +98,6 @@ 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",
@ -114,7 +113,6 @@ 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",
@ -131,7 +129,6 @@ 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",
@ -148,7 +145,6 @@ 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",
@ -165,7 +161,6 @@ 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",
@ -181,7 +176,6 @@ tf_py_test(
size = "small",
srcs = ["kubernetes_cluster_resolver_test.py"],
main = "kubernetes_cluster_resolver_test.py",
tfrt_enabled = True,
deps = [
":kubernetes_cluster_resolver_py",
"//tensorflow/python:client_testlib",

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

@ -32,7 +32,6 @@ 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",

14
tensorflow/python/eager/BUILD
View File

@ -103,7 +103,6 @@ cuda_py_test(
"no_oss", # TODO(b/168051787): Enable.
"no_pip", # TODO(b/168051787): Enable.
],
tfrt_enabled = True,
deps = [
":pywrap_tensor_test_util",
":test",
@ -179,7 +178,6 @@ cuda_py_test(
size = "small",
srcs = ["cancellation_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":cancellation",
":test",
@ -264,7 +262,6 @@ cuda_py_test(
size = "small",
srcs = ["context_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":context",
":test",
@ -288,7 +285,6 @@ cuda_py_test(
name = "monitoring_test",
srcs = ["monitoring_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":monitoring",
":test",
@ -361,7 +357,6 @@ cuda_py_test(
name = "tensor_test",
srcs = ["tensor_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":context",
":test",
@ -415,7 +410,6 @@ cuda_py_test(
size = "small",
srcs = ["core_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":context",
":core",
@ -451,7 +445,6 @@ cuda_py_test(
size = "medium",
srcs = ["function_defun_collection_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":backprop",
":def_function",
@ -567,7 +560,6 @@ cuda_py_test(
name = "graph_only_ops_test",
srcs = ["graph_only_ops_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"graph_only_ops",
"//tensorflow/python:client_testlib",
@ -703,7 +695,6 @@ cuda_py_test(
name = "benchmarks_test",
srcs = ["benchmarks_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":backprop",
":benchmarks_test_base",
@ -723,7 +714,6 @@ cuda_py_test(
name = "remote_benchmarks_test",
srcs = ["remote_benchmarks_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":backprop",
":benchmarks_test_base",
@ -749,7 +739,6 @@ tf_py_test(
name = "tape_test",
srcs = ["tape_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":backprop",
":context",
@ -769,7 +758,6 @@ cuda_py_test(
name = "ops_test",
srcs = ["ops_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":context",
":execute",
@ -795,7 +783,6 @@ tf_py_test(
name = "pywrap_tfe_test",
srcs = ["pywrap_tfe_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":backprop",
":context",
@ -861,7 +848,6 @@ tf_py_test(
size = "medium",
srcs = ["lift_to_graph_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"lift_to_graph",
"//tensorflow/python:framework_ops",

1
tensorflow/python/eager/benchmarks/resnet50/BUILD
View File

@ -46,7 +46,6 @@ cuda_py_test(
"oss_serial",
"v1only",
],
tfrt_enabled = True,
deps = [
":resnet50",
":resnet50_test_util",

1
tensorflow/python/framework/experimental/BUILD
View File

@ -139,7 +139,6 @@ cuda_py_test(
"no_pip",
"no_windows", # b/168218876
],
tfrt_enabled = True,
deps = [
":_unified_api",
":context_stack",

7
tensorflow/python/keras/BUILD
View File

@ -339,7 +339,6 @@ tf_py_test(
size = "small",
srcs = ["activations_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":activations",
":backend",
@ -359,7 +358,6 @@ tf_py_test(
size = "small",
srcs = ["combinations_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":combinations",
":testing_utils",
@ -376,7 +374,6 @@ tf_py_test(
size = "small",
srcs = ["constraints_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":backend",
":combinations",
@ -391,7 +388,6 @@ tf_py_test(
size = "small",
srcs = ["initializers_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":backend",
":combinations",
@ -426,7 +422,6 @@ tf_py_test(
python_version = "PY3",
shard_count = 8,
tags = ["notsan"],
tfrt_enabled = True,
deps = [
":keras",
"//tensorflow/python:client_testlib",
@ -630,7 +625,6 @@ tf_py_test(
size = "medium",
srcs = ["backend_config_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":backend",
":backend_config",
@ -645,7 +639,6 @@ tf_py_test(
srcs = ["keras_parameterized_test.py"],
python_version = "PY3",
tags = ["notsan"],
tfrt_enabled = True,
deps = [
":keras",
"//tensorflow/python:client_testlib",

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

@ -63,7 +63,6 @@ tf_py_test(
"no_rocm",
"notsan", # b/168814536
],
tfrt_enabled = True,
deps = [
":applications",
"//tensorflow/python:client_testlib",

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

@ -73,7 +73,6 @@ cuda_py_test(
tags = COMMON_TAGS + [
"no_oss_py38", # TODO(b/162044699)
],
tfrt_enabled = True,
deps = [
":profiler_lib",
"//tensorflow:tensorflow_py",
@ -85,7 +84,6 @@ 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",
@ -107,7 +105,6 @@ 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",
@ -120,7 +117,6 @@ 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",
@ -132,7 +128,6 @@ 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",
@ -144,7 +139,6 @@ 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",
@ -157,7 +151,6 @@ 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",
@ -169,7 +162,6 @@ 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",
@ -181,7 +173,6 @@ 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",
@ -194,7 +185,6 @@ 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",
@ -206,7 +196,6 @@ cuda_py_test(
srcs = ["keras_examples_benchmarks/mnist_conv_custom_training_benchmark_test.py"],
python_version = "PY3",
tags = COMMON_TAGS,
tfrt_enabled = True,
deps = [
":distribution_util",
"//tensorflow:tensorflow_py",

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

@ -43,7 +43,6 @@ cuda_py_test(
"no_pip", # b/161253163
"no_windows", # b/160628318
],
tfrt_enabled = True,
deps = [
":saved_model_benchmark_util",
"//tensorflow:tensorflow_py",
@ -58,7 +57,6 @@ cuda_py_test(
"no_pip", # b/161253163
"no_windows", # b/160628318
],
tfrt_enabled = True,
deps = [
":saved_model_benchmark_util",
"//tensorflow:tensorflow_py",
@ -73,7 +71,6 @@ cuda_py_test(
"no_pip", # b/161253163
"no_windows", # b/160628318
],
tfrt_enabled = True,
deps = [
":saved_model_benchmark_util",
"//tensorflow:tensorflow_py",
@ -88,7 +85,6 @@ cuda_py_test(
"no_pip", # b/161253163
"no_windows", # b/160628318
],
tfrt_enabled = True,
deps = [
":saved_model_benchmark_util",
"//tensorflow:tensorflow_py",
@ -103,7 +99,6 @@ cuda_py_test(
"no_pip", # b/161253163
"no_windows", # b/160628318
],
tfrt_enabled = True,
deps = [
":saved_model_benchmark_util",
"//tensorflow:tensorflow_py",
@ -118,7 +113,6 @@ cuda_py_test(
"no_pip", # b/161253163
"no_windows", # b/160628318
],
tfrt_enabled = True,
deps = [
":saved_model_benchmark_util",
"//tensorflow:tensorflow_py",
@ -133,7 +127,6 @@ cuda_py_test(
"no_pip", # b/161253163
"no_windows", # b/160628318
],
tfrt_enabled = True,
deps = [
":saved_model_benchmark_util",
"//tensorflow:tensorflow_py",
@ -148,7 +141,6 @@ cuda_py_test(
"no_pip", # b/161253163
"no_windows", # b/160628318
],
tfrt_enabled = True,
deps = [
":saved_model_benchmark_util",
"//tensorflow:tensorflow_py",

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

@ -136,7 +136,6 @@ 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",

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

@ -340,7 +340,6 @@ tf_py_test(
tags = [
"nomac", # TODO(mihaimaruseac): b/127695564
],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python/keras",
@ -356,7 +355,6 @@ tf_py_test(
tags = [
"nomac", # TODO(mihaimaruseac): b/127695564
],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python/keras",
@ -372,7 +370,6 @@ tf_py_test(
tags = [
"nomac", # TODO(mihaimaruseac): b/127695564
],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python/keras",
@ -641,7 +638,6 @@ tf_py_test(
tags = [
"nomac", # TODO(mihaimaruseac): b/127695564
],
tfrt_enabled = True,
deps = [
":base_layer",
":engine",

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

@ -590,7 +590,6 @@ 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",
@ -808,7 +807,6 @@ cuda_py_test(
size = "medium",
srcs = ["separable_convolutional_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python/keras",
@ -855,7 +853,6 @@ tf_py_test(
size = "small",
srcs = ["serialization_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python/keras",
@ -922,7 +919,6 @@ tf_py_test(
tags = [
"notsan",
],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python/keras",
@ -939,7 +935,6 @@ tf_py_test(
size = "small",
srcs = ["layers_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":layers",
"//tensorflow/python:client_testlib",

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

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

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

@ -18,7 +18,6 @@ tf_py_test(
name = "category_encoding_benchmark",
srcs = ["category_encoding_benchmark.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
@ -36,7 +35,6 @@ tf_py_test(
name = "category_crossing_benchmark",
srcs = ["category_crossing_benchmark.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
@ -55,7 +53,6 @@ tf_py_test(
name = "hashing_benchmark",
srcs = ["hashing_benchmark.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
@ -74,7 +71,6 @@ tf_py_test(
name = "index_lookup_adapt_benchmark",
srcs = ["index_lookup_adapt_benchmark.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:dtypes",
@ -92,7 +88,6 @@ tf_py_test(
name = "normalization_adapt_benchmark",
srcs = ["normalization_adapt_benchmark.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
@ -112,7 +107,6 @@ cuda_py_test(
name = "image_preproc_benchmark",
srcs = ["image_preproc_benchmark.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",

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

@ -88,7 +88,6 @@ tf_py_test(
srcs = ["base_test.py"],
main = "base_test.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":core",
":layers_base",
@ -119,7 +118,6 @@ tf_py_test(
srcs = ["core_test.py"],
main = "core_test.py",
python_version = "PY3",
tfrt_enabled = True,
deps = [
":core",
"//tensorflow/python:array_ops",
@ -171,7 +169,6 @@ tf_py_test(
main = "pooling_test.py",
python_version = "PY3",
tags = ["no_rocm"],
tfrt_enabled = True,
deps = [
":pooling",
"//tensorflow/python:array_ops",
@ -189,7 +186,6 @@ cuda_py_test(
main = "normalization_test.py",
python_version = "PY3",
shard_count = 10,
tfrt_enabled = True,
deps = [
":convolutional",
":normalization",

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

@ -92,7 +92,6 @@ cuda_py_test(
size = "medium",
srcs = ["adagrad_test.py"],
shard_count = 4,
tfrt_enabled = True,
deps = [
":optimizer_v2",
"//tensorflow/python:client_testlib",
@ -180,7 +179,6 @@ cuda_py_test(
size = "medium",
srcs = ["ftrl_test.py"],
shard_count = 4,
tfrt_enabled = True,
deps = [
":optimizer_v2",
"//tensorflow/python:client_testlib",
@ -220,7 +218,6 @@ cuda_py_test(
size = "medium",
srcs = ["nadam_test.py"],
shard_count = 4,
tfrt_enabled = True,
deps = [
":optimizer_v2",
"//tensorflow/python:client_testlib",
@ -267,7 +264,6 @@ cuda_py_test(
size = "medium",
srcs = ["learning_rate_schedule_test.py"],
shard_count = 4,
tfrt_enabled = True,
deps = [
":optimizer_v2",
"//tensorflow/python:client_testlib",
@ -282,7 +278,6 @@ 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",

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

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

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

@ -210,7 +210,6 @@ tf_py_test(
size = "small",
srcs = ["saved_model/json_utils_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":saving",
"//tensorflow/python:client_testlib",

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

@ -29,7 +29,6 @@ tf_py_test(
tags = [
"no_pip",
],
tfrt_enabled = True,
deps = [
":get_config_samples",
"//tensorflow/python:client_testlib",
@ -128,7 +127,6 @@ 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",
@ -271,7 +269,6 @@ 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",
@ -304,7 +301,6 @@ 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",
@ -322,7 +318,6 @@ 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",
@ -369,7 +364,6 @@ 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",
@ -402,7 +396,6 @@ tf_py_test(
"no_windows",
"nomac",
],
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:constant_op",
@ -431,7 +424,6 @@ 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",

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

@ -276,7 +276,6 @@ tf_py_test(
size = "small",
srcs = ["version_utils_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":version_utils",
"//tensorflow/python:client_testlib",
@ -290,7 +289,6 @@ tf_py_test(
size = "small",
srcs = ["tf_utils_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":tf_utils",
"//tensorflow/python:client_testlib",
@ -332,7 +330,6 @@ tf_py_test(
"no_windows", # TODO: needs investigation on Windows
"notsan",
],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python/keras",
@ -346,7 +343,6 @@ tf_py_test(
size = "small",
srcs = ["layer_utils_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":layer_utils",
"//tensorflow/python:client_testlib",
@ -360,7 +356,6 @@ tf_py_test(
size = "small",
srcs = ["np_utils_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python/keras",
@ -374,7 +369,6 @@ tf_py_test(
size = "small",
srcs = ["kernelized_utils_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":layer_utils",
"//tensorflow/python:client_testlib",
@ -406,7 +400,6 @@ tf_py_test(
size = "small",
srcs = ["vis_utils_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python/keras",
@ -420,7 +413,6 @@ tf_py_test(
size = "small",
srcs = ["conv_utils_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python/keras",
@ -434,7 +426,6 @@ 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",

175
tensorflow/python/kernel_tests/BUILD
View File

File diff suppressed because it is too large Load Diff

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

@ -10,7 +10,6 @@ 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",
@ -24,7 +23,6 @@ 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",
@ -39,7 +37,6 @@ 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",

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

@ -24,7 +24,6 @@ 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",
@ -40,7 +39,6 @@ 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",
@ -55,7 +53,6 @@ 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",
@ -72,7 +69,6 @@ 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",
@ -87,7 +83,6 @@ 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",

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

@ -11,7 +11,6 @@ cuda_py_test(
name = "bijector_test",
size = "small",
srcs = ["bijector_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
@ -49,7 +48,6 @@ 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",
@ -62,7 +60,6 @@ cuda_py_test(
name = "beta_test",
size = "small",
srcs = ["beta_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client",
"//tensorflow/python:client_testlib",
@ -268,7 +265,6 @@ 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",
@ -285,7 +281,6 @@ cuda_py_test(
name = "identity_bijector_test",
size = "small",
srcs = ["identity_bijector_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",

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

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

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

@ -11,7 +11,6 @@ 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",
@ -29,7 +28,6 @@ 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",
],
@ -55,7 +53,6 @@ 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",
],
@ -67,7 +64,6 @@ 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",
],
@ -79,7 +75,6 @@ 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",
],

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

@ -126,7 +126,6 @@ tf_py_test(
tags = [
"no_pip",
],
tfrt_enabled = True,
deps = [
":descriptor_source_test_base",
"//tensorflow/python:client_testlib",

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

@ -24,7 +24,6 @@ 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",
@ -46,7 +45,6 @@ tf_py_test(
tags = [
"no_cuda_on_cpu_tap", # TODO(b/171060960) flakyly broken assertions
],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:data_flow_ops",
@ -62,7 +60,6 @@ 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",
@ -82,7 +79,6 @@ 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",
@ -101,7 +97,6 @@ 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",
@ -113,7 +108,6 @@ cuda_py_test(
name = "random_ops_test",
size = "medium",
srcs = ["random_ops_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
@ -128,7 +122,6 @@ cuda_py_test(
size = "medium",
srcs = ["stateless_random_ops_test.py"],
shard_count = 10,
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
@ -162,7 +155,6 @@ 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",
@ -183,7 +175,6 @@ tf_py_test(
srcs = ["random_binomial_test.py"],
shard_count = 3,
tags = ["no_oss"],
tfrt_enabled = True,
deps = [
":util",
"//tensorflow/python:array_ops",
@ -200,7 +191,6 @@ cuda_py_test(
name = "random_poisson_test",
size = "medium",
srcs = ["random_poisson_test.py"],
tfrt_enabled = True,
deps = [
":util",
"//tensorflow/python:array_ops",

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

@ -12,7 +12,6 @@ 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",
@ -24,7 +23,6 @@ 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",
@ -37,7 +35,6 @@ cuda_py_test(
name = "session_ops_test",
size = "small",
srcs = ["session_ops_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",

1
tensorflow/python/module/BUILD
View File

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

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

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

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

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

4
tensorflow/python/profiler/BUILD
View File

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

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

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

9
tensorflow/python/saved_model/BUILD
View File

@ -107,7 +107,6 @@ tf_py_test(
name = "loader_test",
size = "small",
srcs = ["loader_test.py"],
tfrt_enabled = True,
deps = [
":builder",
":loader",
@ -166,7 +165,6 @@ 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",
@ -217,7 +215,6 @@ tf_py_test(
name = "utils_test",
size = "small",
srcs = ["utils_test.py"],
tfrt_enabled = True,
deps = [
":utils",
"//tensorflow/core:protos_all_py",
@ -250,7 +247,6 @@ 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",
@ -266,7 +262,6 @@ tf_py_test(
name = "simple_save_test",
size = "small",
srcs = ["simple_save_test.py"],
tfrt_enabled = True,
deps = [
":loader",
":signature_constants",
@ -311,7 +306,6 @@ tf_py_test(
name = "save_context_test",
srcs = ["save_context_test.py"],
srcs_version = "PY2AND3",
tfrt_enabled = True,
deps = [
":save_context",
":save_options",
@ -533,7 +527,6 @@ 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",
@ -608,7 +601,6 @@ 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",
@ -653,7 +645,6 @@ py_strict_library(
tf_py_test(
name = "method_name_updater_test",
srcs = ["method_name_updater_test.py"],
tfrt_enabled = True,
deps = [
":method_name_updater",
"//tensorflow/core:protos_all_py",

8
tensorflow/python/tpu/BUILD
View File

@ -337,7 +337,6 @@ 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",
@ -351,7 +350,6 @@ tf_py_test(
name = "tpu_sharding_test",
size = "small",
srcs = ["tpu_sharding_test.py"],
tfrt_enabled = True,
deps = [
":tpu",
"//tensorflow/python:client_testlib",
@ -363,7 +361,6 @@ tf_py_test(
name = "bfloat16_test",
size = "small",
srcs = ["bfloat16_test.py"],
tfrt_enabled = True,
deps = [
":tpu",
"//tensorflow/python:client_testlib",
@ -375,7 +372,6 @@ tf_py_test(
name = "tpu_infeed_test",
size = "small",
srcs = ["tpu_infeed_test.py"],
tfrt_enabled = True,
deps = [
":tpu",
"//tensorflow/python:framework",
@ -387,7 +383,6 @@ tf_py_test(
name = "topology_test",
size = "medium",
srcs = ["topology_test.py"],
tfrt_enabled = True,
deps = [
":tpu",
"//tensorflow/python:framework_test_lib",
@ -464,7 +459,6 @@ tf_py_test(
"feature_column_test.py",
],
main = "feature_column_test.py",
tfrt_enabled = True,
deps = [
":feature_column",
"//tensorflow/python:client_testlib",
@ -487,7 +481,6 @@ 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",
@ -622,7 +615,6 @@ tf_py_test(
],
python_version = "PY3",
srcs_version = "PY2AND3",
tfrt_enabled = True,
deps = [
":tpu_embedding_v2",
"//tensorflow/python/compat:v2_compat",

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

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

20
tensorflow/python/training/BUILD
View File

@ -602,7 +602,6 @@ tf_py_test(
tags = [
"noasan", # TODO(b/161236904): flaky timeout in trying to start gRPC server
],
tfrt_enabled = True,
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:array_ops",
@ -624,7 +623,6 @@ tf_py_test(
srcs = ["server_lib_multiple_containers_test.py"],
grpc_enabled = True,
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:array_ops",
@ -645,7 +643,6 @@ tf_py_test(
srcs = ["server_lib_same_variables_clear_container_test.py"],
grpc_enabled = True,
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:array_ops",
@ -666,7 +663,6 @@ tf_py_test(
srcs = ["server_lib_same_variables_clear_test.py"],
grpc_enabled = True,
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:array_ops",
@ -687,7 +683,6 @@ tf_py_test(
srcs = ["server_lib_same_variables_no_clear_test.py"],
grpc_enabled = True,
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:array_ops",
@ -708,7 +703,6 @@ tf_py_test(
srcs = ["server_lib_sparse_job_test.py"],
grpc_enabled = True,
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:array_ops",
@ -735,7 +729,6 @@ cuda_py_test(
"no_oss", # Test flaky due to port collisions.
"oss_serial",
],
tfrt_enabled = True,
deps = [
":device_setter",
"//tensorflow/python:client_testlib",
@ -762,7 +755,6 @@ tf_py_test(
"notsan", # data race due to b/62910646
"oss_serial",
],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
@ -840,7 +832,6 @@ cuda_py_test(
"checkpoint_management_test.py",
],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":checkpoint_management",
":saver",
@ -962,7 +953,6 @@ tf_py_test(
"noasan", # http://b/30379628
"notsan", # http://b/30379628
],
tfrt_enabled = True,
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:client",
@ -982,7 +972,6 @@ tf_py_test(
"noasan", # http://b/30782289
"notsan", # http://b/30782289
],
tfrt_enabled = True,
deps = [
"//tensorflow/python:client",
"//tensorflow/python:client_testlib",
@ -1050,7 +1039,6 @@ tf_py_test(
grpc_enabled = True,
python_version = "PY3",
tags = ["no_windows"],
tfrt_enabled = True,
deps = [
":checkpoint_management",
":saver",
@ -1106,7 +1094,6 @@ tf_py_test(
size = "small",
srcs = ["training_util_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":training_util",
"//tensorflow/python:client_testlib",
@ -1121,7 +1108,6 @@ cuda_py_test(
size = "medium",
srcs = ["adam_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":adam",
"//tensorflow/python:array_ops",
@ -1150,7 +1136,6 @@ cuda_py_test(
"no_windows", # b/139083295: bfloat16 tests fail on Windows
"notsan",
],
tfrt_enabled = True,
deps = [
":moving_averages",
":saver",
@ -1264,7 +1249,6 @@ tf_py_test(
"no_windows",
"notsan", # intermittent races on a few percent of runs
],
tfrt_enabled = True,
deps = [
"//tensorflow/core:protos_all_py",
"//tensorflow/python:client",
@ -1313,7 +1297,6 @@ tf_py_test(
size = "small",
srcs = ["checkpoint_ops_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow/python:checkpoint_ops_gen",
"//tensorflow/python:client",
@ -1334,7 +1317,6 @@ tf_py_test(
size = "medium",
srcs = ["warm_starting_util_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
@ -1382,7 +1364,6 @@ tf_py_test(
"no_pip",
"notsan", # b/67945581
],
tfrt_enabled = True,
deps = [
":checkpoint_management",
":monitored_session",
@ -1408,7 +1389,6 @@ tf_py_test(
size = "medium",
srcs = ["input_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",

1
tensorflow/python/training/experimental/BUILD
View File

@ -107,7 +107,6 @@ cuda_py_test(
size = "small",
srcs = ["mixed_precision_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":mixed_precision",
"//tensorflow/python:client_testlib",

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

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