aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/examples/benchmark/BUILD
blob: c4bb0a5bd952ea175a4fd2444a3d632dc13445de (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
26
27
28
29
30
31
# 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",
)

filegroup(
    name = "all_files",
    srcs = glob(["**/*"]),
    visibility = ["//tensorflow:__subpackages__"],
)