Sync OSS keras to head.

PiperOrigin-RevId: 348083084 Change-Id: I7ee1cf13c157952929854654ea081bb6c3e4db0d
2020-12-17 13:31:31 -08:00 · 2020-12-17 13:31:31 -08:00 · 684bbf8dec
commit 684bbf8dec
parent 50c75992c7
1 changed files with 1 additions and 0 deletions
--- 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",
    ],
 )