diff options
-rw-r--r-- | tensorflow/BUILD | 5 | ||||
-rw-r--r-- | tensorflow/python/tools/api/generator/BUILD | 3 |
2 files changed, 7 insertions, 1 deletions
diff --git a/tensorflow/BUILD b/tensorflow/BUILD index 9cc4c4567b..b5e0a4e98b 100644 --- a/tensorflow/BUILD +++ b/tensorflow/BUILD @@ -24,6 +24,10 @@ load( "gen_api_init_files", # @unused ) load( + "//tensorflow/python/tools/api/generator:api_init_files_v1.bzl", + "TENSORFLOW_API_INIT_FILES_V1", # @unused +) +load( "//third_party/ngraph:build_defs.bzl", "if_ngraph", ) @@ -589,6 +593,7 @@ gen_api_init_files( name = "tensorflow_python_api_gen", srcs = ["api_template.__init__.py"], api_version = 1, + output_files = TENSORFLOW_API_INIT_FILES_V1, root_init_template = "api_template.__init__.py", ) diff --git a/tensorflow/python/tools/api/generator/BUILD b/tensorflow/python/tools/api/generator/BUILD index 36af091163..90be2cc4f7 100644 --- a/tensorflow/python/tools/api/generator/BUILD +++ b/tensorflow/python/tools/api/generator/BUILD @@ -6,6 +6,7 @@ licenses(["notice"]) # Apache 2.0 load("//tensorflow:tensorflow.bzl", "py_test") load("//tensorflow/python/tools/api/generator:api_gen.bzl", "ESTIMATOR_API_INIT_FILES") load("//tensorflow/python/tools/api/generator:api_init_files.bzl", "TENSORFLOW_API_INIT_FILES") +load("//tensorflow/python/tools/api/generator:api_init_files_v1.bzl", "TENSORFLOW_API_INIT_FILES_V1") exports_files( [ @@ -55,7 +56,7 @@ py_test( args = [ "--package=tensorflow.python", "--api_name=tensorflow", - ] + TENSORFLOW_API_INIT_FILES, + ] + TENSORFLOW_API_INIT_FILES + TENSORFLOW_API_INIT_FILES_V1, main = "doc_srcs_test.py", srcs_version = "PY2AND3", deps = [ |