re-classing the long running save/restore tests as "large"

ROCm · Feb 13, 2020 · afabf2f · afabf2f
1 parent 98149e6
commit afabf2f
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/tensorflow/python/distribute/BUILD b/tensorflow/python/distribute/BUILD
@@ -1199,14 +1199,13 @@ 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",
     shard_count = 5,
     tags = [
         "multi_and_single_gpu",
-        "no_rocm",
     ],
     deps = [
         ":saved_model_test_base",
@@ -1216,14 +1215,13 @@ 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",
     shard_count = 5,
     tags = [
         "multi_and_single_gpu",
-        "no_rocm",
     ],
     deps = [
         ":saved_model_test_base",
@@ -1233,14 +1231,13 @@ 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",
     shard_count = 5,
     tags = [
         "multi_and_single_gpu",
-        "no_rocm",
     ],
     deps = [
         ":saved_model_test_base",