From 8f1e63d5629bda4f6c91fdec7a3b8418ed96786e Mon Sep 17 00:00:00 2001
From: Anna R <annarev@google.com>
Date: Mon, 4 Dec 2017 12:31:03 -0800
Subject: Actually use ApiDef when generating Python API.

PiperOrigin-RevId: 177851421
---
 tensorflow/python/build_defs.bzl | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'tensorflow/python/build_defs.bzl')

diff --git a/tensorflow/python/build_defs.bzl b/tensorflow/python/build_defs.bzl
index 2d8625933f..48b03fab0f 100644
--- a/tensorflow/python/build_defs.bzl
+++ b/tensorflow/python/build_defs.bzl
@@ -27,4 +27,8 @@ def tf_gen_op_wrapper_private_py(name, out=None, deps=[],
     deps=deps,
     require_shape_functions=require_shape_functions,
     generated_target_name=name,
+    api_def_srcs = [
+        "//tensorflow/core:base_api_def",
+        "//tensorflow/core:python_api_def",
+    ],
   )
-- 
cgit v1.2.3