From dca04c0a9b4dad68caca50859252b7b14ec43cf1 Mon Sep 17 00:00:00 2001
From: Jiri Simsa <jsimsa@google.com>
Date: Tue, 2 Jun 2020 13:06:15 -0700
Subject: [PATCH] [tf.data] Internal change.

PiperOrigin-RevId: 314389120
Change-Id: I87b049e02b1aa7a658716b62acd1420bb5ac76ee
---
 .../python/data/experimental/benchmarks/BUILD    | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

diff --git a/tensorflow/python/data/experimental/benchmarks/BUILD b/tensorflow/python/data/experimental/benchmarks/BUILD
index 9a8c7e2164a..e3ca2d52ab5 100644
--- a/tensorflow/python/data/experimental/benchmarks/BUILD
+++ b/tensorflow/python/data/experimental/benchmarks/BUILD
@@ -7,9 +7,19 @@ package(
 
 exports_files(["LICENSE"])
 
-exports_files(
-    ["autotune_benchmark.py"],
-    visibility = ["//tensorflow:internal"],
+py_binary(
+    name = "autotune_benchmark_binary",
+    srcs = ["autotune_benchmark.py"],
+    main = "autotune_benchmark.py",
+    python_version = "PY3",
+    srcs_version = "PY2AND3",
+    deps = [
+        "//tensorflow/python:client_testlib",
+        "//tensorflow/python:math_ops",
+        "//tensorflow/python:session",
+        "//tensorflow/python/data/ops:dataset_ops",
+        "//third_party/py/numpy",
+    ],
 )
 
 tf_py_test(