From 684bbf8dec1ca7e73394caba0dc883334d4cba5a Mon Sep 17 00:00:00 2001
From: Scott Zhu <scottzhu@google.com>
Date: Thu, 17 Dec 2020 13:31:31 -0800
Subject: [PATCH] Sync OSS keras to head.

PiperOrigin-RevId: 348083084
Change-Id: I7ee1cf13c157952929854654ea081bb6c3e4db0d
---
 tensorflow/python/keras/benchmarks/BUILD | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tensorflow/python/keras/benchmarks/BUILD b/tensorflow/python/keras/benchmarks/BUILD
index 631ef62f856..3087db3d3ed 100644
--- a/tensorflow/python/keras/benchmarks/BUILD
+++ b/tensorflow/python/keras/benchmarks/BUILD
@@ -232,6 +232,7 @@ py_test(
         ":benchmark_util",
         ":profiler_lib",
         "//tensorflow:tensorflow_py",
+        "//tensorflow/python/keras/optimizer_v2",
     ],
 )