aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib
diff options
context:
space:
mode:
authorGravatar Smit Hinsu <hinsu@google.com>2018-10-03 16:38:22 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-10-03 16:48:17 -0700
commitd5b362a67a57f53f610536ed6068a5b67bc37b88 (patch)
tree09bd46edb9ec02e54053ad226b2f225fb4a8d8ea /tensorflow/contrib
parent9801b8810e07859141d4417746317cc3dbebc227 (diff)
Update size of mvn_diag_test and core_rnn_cell_test to medium to fix timeouts
PiperOrigin-RevId: 215651746
Diffstat (limited to 'tensorflow/contrib')
-rw-r--r--tensorflow/contrib/distributions/BUILD2
-rw-r--r--tensorflow/contrib/rnn/BUILD2
2 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/distributions/BUILD b/tensorflow/contrib/distributions/BUILD
index 3ff7da4f89..60f6b90edc 100644
--- a/tensorflow/contrib/distributions/BUILD
+++ b/tensorflow/contrib/distributions/BUILD
@@ -299,7 +299,7 @@ cuda_py_test(
cuda_py_test(
name = "mvn_diag_test",
- size = "small",
+ size = "medium",
srcs = ["python/kernel_tests/mvn_diag_test.py"],
additional_deps = [
":distributions_py",
diff --git a/tensorflow/contrib/rnn/BUILD b/tensorflow/contrib/rnn/BUILD
index 4e67d80558..1385a9ddc1 100644
--- a/tensorflow/contrib/rnn/BUILD
+++ b/tensorflow/contrib/rnn/BUILD
@@ -108,7 +108,7 @@ cuda_py_tests(
cuda_py_tests(
name = "core_rnn_cell_test",
- size = "small",
+ size = "medium",
srcs = ["python/kernel_tests/core_rnn_cell_test.py"],
additional_deps = [
":rnn_py",