diff options
author | 2016-07-29 14:53:45 -0800 | |
---|---|---|
committer | 2016-07-29 16:03:48 -0700 | |
commit | 788f359b7218ad46696c15459c89688ffe70955e (patch) | |
tree | b1d3becb5cccdd977e6438d6b2cfe079aaa4ad15 /tensorflow/c/BUILD | |
parent | 76c39bc1e7b06bf427a478d5c453777cf7882389 (diff) |
Move C API files (and related files used by SWIG wrappers)
to new tensorflow/c directory.
Change: 128855990
Diffstat (limited to 'tensorflow/c/BUILD')
-rw-r--r-- | tensorflow/c/BUILD | 95 |
1 files changed, 95 insertions, 0 deletions
diff --git a/tensorflow/c/BUILD b/tensorflow/c/BUILD new file mode 100644 index 0000000000..fec8ca759e --- /dev/null +++ b/tensorflow/c/BUILD @@ -0,0 +1,95 @@ +# Description: +# C API for TensorFlow, for use by client language bindings. + +licenses(["notice"]) # Apache 2.0 + +load( + "//tensorflow:tensorflow.bzl", + "tf_cc_test", + "tf_cuda_library", +) + +# For platform specific build config +load( + "//tensorflow/core:platform/default/build_config.bzl", + "tf_kernel_tests_linkstatic", +) + +# ----------------------------------------------------------------------------- +# Public targets + +tf_cuda_library( + name = "c_api", + srcs = ["c_api.cc"], + hdrs = ["c_api.h"], + visibility = ["//visibility:public"], + deps = [ + "//tensorflow/core:core_cpu", + "//tensorflow/core:framework", + "//tensorflow/core:lib", + ], +) + +tf_cuda_library( + name = "tf_status_helper", + srcs = ["tf_status_helper.cc"], + hdrs = ["tf_status_helper.h"], + visibility = ["//visibility:public"], + deps = [ + ":c_api", + "//tensorflow/core:lib", + ], +) + +tf_cuda_library( + name = "checkpoint_reader", + srcs = ["checkpoint_reader.cc"], + hdrs = ["checkpoint_reader.h"], + visibility = ["//visibility:public"], + deps = [ + ":tf_status_helper", + "//tensorflow/core:framework", + "//tensorflow/core:lib", + ], +) + +# ----------------------------------------------------------------------------- +# Tests + +tf_cc_test( + name = "c_api_test", + size = "small", + linkopts = select({ + "//tensorflow:darwin": ["-headerpad_max_install_names"], + "//conditions:default": [], + }), + linkstatic = tf_kernel_tests_linkstatic(), + deps = [ + ":c_api", + "//tensorflow/core:direct_session", + "//tensorflow/core:framework", + "//tensorflow/core:proto_text", + "//tensorflow/core:protos_all_cc", + "//tensorflow/core:test", + "//tensorflow/core:test_main", + "//tensorflow/core:testlib", + "//tensorflow/core/kernels:array", + "//tensorflow/core/kernels:math", + "//third_party/eigen3", + ], +) + +# ----------------------------------------------------------------------------- +# Google-internal targets. + +filegroup( + name = "all_files", + srcs = glob( + ["**/*"], + exclude = [ + "**/METADATA", + "**/OWNERS", + ], + ), + visibility = ["//tensorflow:__subpackages__"], +) |