aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/c/BUILD
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2016-07-29 14:53:45 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2016-07-29 16:03:48 -0700
commit788f359b7218ad46696c15459c89688ffe70955e (patch)
treeb1d3becb5cccdd977e6438d6b2cfe079aaa4ad15 /tensorflow/c/BUILD
parent76c39bc1e7b06bf427a478d5c453777cf7882389 (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/BUILD95
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__"],
+)