aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/batching
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2017-06-09 14:45:22 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-06-09 14:48:59 -0700
commit435599f5d896d7e1f721ffe6fd092d39efe2b027 (patch)
tree609a81818fef9c899a628461654eba6683537e56 /tensorflow/contrib/batching
parent0058c1f134c7de0733ffed754fe3a3091dea60ca (diff)
Adjust test sizes
PiperOrigin-RevId: 158565259
Diffstat (limited to 'tensorflow/contrib/batching')
-rw-r--r--tensorflow/contrib/batching/BUILD3
-rw-r--r--tensorflow/contrib/batching/util/BUILD1
2 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/contrib/batching/BUILD b/tensorflow/contrib/batching/BUILD
index b16fb9b5bb..6478ad2ea7 100644
--- a/tensorflow/contrib/batching/BUILD
+++ b/tensorflow/contrib/batching/BUILD
@@ -27,6 +27,7 @@ cc_library(
cc_test(
name = "batch_scheduler_test",
+ size = "small",
srcs = [
"batch_scheduler_test.cc",
],
@@ -60,6 +61,7 @@ cc_library(
cc_test(
name = "shared_batch_scheduler_test",
+ size = "small",
srcs = [
"shared_batch_scheduler_test.cc",
],
@@ -83,6 +85,7 @@ cc_library(
cc_test(
name = "basic_batch_scheduler_test",
+ size = "small",
srcs = [
"basic_batch_scheduler_test.cc",
],
diff --git a/tensorflow/contrib/batching/util/BUILD b/tensorflow/contrib/batching/util/BUILD
index 3d140c73a4..48d03746bb 100644
--- a/tensorflow/contrib/batching/util/BUILD
+++ b/tensorflow/contrib/batching/util/BUILD
@@ -43,6 +43,7 @@ cc_library(
cc_test(
name = "periodic_function_test",
+ size = "small",
srcs = ["periodic_function_test.cc"],
deps = [
":periodic_function",