aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/build_defs.bzl
diff options
context:
space:
mode:
authorGravatar Anna R <annarev@google.com>2017-12-04 12:31:03 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-12-04 12:37:12 -0800
commit8f1e63d5629bda4f6c91fdec7a3b8418ed96786e (patch)
tree0f56fddabcd4e5cf91090acf5152cc54b3651e6a /tensorflow/python/build_defs.bzl
parenta1c29139ccf441ad4de97c4e7fe2729e6130fcb8 (diff)
Actually use ApiDef when generating Python API.
PiperOrigin-RevId: 177851421
Diffstat (limited to 'tensorflow/python/build_defs.bzl')
-rw-r--r--tensorflow/python/build_defs.bzl4
1 files changed, 4 insertions, 0 deletions
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",
+ ],
)