Update the test size to avoid timeout.

2019-10-16 21:07:20 -07:00 · 2019-10-16 21:07:20 -07:00 · ff97edee77
commit ff97edee77
parent aff49a1b9a
1 changed files with 2 additions and 2 deletions
--- 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",