From cc686693f3ca5a6b02ecb55dffd529b7eb15017c Mon Sep 17 00:00:00 2001
From: Reed Wanderman-Milne <reedwm@google.com>
Date: Wed, 3 Apr 2019 16:34:26 -0700
Subject: [PATCH] Increase size of loss_scale_test to fix nightly build

PiperOrigin-RevId: 241830333
---
 tensorflow/python/keras/mixed_precision/experimental/BUILD | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tensorflow/python/keras/mixed_precision/experimental/BUILD b/tensorflow/python/keras/mixed_precision/experimental/BUILD
index 1b6687dce68..628944c2b05 100644
--- a/tensorflow/python/keras/mixed_precision/experimental/BUILD
+++ b/tensorflow/python/keras/mixed_precision/experimental/BUILD
@@ -126,7 +126,7 @@ py_library(
 
 py_test(
     name = "loss_scale_test",
-    size = "small",
+    size = "medium",
     srcs = ["loss_scale_test.py"],
     deps = [
         ":loss_scale",