diff --git a/tensorflow/python/kernel_tests/BUILD b/tensorflow/python/kernel_tests/BUILD
index 90bbd982fa5..6f49fc6e766 100644
--- a/tensorflow/python/kernel_tests/BUILD
+++ b/tensorflow/python/kernel_tests/BUILD
@@ -3231,7 +3231,7 @@ cuda_py_test(
     name = "extract_image_patches_grad_test",
     size = "medium",
     srcs = ["extract_image_patches_grad_test.py"],
-    shard_count = 9,
+    shard_count = 15,
     tags = ["notap"],  # http://b/31080670
     deps = [
         "//tensorflow/python:array_ops",