From 62bcfd96774f3849dd01ba08dd7e9f72d38e2fe5 Mon Sep 17 00:00:00 2001
From: Anna R <annarev@google.com>
Date: Wed, 29 Aug 2018 15:07:59 -0700
Subject: 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
---
 tensorflow/BUILD | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'tensorflow/BUILD')

diff --git a/tensorflow/BUILD b/tensorflow/BUILD
index 9cc4c4567b..b5e0a4e98b 100644
--- a/tensorflow/BUILD
+++ b/tensorflow/BUILD
@@ -23,6 +23,10 @@ load(
     "//tensorflow/python/tools/api/generator:api_gen.bzl",
     "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",
 )
 
-- 
cgit v1.2.3