Update benchmark to be a binary instead of test

PiperOrigin-RevId: 322477922
Change-Id: I1fd810f37e26e5141f96bbdd7dbb14bd84d8c1f6
This commit is contained in:
Akshay Modi 2020-07-21 17:34:25 -07:00 committed by TensorFlower Gardener
parent 4da55605f0
commit 147843127e

View File

@ -1,5 +1,3 @@
load("//tensorflow:tensorflow.bzl", "cuda_py_test")
package(
default_visibility = [
"//tensorflow:internal",
@ -7,11 +5,10 @@ package(
licenses = ["notice"], # Apache 2.0
)
cuda_py_test(
py_binary(
name = "micro_benchmarks",
srcs = ["micro_benchmarks.py"],
python_version = "PY3",
shard_count = 5,
deps = [
":numpy_mlp",
":tf_numpy_mlp",