aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/compiler/xla/client/BUILD
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/compiler/xla/client/BUILD')
-rw-r--r--tensorflow/compiler/xla/client/BUILD175
1 files changed, 175 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/client/BUILD b/tensorflow/compiler/xla/client/BUILD
new file mode 100644
index 0000000000..3e9dfe2a92
--- /dev/null
+++ b/tensorflow/compiler/xla/client/BUILD
@@ -0,0 +1,175 @@
+# Description:
+# XLA client libraries.
+
+licenses(["notice"]) # Apache 2.0
+
+package(default_visibility = [":friends"])
+
+package_group(
+ name = "friends",
+ includes = [
+ "//tensorflow/compiler/xla:friends",
+ ],
+)
+
+# Filegroup used to collect source files for dependency checking.
+filegroup(
+ name = "c_srcs",
+ data = glob([
+ "**/*.cc",
+ "**/*.h",
+ ]),
+)
+
+cc_library(
+ name = "global_data",
+ srcs = ["global_data.cc"],
+ hdrs = ["global_data.h"],
+ deps = [
+ "//tensorflow/compiler/xla:service_interface",
+ "//tensorflow/compiler/xla:types",
+ "//tensorflow/compiler/xla:xla_data_proto",
+ "//tensorflow/compiler/xla:xla_proto",
+ "//tensorflow/core:lib",
+ ],
+)
+
+cc_library(
+ name = "padding",
+ srcs = ["padding.cc"],
+ hdrs = ["padding.h"],
+ deps = [
+ "//tensorflow/compiler/xla:types",
+ "//tensorflow/core:lib",
+ ],
+)
+
+cc_test(
+ name = "padding_test",
+ srcs = ["padding_test.cc"],
+ deps = [
+ ":padding",
+ "//tensorflow/core:test",
+ "//tensorflow/core:test_main",
+ ],
+)
+
+cc_library(
+ name = "client",
+ srcs = ["client.cc"],
+ hdrs = ["client.h"],
+ deps = [
+ ":computation",
+ ":global_data",
+ "//tensorflow/compiler/xla:literal_util",
+ "//tensorflow/compiler/xla:service_interface",
+ "//tensorflow/compiler/xla:status_macros",
+ "//tensorflow/compiler/xla:statusor",
+ "//tensorflow/compiler/xla:types",
+ "//tensorflow/compiler/xla:util",
+ "//tensorflow/compiler/xla:xla_data_proto",
+ "//tensorflow/compiler/xla:xla_proto",
+ "//tensorflow/compiler/xla/service:session_proto",
+ "//tensorflow/core:lib",
+ ],
+)
+
+cc_library(
+ name = "local_client",
+ srcs = ["local_client.cc"],
+ hdrs = ["local_client.h"],
+ deps = [
+ ":client",
+ ":computation",
+ "//tensorflow/compiler/xla:executable_run_options",
+ "//tensorflow/compiler/xla:status_macros",
+ "//tensorflow/compiler/xla:statusor",
+ "//tensorflow/compiler/xla:util",
+ "//tensorflow/compiler/xla:xla_data_proto",
+ "//tensorflow/compiler/xla/service:backend",
+ "//tensorflow/compiler/xla/service:compiler",
+ "//tensorflow/compiler/xla/service:device_memory_allocator",
+ "//tensorflow/compiler/xla/service:executable",
+ "//tensorflow/compiler/xla/service:local_service",
+ "//tensorflow/compiler/xla/service:shaped_buffer",
+ "//tensorflow/compiler/xla/service/llvm_ir:llvm_util",
+ "//tensorflow/core:lib",
+ "//tensorflow/core:stream_executor_no_cuda",
+ "@llvm//:support",
+ ],
+)
+
+# This target is used to instantiate the XLA service in-process and create
+# a client for it.
+cc_library(
+ name = "client_library",
+ srcs = ["client_library.cc"],
+ hdrs = ["client_library.h"],
+ deps = [
+ ":local_client",
+ "//tensorflow/compiler/xla:status_macros",
+ "//tensorflow/compiler/xla:statusor",
+ "//tensorflow/compiler/xla:types",
+ "//tensorflow/compiler/xla:util",
+ "//tensorflow/compiler/xla/service:backend",
+ "//tensorflow/compiler/xla/service:device_memory_allocator",
+ "//tensorflow/compiler/xla/service:local_service",
+ "//tensorflow/compiler/xla/service:platform_util",
+ "//tensorflow/core:lib",
+ "//tensorflow/core:stream_executor_no_cuda",
+ ],
+)
+
+cc_library(
+ name = "computation",
+ srcs = ["computation.cc"],
+ hdrs = ["computation.h"],
+ deps = [
+ "//tensorflow/compiler/xla:service_interface",
+ "//tensorflow/compiler/xla:status_macros",
+ "//tensorflow/compiler/xla:statusor",
+ "//tensorflow/compiler/xla:util",
+ "//tensorflow/compiler/xla:xla_data_proto",
+ "//tensorflow/compiler/xla:xla_proto",
+ "//tensorflow/compiler/xla/service:session_proto",
+ "//tensorflow/core:lib",
+ ],
+)
+
+cc_library(
+ name = "computation_builder",
+ srcs = ["computation_builder.cc"],
+ hdrs = ["computation_builder.h"],
+ deps = [
+ ":client",
+ ":computation",
+ ":global_data",
+ ":padding",
+ "//tensorflow/compiler/xla:array2d",
+ "//tensorflow/compiler/xla:array3d",
+ "//tensorflow/compiler/xla:array4d",
+ "//tensorflow/compiler/xla:literal_util",
+ "//tensorflow/compiler/xla:shape_util",
+ "//tensorflow/compiler/xla:status_macros",
+ "//tensorflow/compiler/xla:statusor",
+ "//tensorflow/compiler/xla:types",
+ "//tensorflow/compiler/xla:util",
+ "//tensorflow/compiler/xla:xla_data_proto",
+ "//tensorflow/compiler/xla:xla_proto",
+ "//tensorflow/core:lib",
+ ],
+)
+
+# -----------------------------------------------------------------------------
+
+filegroup(
+ name = "all_files",
+ srcs = glob(
+ ["**/*"],
+ exclude = [
+ "**/METADATA",
+ "**/OWNERS",
+ ],
+ ),
+ visibility = ["//tensorflow:__subpackages__"],
+)