aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/examples/benchmark/BUILD
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",
)