aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/tfprof/BUILD
blob: e7f4ebdd36aa9d21ec1dc71ed200001eb0331704 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
licenses(["notice"])  # Apache 2.0

exports_files(["LICENSE"])

package(default_visibility = ["//tensorflow:__subpackages__"])

py_library(
    name = "tfprof",
    srcs = [
        "__init__.py",
        "model_analyzer.py",
        "tfprof_logger.py",
    ],
    srcs_version = "PY2AND3",
    visibility = ["//tensorflow:__subpackages__"],
    deps = [
        "//tensorflow/python:util",
        "//tensorflow/python/profiler:model_analyzer",
        "//tensorflow/python/profiler:profile_context",
        "//tensorflow/python/profiler:tfprof_logger",
    ],
)