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