re-classing the long running tests as "large"

ROCm · Feb 14, 2020 · c97e075 · c97e075
1 parent 39ab272
commit c97e075
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tensorflow/python/distribute/BUILD b/tensorflow/python/distribute/BUILD
@@ -1256,7 +1256,7 @@ py_library(
 
 distribute_py_test(
     name = "saved_model_save_load_test",
-    size = "medium",
+    size = "large",
     srcs = ["saved_model_save_load_test.py"],
     full_precision = True,
     main = "saved_model_save_load_test.py",
@@ -1272,7 +1272,7 @@ distribute_py_test(
 
 distribute_py_test(
     name = "keras_save_load_test",
-    size = "medium",
+    size = "large",
     srcs = ["keras_save_load_test.py"],
     full_precision = True,
     main = "keras_save_load_test.py",
@@ -1288,7 +1288,7 @@ distribute_py_test(
 
 distribute_py_test(
     name = "saved_model_mixed_api_test",
-    size = "medium",
+    size = "large",
     srcs = ["saved_model_mixed_api_test.py"],
     full_precision = True,
     main = "saved_model_mixed_api_test.py",

diff --git a/tensorflow/python/kernel_tests/BUILD b/tensorflow/python/kernel_tests/BUILD
@@ -2912,7 +2912,7 @@ cuda_py_test(
 
 cuda_py_test(
     name = "conv_ops_test",
-    size = "medium",
+    size = "large",
     srcs = ["conv_ops_test.py"],
     shard_count = 4,
     tags = [