• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1package(default_visibility = ["//tensorflow/python/profiler:__subpackages__"])
2
3licenses(["notice"])  # Apache 2.0
4
5load("//tensorflow:tensorflow.bzl", "cuda_py_test")
6load("//tensorflow:tensorflow.bzl", "py_test")
7load("//tensorflow:tensorflow.bzl", "tf_py_test")
8load("//tensorflow:tensorflow.bzl", "tf_py_wrap_cc")
9
10py_library(
11    name = "flops_registry",
12    srcs = ["flops_registry.py"],
13    srcs_version = "PY2AND3",
14    deps = [
15        "//tensorflow/python:framework_ops",
16        "//tensorflow/python:graph_util",
17    ],
18)
19
20py_library(
21    name = "model_analyzer_testlib",
22    srcs = ["model_analyzer_testlib.py"],
23    srcs_version = "PY2AND3",
24    visibility = ["//visibility:public"],
25    deps = [
26        "//tensorflow/python:array_ops",
27        "//tensorflow/python:framework_for_generated_wrappers",
28        "//tensorflow/python:init_ops",
29        "//tensorflow/python:math_ops",
30        "//tensorflow/python:nn_grad",
31        "//tensorflow/python:nn_ops",
32        "//tensorflow/python:rnn",
33        "//tensorflow/python:rnn_cell",
34        "//tensorflow/python:tensor_array_grad",
35        "//tensorflow/python:training",
36        "//tensorflow/python:variable_scope",
37    ],
38)
39
40py_test(
41    name = "print_model_analysis_test",
42    srcs = ["print_model_analysis_test.py"],
43    srcs_version = "PY2AND3",
44    deps = [
45        "//tensorflow/core/profiler:protos_all_py",
46        "//tensorflow/python:array_ops",
47        "//tensorflow/python:client",
48        "//tensorflow/python:client_testlib",
49        "//tensorflow/python:framework_for_generated_wrappers",
50        "//tensorflow/python:init_ops",
51        "//tensorflow/python:nn_ops",
52        "//tensorflow/python:pywrap_tensorflow",
53        "//tensorflow/python:variable_scope",
54    ],
55)
56
57cuda_py_test(
58    name = "run_metadata_test",
59    srcs = ["run_metadata_test.py"],
60    additional_deps = [
61        ":model_analyzer_testlib",
62        "//tensorflow/core/profiler:protos_all_py",
63        "//tensorflow/python:array_ops",
64        "//tensorflow/python:client_testlib",
65        "//tensorflow/python:math_ops",
66        "//tensorflow/python/profiler:model_analyzer",
67        "//tensorflow/python:random_ops",
68    ],
69    tags = [
70        "no_pip",
71    ],
72    xla_enable_strict_auto_jit = False,  # Node names are different with autojit
73)
74