experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[tf.data] Internal change.

Browse Source 
PiperOrigin-RevId: 314389120
Change-Id: I87b049e02b1aa7a658716b62acd1420bb5ac76ee
This commit is contained in:
Jiri Simsa 2020-06-02 13:06:15 -07:00 committed by TensorFlower Gardener
parent 255f590ab6
commit dca04c0a9b
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tensorflow/python/data/experimental/benchmarks/BUILD
View File

@ -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(