commit eaabc7207aa8136fc08bf316fc36d8d91f5bbf56
author A. Unique TensorFlower <gardener@tensorflow.org> Fri Sep 25 14:22:55 2020 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Fri Sep 25 14:30:44 2020 -0700
tree 829bb085ed584889d2538c3860dc5970d8692610
parent befe9f45f2f0e8a07e73074a06322826e6b354a8

Quick fixes for tests that are timing out and blocking presubmit.

These tests are now disabled from presubmit because they now have size=large:
//third_party/tensorflow/python:image_grad_deterministic_test
//third_party/tensorflow/python:image_grad_test

PiperOrigin-RevId: 333808481
Change-Id: Ibe3c55c5d916323e6e7c52a17d0d67062153edd8

tensorflow/python/BUILD

diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD
index 419b8b0..4ec33a2 100644
--- a/tensorflow/python/BUILD
+++ b/tensorflow/python/BUILD
@@ -5137,7 +5137,7 @@
 
 cuda_py_test(
     name = "image_grad_deterministic_test",
-    size = "medium",
+    size = "large",
     srcs = ["ops/image_grad_deterministic_test.py"],
     python_version = "PY3",
     shard_count = 5,
@@ -5148,7 +5148,7 @@
 
 cuda_py_test(
     name = "image_grad_test",
-    size = "medium",
+    size = "large",
     srcs = ["ops/image_grad_test.py"],
     python_version = "PY3",
     shard_count = 5,