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

Update benchmark to be a binary instead of test

Browse Source 
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
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/python/ops/numpy_ops/integration_test/benchmarks/BUILD
View File

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