aboutsummaryrefslogtreecommitdiffhomepage
diff options
context:
space:
mode:
Diffstat
-rw-r--r--tensorflow/compiler/tests/BUILD2
-rw-r--r--tensorflow/contrib/data/python/kernel_tests/BUILD15
-rw-r--r--tensorflow/contrib/distributions/BUILD1
-rw-r--r--tensorflow/contrib/estimator/BUILD4
-rw-r--r--tensorflow/contrib/legacy_seq2seq/BUILD5
-rw-r--r--tensorflow/contrib/linalg/BUILD10
-rw-r--r--tensorflow/contrib/lookup/BUILD2
-rw-r--r--tensorflow/contrib/optimizer_v2/BUILD1
-rw-r--r--tensorflow/core/BUILD16
-rw-r--r--tensorflow/examples/tutorials/mnist/BUILD1
-rw-r--r--tensorflow/python/BUILD4
-rw-r--r--tensorflow/python/estimator/BUILD1
-rwxr-xr-xtensorflow/python/keras/BUILD2
-rw-r--r--tensorflow/python/kernel_tests/BUILD7
-rw-r--r--tensorflow/python/kernel_tests/linalg/BUILD1
-rw-r--r--tensorflow/tools/docs/BUILD3
16 files changed, 58 insertions, 17 deletions
diff --git a/tensorflow/compiler/tests/BUILD b/tensorflow/compiler/tests/BUILD
index b9e42ca677..46b86c53aa 100644
--- a/tensorflow/compiler/tests/BUILD
+++ b/tensorflow/compiler/tests/BUILD
@@ -340,7 +340,7 @@ tf_xla_py_test(
tf_xla_py_test(
name = "ftrl_test",
- size = "small",
+ size = "medium",
srcs = ["ftrl_test.py"],
deps = [
":xla_test",
diff --git a/tensorflow/contrib/data/python/kernel_tests/BUILD b/tensorflow/contrib/data/python/kernel_tests/BUILD
index b475c9fa6b..c554607960 100644
--- a/tensorflow/contrib/data/python/kernel_tests/BUILD
+++ b/tensorflow/contrib/data/python/kernel_tests/BUILD
@@ -122,7 +122,10 @@ py_test(
size = "small",
srcs = ["filter_dataset_op_test.py"],
srcs_version = "PY2AND3",
- tags = ["no_pip"],
+ tags = [
+ "no_pip",
+ "optonly",
+ ],
deps = [
":dataset_serialization_test",
"//tensorflow/python:array_ops",
@@ -211,7 +214,10 @@ py_test(
size = "medium",
srcs = ["map_dataset_op_test.py"],
srcs_version = "PY2AND3",
- tags = ["no_pip"],
+ tags = [
+ "no_pip",
+ "optonly",
+ ],
deps = [
":dataset_serialization_test",
"//tensorflow/contrib/data/python/ops:error_ops",
@@ -306,7 +312,10 @@ py_test(
srcs = ["resample_test.py"],
shard_count = 2,
srcs_version = "PY2AND3",
- tags = ["noasan"],
+ tags = [
+ "noasan",
+ "optonly",
+ ],
deps = [
"//tensorflow/contrib/data/python/ops:resampling",
"//tensorflow/python:client_testlib",
diff --git a/tensorflow/contrib/distributions/BUILD b/tensorflow/contrib/distributions/BUILD
index 20e432b88d..2d99e8172d 100644
--- a/tensorflow/contrib/distributions/BUILD
+++ b/tensorflow/contrib/distributions/BUILD
@@ -877,6 +877,7 @@ cuda_py_test(
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:platform_test",
],
+ tags = ["optonly"],
)
cuda_py_test(
diff --git a/tensorflow/contrib/estimator/BUILD b/tensorflow/contrib/estimator/BUILD
index 9f4cd44afb..9e88bc7de1 100644
--- a/tensorflow/contrib/estimator/BUILD
+++ b/tensorflow/contrib/estimator/BUILD
@@ -210,7 +210,7 @@ py_library(
py_test(
name = "head_test",
- size = "small",
+ size = "medium",
srcs = ["python/estimator/head_test.py"],
srcs_version = "PY2AND3",
deps = [
@@ -250,7 +250,7 @@ py_library(
py_test(
name = "linear_test",
- size = "small",
+ size = "medium",
srcs = ["python/estimator/linear_test.py"],
srcs_version = "PY2AND3",
tags = [
diff --git a/tensorflow/contrib/legacy_seq2seq/BUILD b/tensorflow/contrib/legacy_seq2seq/BUILD
index 8c2c4fd29c..4ce91a140f 100644
--- a/tensorflow/contrib/legacy_seq2seq/BUILD
+++ b/tensorflow/contrib/legacy_seq2seq/BUILD
@@ -58,5 +58,8 @@ cuda_py_tests(
"//tensorflow/python:variable_scope",
"//tensorflow/python:variables",
],
- tags = ["noasan"], # times out b/63678675
+ tags = [
+ "noasan", # times out b/63678675
+ "optonly", # times out (flaky)
+ ],
)
diff --git a/tensorflow/contrib/linalg/BUILD b/tensorflow/contrib/linalg/BUILD
index 2c5fa7af89..2e92ad6eb3 100644
--- a/tensorflow/contrib/linalg/BUILD
+++ b/tensorflow/contrib/linalg/BUILD
@@ -59,7 +59,10 @@ cuda_py_test(
"//tensorflow/python:platform_test",
],
shard_count = 5,
- tags = ["noasan"],
+ tags = [
+ "noasan",
+ "optonly",
+ ],
)
cuda_py_test(
@@ -78,5 +81,8 @@ cuda_py_test(
"//tensorflow/python:platform_test",
],
shard_count = 8,
- tags = ["noasan"],
+ tags = [
+ "noasan",
+ "optonly",
+ ],
)
diff --git a/tensorflow/contrib/lookup/BUILD b/tensorflow/contrib/lookup/BUILD
index f616207d46..e3928a82a2 100644
--- a/tensorflow/contrib/lookup/BUILD
+++ b/tensorflow/contrib/lookup/BUILD
@@ -28,7 +28,7 @@ py_library(
tf_py_test(
name = "lookup_ops_test",
- size = "small",
+ size = "medium",
srcs = ["lookup_ops_test.py"],
additional_deps = [
":lookup_py",
diff --git a/tensorflow/contrib/optimizer_v2/BUILD b/tensorflow/contrib/optimizer_v2/BUILD
index 86e5f4a437..85cfce346c 100644
--- a/tensorflow/contrib/optimizer_v2/BUILD
+++ b/tensorflow/contrib/optimizer_v2/BUILD
@@ -203,4 +203,5 @@ cuda_py_test(
"//tensorflow/python:client_testlib",
"//third_party/py/numpy",
],
+ tags = ["optonly"],
)
diff --git a/tensorflow/core/BUILD b/tensorflow/core/BUILD
index 01bda8e09b..21f929894c 100644
--- a/tensorflow/core/BUILD
+++ b/tensorflow/core/BUILD
@@ -2759,7 +2759,6 @@ tf_cc_tests(
"lib/monitoring/sampler_test.cc",
"lib/random/distribution_sampler_test.cc",
"lib/random/philox_random_test.cc",
- "lib/random/random_distributions_test.cc",
"lib/random/random_test.cc",
"lib/random/simple_philox_test.cc",
"lib/strings/base64_test.cc",
@@ -2790,6 +2789,21 @@ tf_cc_tests(
)
tf_cc_test(
+ name = "lib_random_random_distributions_test",
+ srcs = ["lib/random/random_distributions_test.cc"],
+ tags = ["optonly"],
+ deps = [
+ ":lib",
+ ":lib_internal",
+ ":lib_test_internal",
+ ":protos_all_cc",
+ ":test",
+ ":test_main",
+ "//third_party/eigen3",
+ ],
+)
+
+tf_cc_test(
name = "platform_env_test",
size = "small",
srcs = ["platform/env_test.cc"],
diff --git a/tensorflow/examples/tutorials/mnist/BUILD b/tensorflow/examples/tutorials/mnist/BUILD
index aa1b2ec2db..d7bc6a5a7d 100644
--- a/tensorflow/examples/tutorials/mnist/BUILD
+++ b/tensorflow/examples/tutorials/mnist/BUILD
@@ -51,6 +51,7 @@ py_binary(
"fully_connected_feed.py",
],
srcs_version = "PY2AND3",
+ tags = ["optonly"],
deps = [
":input_data",
":mnist",
diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD
index 569d3eb2ce..c2bedab4f9 100644
--- a/tensorflow/python/BUILD
+++ b/tensorflow/python/BUILD
@@ -2802,7 +2802,7 @@ cuda_py_test(
cuda_py_test(
name = "image_ops_test",
- size = "small",
+ size = "medium",
srcs = ["ops/image_ops_test.py"],
additional_deps = [
":array_ops",
@@ -4333,7 +4333,7 @@ py_test(
tf_py_test(
name = "input_test",
- size = "small",
+ size = "medium",
srcs = ["training/input_test.py"],
additional_deps = [
":array_ops",
diff --git a/tensorflow/python/estimator/BUILD b/tensorflow/python/estimator/BUILD
index 7bf4447491..c6bb9b9be7 100644
--- a/tensorflow/python/estimator/BUILD
+++ b/tensorflow/python/estimator/BUILD
@@ -205,6 +205,7 @@ py_test(
"no_pip",
"noasan", # test flakily times out in asan mode.
"notsan", # b/67510291
+ "optonly", # flakily times out in fastbuild
],
deps = [
":baseline",
diff --git a/tensorflow/python/keras/BUILD b/tensorflow/python/keras/BUILD
index 024a8cd3d1..ca7686b1d1 100755
--- a/tensorflow/python/keras/BUILD
+++ b/tensorflow/python/keras/BUILD
@@ -835,7 +835,7 @@ py_test(
py_test(
name = "saving_test",
- size = "small",
+ size = "medium",
srcs = ["_impl/keras/engine/saving_test.py"],
srcs_version = "PY2AND3",
deps = [
diff --git a/tensorflow/python/kernel_tests/BUILD b/tensorflow/python/kernel_tests/BUILD
index 11adb1ccfc..a02783e7e7 100644
--- a/tensorflow/python/kernel_tests/BUILD
+++ b/tensorflow/python/kernel_tests/BUILD
@@ -592,7 +592,7 @@ cuda_py_test(
cuda_py_test(
name = "matrix_solve_op_test",
- size = "small",
+ size = "medium",
srcs = ["matrix_solve_op_test.py"],
additional_deps = [
"//third_party/py/numpy",
@@ -1603,7 +1603,10 @@ cuda_py_test(
"//tensorflow/python:variable_scope",
"//tensorflow/python:variables",
],
- tags = ["noasan"],
+ tags = [
+ "noasan",
+ "optonly",
+ ],
)
cuda_py_test(
diff --git a/tensorflow/python/kernel_tests/linalg/BUILD b/tensorflow/python/kernel_tests/linalg/BUILD
index 9555e51099..4e3f24890b 100644
--- a/tensorflow/python/kernel_tests/linalg/BUILD
+++ b/tensorflow/python/kernel_tests/linalg/BUILD
@@ -123,6 +123,7 @@ cuda_py_test(
"//tensorflow/python:platform_test",
],
shard_count = 5,
+ tags = ["optonly"],
)
cuda_py_test(
diff --git a/tensorflow/tools/docs/BUILD b/tensorflow/tools/docs/BUILD
index 0c1fd0cf9d..58b5ef8345 100644
--- a/tensorflow/tools/docs/BUILD
+++ b/tensorflow/tools/docs/BUILD
@@ -103,10 +103,11 @@ py_test(
data = ["//tensorflow:docs_src"],
srcs_version = "PY2AND3",
tags = [
- # No reason to run sanitizers for this test.
+ # No reason to run sanitizers or fastbuild for this test.
"noasan",
"nomsan",
"notsan",
+ "optonly",
],
deps = [
":generate_lib",
generated by cgit v1.2.3 (git 2.39.1) at 2024-06-14 15:26:25 +0000