aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/cc/BUILD
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/cc/BUILD')
-rw-r--r--tensorflow/cc/BUILD89
1 files changed, 89 insertions, 0 deletions
diff --git a/tensorflow/cc/BUILD b/tensorflow/cc/BUILD
new file mode 100644
index 0000000000..8a5bf87a29
--- /dev/null
+++ b/tensorflow/cc/BUILD
@@ -0,0 +1,89 @@
+# Description:
+# TensorFlow is a computational framework, primarily for use in machine
+# learning applications.
+
+package(default_visibility = ["//tensorflow:internal"])
+
+licenses(["notice"]) # Apache 2.0
+
+exports_files(["LICENSE"])
+
+load("/tensorflow/tensorflow", "tf_copts")
+load("/tensorflow/tensorflow", "tf_gen_op_wrappers_cc")
+
+cc_library(
+ name = "cc_op_gen_main",
+ srcs = [
+ "ops/cc_op_gen.cc",
+ "ops/cc_op_gen_main.cc",
+ ],
+ hdrs = ["ops/cc_op_gen.h"],
+ copts = tf_copts(),
+ deps = [
+ "//tensorflow/core:framework",
+ ],
+)
+
+# Generates a library that contains C++ wrappers for ops.
+tf_gen_op_wrappers_cc(
+ name = "cc_ops",
+ op_lib_names = [
+ "array_ops",
+ "attention_ops",
+ "candidate_sampling_ops",
+ "control_flow_ops",
+ "data_flow_ops",
+ "image_ops",
+ "io_ops",
+ "linalg_ops",
+ "logging_ops",
+ "math_ops",
+ "nn_ops",
+ "no_op",
+ "parsing_ops",
+ "random_ops",
+ "sendrecv_ops",
+ "sparse_ops",
+ "state_ops",
+ "string_ops",
+ "summary_ops",
+ "training_ops",
+ "user_ops",
+ ],
+ other_hdrs = [
+ "ops/const_op.h",
+ "ops/standard_ops.h",
+ ],
+ other_srcs = [
+ "ops/const_op.cc",
+ ] + glob(["ops/*_grad.cc"]),
+ pkg = "//tensorflow/core",
+)
+
+cc_binary(
+ name = "tutorials_example_trainer",
+ srcs = ["tutorials/example_trainer.cc"],
+ copts = tf_copts(),
+ linkopts = [
+ "-lpthread",
+ "-lm",
+ ],
+ deps = [
+ ":cc_ops",
+ "//tensorflow/core:kernels",
+ "//tensorflow/core:local",
+ "//tensorflow/core:tensorflow",
+ ],
+)
+
+filegroup(
+ name = "all_files",
+ srcs = glob(
+ ["**/*"],
+ exclude = [
+ "**/METADATA",
+ "**/OWNERS",
+ ],
+ ),
+ visibility = ["//tensorflow:__subpackages__"],
+)