• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# Description:
2#   Common functionality for TensorFlow tooling
3
4load("//tensorflow:tensorflow.bzl", "py_test")
5
6package(
7    default_visibility = ["//tensorflow:__subpackages__"],
8    licenses = ["notice"],  # Apache 2.0
9)
10
11exports_files(["LICENSE"])
12
13py_library(
14    name = "public_api",
15    srcs = ["public_api.py"],
16    srcs_version = "PY2AND3",
17    deps = [
18        "//tensorflow/python:util",
19        "@six_archive//:six",
20    ],
21)
22
23py_test(
24    name = "public_api_test",
25    srcs = ["public_api_test.py"],
26    python_version = "PY3",
27    srcs_version = "PY2AND3",
28    deps = [
29        ":public_api",
30        "//tensorflow/python:platform_test",
31    ],
32)
33
34py_library(
35    name = "traverse",
36    srcs = ["traverse.py"],
37    srcs_version = "PY2AND3",
38    deps = [
39        "//tensorflow/python:util",
40        "@six_archive//:six",
41    ],
42)
43
44py_test(
45    name = "traverse_test",
46    srcs = ["traverse_test.py"],
47    python_version = "PY3",
48    srcs_version = "PY2AND3",
49    deps = [
50        ":test_module1",
51        ":test_module2",
52        ":traverse",
53        "//tensorflow/python:platform_test",
54    ],
55)
56
57py_library(
58    name = "test_module1",
59    srcs = ["test_module1.py"],
60    srcs_version = "PY2AND3",
61    deps = [
62        ":test_module2",
63    ],
64)
65
66py_library(
67    name = "test_module2",
68    srcs = ["test_module2.py"],
69    srcs_version = "PY2AND3",
70)
71