From ff97edee775e6d13912f82897024baf699ff6329 Mon Sep 17 00:00:00 2001
From: Guangda Lai <31743510+aaroey@users.noreply.github.com>
Date: Wed, 16 Oct 2019 21:07:20 -0700
Subject: [PATCH] Update the test size to avoid timeout.

---
 tensorflow/python/kernel_tests/BUILD | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tensorflow/python/kernel_tests/BUILD b/tensorflow/python/kernel_tests/BUILD
index 47523d1d877..ed0a92d59f5 100644
--- a/tensorflow/python/kernel_tests/BUILD
+++ b/tensorflow/python/kernel_tests/BUILD
@@ -1699,7 +1699,7 @@ py_library(
 
 cuda_py_test(
     name = "bias_op_deterministic_test",
-    size = "small",
+    size = "medium",
     srcs = ["bias_op_deterministic_test.py"],
     additional_deps = [
         ":bias_op_base",
@@ -1709,7 +1709,7 @@ cuda_py_test(
 
 cuda_py_test(
     name = "bias_op_test",
-    size = "small",
+    size = "medium",
     srcs = ["bias_op_test.py"],
     additional_deps = [
         ":bias_op_base",