diff options
author | Skye Wanderman-Milne <skyewm@google.com> | 2017-07-12 13:08:51 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-07-12 13:13:16 -0700 |
commit | 45a58d378ea081907d1aa7d8963c40e7fb3dcfe8 (patch) | |
tree | bd43bbeeb166561fca5e9e3ae1a0149305094b39 /tensorflow/c/BUILD | |
parent | 4f54336348fb40fb5eea2166b949f4457f447ef5 (diff) |
Introduce Python-only extensions to the C API
Implements an incomplete version of Operation._add_control_input()
using a new extension to make sure the plumbing works.
This also adds header guards to c_api_internal.h, which were missing. For some reason the missing guards caused problems in the cmake build even though there doesn't appear to be any #include cycles.
PiperOrigin-RevId: 161705859
Diffstat (limited to 'tensorflow/c/BUILD')
-rw-r--r-- | tensorflow/c/BUILD | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/tensorflow/c/BUILD b/tensorflow/c/BUILD index 9267ef77ef..85429f8c66 100644 --- a/tensorflow/c/BUILD +++ b/tensorflow/c/BUILD @@ -145,6 +145,19 @@ tf_custom_op_library( ) # ----------------------------------------------------------------------------- +# Python API target + +tf_cuda_library( + name = "python_api", + srcs = ["python_api.cc"], + hdrs = ["python_api.h"], + visibility = ["//tensorflow/python:__pkg__"], + deps = [ + ":c_api_internal", + ], +) + +# ----------------------------------------------------------------------------- # Google-internal targets. filegroup( |