• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# Description:
2# Scripts used to generate TensorFlow Python API.
3
4licenses(["notice"])  # Apache 2.0
5
6load("//tensorflow:tensorflow.bzl", "py_test")
7load("//tensorflow/python/tools/api/generator:api_init_files.bzl", "TENSORFLOW_API_INIT_FILES")
8load("//tensorflow/python/tools/api/generator:api_init_files_v1.bzl", "TENSORFLOW_API_INIT_FILES_V1")
9load("//tensorflow/python/tools/api/generator:api_init_files.bzl", "KERAS_API_INIT_FILES")
10load("//tensorflow/python/tools/api/generator:api_init_files_v1.bzl", "KERAS_API_INIT_FILES_V1")
11
12exports_files(
13    [
14        "LICENSE",
15        "create_python_api.py",
16    ],
17)
18
19py_library(
20    name = "create_python_api",
21    srcs = ["//tensorflow/python/tools/api/generator:create_python_api.py"],
22    srcs_version = "PY2AND3",
23    visibility = ["//visibility:public"],
24    deps = [
25        "//tensorflow/python:util",
26        "//tensorflow/python/tools/api/generator:doc_srcs",
27    ],
28)
29
30py_library(
31    name = "doc_srcs",
32    srcs = ["doc_srcs.py"],
33    srcs_version = "PY2AND3",
34    visibility = ["//visibility:public"],
35    deps = [
36        "//tensorflow/python:util",
37    ],
38)
39
40py_test(
41    name = "create_python_api_test",
42    srcs = [
43        "create_python_api.py",
44        "create_python_api_test.py",
45    ],
46    srcs_version = "PY2AND3",
47    deps = [
48        ":doc_srcs",
49        "//tensorflow/python:client_testlib",
50        "//tensorflow/python:no_contrib",
51    ],
52)
53
54py_test(
55    name = "tensorflow_doc_srcs_test",
56    srcs = ["doc_srcs_test.py"],
57    args = [
58        "--package=tensorflow.python",
59        "--api_name=tensorflow",
60    ] + KERAS_API_INIT_FILES + KERAS_API_INIT_FILES_V1 + TENSORFLOW_API_INIT_FILES + TENSORFLOW_API_INIT_FILES_V1,
61    main = "doc_srcs_test.py",
62    srcs_version = "PY2AND3",
63    deps = [
64        ":doc_srcs",
65        "//tensorflow/python:client_testlib",
66        "//tensorflow/python:no_contrib",
67    ],
68)
69
70py_test(
71    name = "output_init_files_test",
72    srcs = ["output_init_files_test.py"],
73    data = [
74        "api_init_files.bzl",
75        "api_init_files_v1.bzl",
76    ],
77    srcs_version = "PY2AND3",
78    tags = ["no_pip"],
79    deps = [
80        "//tensorflow/python:client_testlib",
81        "//tensorflow/python:no_contrib",
82        "//tensorflow/python/tools/api/generator:create_python_api",
83    ],
84)
85