diff options
Diffstat (limited to 'tensorflow/compiler/xla/client/lib/BUILD')
-rw-r--r-- | tensorflow/compiler/xla/client/lib/BUILD | 60 |
1 files changed, 60 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/client/lib/BUILD b/tensorflow/compiler/xla/client/lib/BUILD new file mode 100644 index 0000000000..e185beaedd --- /dev/null +++ b/tensorflow/compiler/xla/client/lib/BUILD @@ -0,0 +1,60 @@ +# Common computation builders for XLA. + +licenses(["notice"]) # Apache 2.0 + +package(default_visibility = ["//tensorflow/compiler/xla/client:friends"]) + +# Filegroup used to collect source files for dependency checking. +filegroup( + name = "c_srcs", + data = glob([ + "**/*.cc", + "**/*.h", + ]), +) + +cc_library( + name = "arithmetic", + srcs = ["arithmetic.cc"], + hdrs = ["arithmetic.h"], + deps = [ + "//tensorflow/compiler/xla:shape_util", + "//tensorflow/compiler/xla:types", + "//tensorflow/compiler/xla:xla_data_proto", + "//tensorflow/compiler/xla/client:computation", + "//tensorflow/compiler/xla/client:computation_builder", + ], +) + +cc_library( + name = "testing", + srcs = ["testing.cc"], + hdrs = ["testing.h"], + deps = [ + "//tensorflow/compiler/xla:literal_util", + "//tensorflow/compiler/xla:shape_util", + "//tensorflow/compiler/xla:statusor", + "//tensorflow/compiler/xla:types", + "//tensorflow/compiler/xla:util", + "//tensorflow/compiler/xla:xla_data_proto", + "//tensorflow/compiler/xla/client", + "//tensorflow/compiler/xla/client:computation", + "//tensorflow/compiler/xla/client:computation_builder", + "//tensorflow/compiler/xla/client:global_data", + "//tensorflow/core:lib", + ], +) + +# ----------------------------------------------------------------------------- + +filegroup( + name = "all_files", + srcs = glob( + ["**/*"], + exclude = [ + "**/METADATA", + "**/OWNERS", + ], + ), + visibility = ["//tensorflow:__subpackages__"], +) |