diff options
author | Anna R <annarev@google.com> | 2018-08-29 15:07:59 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-29 15:13:34 -0700 |
commit | 62bcfd96774f3849dd01ba08dd7e9f72d38e2fe5 (patch) | |
tree | 70b4aa3a96eb953c4fa1f2c2ae7dcbcdfb69e5cf /tensorflow/BUILD | |
parent | 82491c0930922ee6f8675e0953d6a2a1ef8b9fe5 (diff) |
Pass V1 __init__.py file list when generating API. Also, check against both TENSORFLOW_API_INIT_FILES and TENSORFLOW_API_INIT_FILES_V1 in doc_srcs_test.
PiperOrigin-RevId: 210798675
Diffstat (limited to 'tensorflow/BUILD')
-rw-r--r-- | tensorflow/BUILD | 5 |
1 files changed, 5 insertions, 0 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", ) |