Adding missing build deps for benchmark tests.

PiperOrigin-RevId: 334900981 Change-Id: I9c5570e728ce01ce2382b715a3c74b8d35591212
2020-10-01 14:45:33 -07:00 · 2020-10-01 14:45:33 -07:00 · 39070910db
commit 39070910db
parent ac90e7be05
1 changed files with 1 additions and 0 deletions
--- a/tensorflow/python/keras/benchmarks/BUILD
+++ b/tensorflow/python/keras/benchmarks/BUILD
@ -77,6 +77,7 @@ cuda_py_test(
    deps = [
        ":profiler_lib",
        "//tensorflow:tensorflow_py",
+        "//tensorflow/python/keras/utils:tf_inspect",
    ],
 )