• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1load("//tensorflow:tensorflow.bzl", "tf_py_test")
2
3package(
4    licenses = ["notice"],  # Apache 2.0
5)
6
7filegroup(
8    name = "all_files",
9    srcs = glob(
10        ["**/*"],
11        exclude = [
12            "**/METADATA",
13            "**/OWNERS",
14        ],
15    ),
16    visibility = ["//tensorflow:__subpackages__"],
17)
18
19py_library(
20    name = "impl",
21    srcs = [
22        "api.py",
23        "conversion.py",
24    ],
25    srcs_version = "PY2AND3",
26    visibility = ["//tensorflow:__subpackages__"],
27    deps = [
28        "//tensorflow/python:platform",
29        "//tensorflow/python:util",
30        "//tensorflow/python/autograph/converters",
31        "//tensorflow/python/autograph/core",
32        "//tensorflow/python/autograph/operators",
33        "//tensorflow/python/autograph/pyct",
34        "//tensorflow/python/autograph/pyct/static_analysis",
35        "//tensorflow/python/autograph/utils",
36        "@gast_archive//:gast",
37    ],
38)
39
40tf_py_test(
41    name = "api_test",
42    srcs = ["api_test.py"],
43    deps = [
44        ":impl",
45        "//tensorflow/python:client_testlib",
46        "//tensorflow/python/autograph/core:test_lib",
47        "//tensorflow/python/autograph/utils",
48        "//third_party/py/numpy",
49    ],
50)
51
52py_test(
53    name = "api_py3_test",
54    srcs = ["api_py3_test.py"],
55    python_version = "PY3",
56    srcs_version = "PY3",
57    tags = [
58        "no_oss_py2",
59        "no_pip",
60        "nopip",
61    ],
62    deps = [
63        ":impl",
64        "//tensorflow/python:client_testlib",
65        "//tensorflow/python:constant_op",
66        "//tensorflow/python/autograph/core",
67    ],
68)
69
70tf_py_test(
71    name = "conversion_test",
72    srcs = ["conversion_test.py"],
73    deps = [
74        ":impl",
75        "//tensorflow/python:client_testlib",
76        "@gast_archive//:gast",
77    ],
78)
79