Attempt to fx revive_test timeout on macOS

PiperOrigin-RevId: 327900094
Change-Id: I213eb1a62d247445b7b9ea83cec6a718337711f6
diff --git a/tensorflow/python/keras/saving/BUILD b/tensorflow/python/keras/saving/BUILD
index 62000be..2e1b6cd 100644
--- a/tensorflow/python/keras/saving/BUILD
+++ b/tensorflow/python/keras/saving/BUILD
@@ -177,7 +177,7 @@
     size = "medium",
     srcs = ["saved_model/revive_test.py"],
     python_version = "PY3",
-    shard_count = 4,
+    shard_count = 8,
     tags = [
         "no_windows",  # b/158005583
     ],