diff options
Diffstat (limited to 'tensorflow/cc/BUILD')
-rw-r--r-- | tensorflow/cc/BUILD | 89 |
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__"], +) |