blob: 98611a9aadf6f456dd4f9fe4f423e3e2ce9722ec (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
|
# Description:
# Examples of adding a benchmark to TensorFlow.
load(
"//tensorflow/tools/test:performance.bzl",
"tf_py_logged_benchmark",
)
licenses(["notice"]) # Apache 2.0
exports_files(["LICENSE"])
py_test(
name = "sample_benchmark",
srcs = ["sample_benchmark.py"],
srcs_version = "PY2AND3",
deps = [
"//tensorflow:tensorflow_py",
],
)
tf_py_logged_benchmark(
name = "sample_logged_benchmark",
target = "//tensorflow/examples/benchmark:sample_benchmark",
)
|