aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/compiler/xla/BUILD
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/compiler/xla/BUILD')
-rw-r--r--tensorflow/compiler/xla/BUILD561
1 files changed, 561 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/BUILD b/tensorflow/compiler/xla/BUILD
new file mode 100644
index 0000000000..caa5141da4
--- /dev/null
+++ b/tensorflow/compiler/xla/BUILD
@@ -0,0 +1,561 @@
+licenses(["notice"]) # Apache 2.0
+
+package(default_visibility = ["//tensorflow:internal"])
+
+package_group(
+ name = "friends",
+ packages = [
+ "//tensorflow/compiler/...",
+ ],
+)
+
+package_group(
+ name = "internal",
+ packages = [
+ "//tensorflow/compiler/xla/...",
+ ],
+)
+
+load("//tensorflow/compiler/xla:xla.bzl", "xla_proto_library")
+
+# Filegroup used to collect source files for dependency checking.
+filegroup(
+ name = "c_srcs",
+ data = glob([
+ "**/*.cc",
+ "**/*.h",
+ ]),
+)
+
+xla_proto_library(
+ name = "xla_data_proto",
+ srcs = ["xla_data.proto"],
+ visibility = ["//visibility:public"],
+)
+
+xla_proto_library(
+ name = "xla_proto",
+ srcs = ["xla.proto"],
+ visibility = ["//visibility:public"],
+ deps = [
+ ":xla_data_proto",
+ "//tensorflow/compiler/xla/service:session_proto",
+ ],
+)
+
+cc_library(
+ name = "types",
+ hdrs = ["types.h"],
+ visibility = [":friends"],
+ deps = ["//tensorflow/core:lib"],
+)
+
+cc_library(
+ name = "service_interface",
+ srcs = [],
+ hdrs = ["service_interface.h"],
+ visibility = [":friends"],
+ deps = [
+ ":xla_proto",
+ "//tensorflow/core:lib",
+ ],
+)
+
+cc_library(
+ name = "status_macros",
+ srcs = ["status_macros.cc"],
+ hdrs = ["status_macros.h"],
+ visibility = [":friends"],
+ deps = [
+ ":statusor",
+ ":types",
+ "//tensorflow/core:lib",
+ ],
+)
+
+cc_test(
+ name = "status_macros_test",
+ size = "small",
+ srcs = ["status_macros_test.cc"],
+ deps = [
+ ":status_macros",
+ ":statusor",
+ ":test_helpers",
+ "//tensorflow/core:lib",
+ "//tensorflow/core:test",
+ "//tensorflow/core:test_main",
+ ],
+)
+
+cc_library(
+ name = "status",
+ hdrs = ["status.h"],
+ visibility = ["//visibility:public"],
+ deps = [
+ "//tensorflow/core:lib",
+ "//tensorflow/core:lib_internal",
+ ],
+)
+
+cc_library(
+ name = "statusor",
+ srcs = ["statusor.cc"],
+ hdrs = ["statusor.h"],
+ visibility = ["//visibility:public"],
+ deps = [
+ ":status",
+ "//tensorflow/core:lib",
+ "//tensorflow/core:lib_internal",
+ ],
+)
+
+cc_test(
+ name = "statusor_test",
+ size = "small",
+ srcs = ["statusor_test.cc"],
+ deps = [
+ ":statusor",
+ ":types",
+ "//tensorflow/core:lib",
+ "//tensorflow/core:test",
+ "//tensorflow/core:test_main",
+ ],
+)
+
+cc_library(
+ name = "util",
+ srcs = ["util.cc"],
+ hdrs = [
+ "map_util.h",
+ "ptr_util.h",
+ "util.h",
+ ],
+ visibility = ["//visibility:public"],
+ deps = [
+ ":status",
+ ":types",
+ ":xla_data_proto",
+ "//tensorflow/compiler/xla/legacy_flags:util_flags",
+ "//tensorflow/core:lib",
+ ],
+)
+
+cc_library(
+ name = "protobuf_util",
+ srcs = ["protobuf_util.cc"],
+ hdrs = [
+ "protobuf_util.h",
+ ],
+ visibility = ["//visibility:public"],
+ deps = [
+ ":types",
+ "//tensorflow/core:lib",
+ ],
+)
+
+cc_test(
+ name = "util_test",
+ srcs = ["util_test.cc"],
+ deps = [
+ ":types",
+ ":util",
+ "//tensorflow/core:test",
+ "//tensorflow/core:test_main",
+ ],
+)
+
+cc_library(
+ name = "shape_util",
+ srcs = [
+ "index_util.cc",
+ "layout_util.cc",
+ "primitive_util.cc",
+ "shape_util.cc",
+ ],
+ hdrs = [
+ "index_util.h",
+ "layout_util.h",
+ "primitive_util.h",
+ "shape_util.h",
+ ],
+ visibility = ["//visibility:public"],
+ deps = [
+ ":protobuf_util",
+ ":status_macros",
+ ":statusor",
+ ":types",
+ ":util",
+ ":xla_data_proto",
+ "//tensorflow/compiler/xla/legacy_flags:layout_util_flags",
+ "//tensorflow/core:framework_internal",
+ "//tensorflow/core:lib",
+ "//tensorflow/core:regexp_internal",
+ ],
+)
+
+cc_test(
+ name = "shape_util_test",
+ srcs = ["shape_util_test.cc"],
+ deps = [
+ ":shape_util",
+ ":test_helpers",
+ ":types",
+ ":util",
+ "//tensorflow/core:test",
+ "//tensorflow/core:test_main",
+ ],
+)
+
+cc_test(
+ name = "layout_util_test",
+ srcs = ["layout_util_test.cc"],
+ deps = [
+ ":shape_util",
+ ":test_helpers",
+ "//tensorflow/compiler/xla/legacy_flags:layout_util_flags",
+ "//tensorflow/core:test",
+ "//tensorflow/core:test_main",
+ ],
+)
+
+cc_test(
+ name = "index_util_test",
+ srcs = ["index_util_test.cc"],
+ deps = [
+ ":shape_util",
+ ":test_helpers",
+ ":xla_data_proto",
+ "//tensorflow/core:test",
+ "//tensorflow/core:test_main",
+ ],
+)
+
+cc_library(
+ name = "literal_util",
+ srcs = ["literal_util.cc"],
+ hdrs = ["literal_util.h"],
+ visibility = ["//visibility:public"],
+ deps = [
+ ":array2d",
+ ":array3d",
+ ":array4d",
+ ":shape_util",
+ ":types",
+ ":util",
+ ":xla_data_proto",
+ "//tensorflow/core:lib",
+ ],
+)
+
+cc_test(
+ name = "literal_util_test",
+ srcs = ["literal_util_test.cc"],
+ deps = [
+ ":array3d",
+ ":array4d",
+ ":literal_util",
+ ":shape_util",
+ ":test_helpers",
+ ":types",
+ "//tensorflow/core:lib",
+ "//tensorflow/core:test",
+ "//tensorflow/core:test_main",
+ ],
+)
+
+cc_library(
+ name = "device_util",
+ hdrs = ["device_util.h"],
+ visibility = ["//visibility:public"],
+ deps = [
+ ":types",
+ "//tensorflow/core:lib",
+ "//tensorflow/core:stream_executor_no_cuda",
+ ],
+)
+
+cc_library(
+ name = "array2d",
+ srcs = ["array2d.cc"],
+ hdrs = ["array2d.h"],
+ visibility = ["//visibility:public"],
+ deps = [
+ ":types",
+ ":util",
+ "//tensorflow/core:lib",
+ ],
+)
+
+cc_test(
+ name = "array2d_test",
+ srcs = ["array2d_test.cc"],
+ deps = [
+ ":array2d",
+ "//tensorflow/core:test",
+ "//tensorflow/core:test_main",
+ ],
+)
+
+cc_library(
+ name = "array3d",
+ hdrs = ["array3d.h"],
+ visibility = [":friends"],
+ deps = [
+ ":types",
+ "//tensorflow/core:lib",
+ ],
+)
+
+cc_test(
+ name = "array3d_test",
+ srcs = ["array3d_test.cc"],
+ deps = [
+ ":array3d",
+ ":types",
+ "//tensorflow/core:test",
+ "//tensorflow/core:test_main",
+ ],
+)
+
+cc_library(
+ name = "array4d",
+ hdrs = ["array4d.h"],
+ visibility = [":friends"],
+ deps = [
+ ":array2d",
+ ":types",
+ "//tensorflow/core:lib",
+ ],
+)
+
+cc_test(
+ name = "array4d_test",
+ srcs = ["array4d_test.cc"],
+ deps = [
+ ":array4d",
+ "//tensorflow/core:lib",
+ "//tensorflow/core:test",
+ "//tensorflow/core:test_main",
+ ],
+)
+
+cc_library(
+ name = "executable_run_options",
+ srcs = ["executable_run_options.cc"],
+ hdrs = ["executable_run_options.h"],
+ visibility = ["//visibility:public"],
+)
+
+cc_library(
+ name = "differential_set",
+ hdrs = ["differential_set.h"],
+ visibility = [":internal"],
+ deps = [
+ "//tensorflow/core:lib",
+ ],
+)
+
+cc_test(
+ name = "differential_set_test",
+ srcs = ["differential_set_test.cc"],
+ deps = [
+ ":differential_set",
+ "//tensorflow/core:test",
+ "//tensorflow/core:test_main",
+ ],
+)
+
+cc_library(
+ name = "packed_literal_reader",
+ srcs = ["packed_literal_reader.cc"],
+ hdrs = ["packed_literal_reader.h"],
+ visibility = [":internal"],
+ deps = [
+ ":literal_util",
+ ":shape_util",
+ ":status_macros",
+ ":statusor",
+ ":types",
+ ":util",
+ ":xla_data_proto",
+ "//tensorflow/core:lib",
+ ],
+)
+
+cc_library(
+ name = "test_helpers",
+ testonly = 1,
+ srcs = ["test_helpers.cc"],
+ hdrs = ["test_helpers.h"],
+ visibility = [":internal"],
+ deps = [
+ ":statusor",
+ ":types",
+ "//tensorflow/core:lib",
+ "//tensorflow/core:regexp_internal",
+ "//tensorflow/core:test",
+ ],
+)
+
+cc_library(
+ name = "text_literal_reader",
+ srcs = ["text_literal_reader.cc"],
+ hdrs = ["text_literal_reader.h"],
+ visibility = [":internal"],
+ deps = [
+ ":literal_util",
+ ":shape_util",
+ ":status_macros",
+ ":statusor",
+ ":types",
+ ":util",
+ ":xla_data_proto",
+ "//tensorflow/core:lib",
+ "//tensorflow/core:lib_internal",
+ ],
+)
+
+cc_test(
+ name = "text_literal_reader_test",
+ srcs = ["text_literal_reader_test.cc"],
+ deps = [
+ ":literal_util",
+ ":shape_util",
+ ":text_literal_reader",
+ ":types",
+ ":xla_data_proto",
+ "//tensorflow/core:lib",
+ "//tensorflow/core:test",
+ "//tensorflow/core:test_main",
+ ],
+)
+
+cc_library(
+ name = "text_literal_writer",
+ srcs = ["text_literal_writer.cc"],
+ hdrs = ["text_literal_writer.h"],
+ visibility = [":internal"],
+ deps = [
+ ":literal_util",
+ ":shape_util",
+ ":status_macros",
+ ":types",
+ ":xla_data_proto",
+ "//tensorflow/core:lib",
+ ],
+)
+
+cc_test(
+ name = "text_literal_writer_test",
+ srcs = ["text_literal_writer_test.cc"],
+ deps = [
+ ":literal_util",
+ ":test_helpers",
+ ":text_literal_writer",
+ ":types",
+ "//tensorflow/core:lib",
+ "//tensorflow/core:test",
+ "//tensorflow/core:test_main",
+ ],
+)
+
+cc_library(
+ name = "shape_tree",
+ hdrs = ["shape_tree.h"],
+ visibility = ["//visibility:public"],
+ deps = [
+ ":shape_util",
+ ":status_macros",
+ ":util",
+ ":xla_data_proto",
+ "//tensorflow/core:lib",
+ ],
+)
+
+cc_test(
+ name = "shape_tree_test",
+ srcs = ["shape_tree_test.cc"],
+ deps = [
+ ":shape_tree",
+ ":shape_util",
+ ":xla_data_proto",
+ "//tensorflow/core:test",
+ "//tensorflow/core:test_main",
+ ],
+)
+
+cc_library(
+ name = "shape_layout",
+ srcs = ["shape_layout.cc"],
+ hdrs = ["shape_layout.h"],
+ visibility = ["//visibility:public"],
+ deps = [
+ ":shape_util",
+ ":types",
+ ":util",
+ ":xla_data_proto",
+ "//tensorflow/core:lib",
+ ],
+)
+
+cc_library(
+ name = "window_util",
+ srcs = ["window_util.cc"],
+ hdrs = ["window_util.h"],
+ visibility = ["//visibility:public"],
+ deps = [
+ ":types",
+ ":xla_data_proto",
+ "//tensorflow/core:lib",
+ ],
+)
+
+cc_library(
+ name = "reference_util",
+ srcs = ["reference_util.cc"],
+ hdrs = ["reference_util.h"],
+ visibility = ["//visibility:public"],
+ deps = [
+ ":array2d",
+ ":array3d",
+ ":array4d",
+ ":util",
+ ":window_util",
+ ":xla_data_proto",
+ "//tensorflow/compiler/xla/client:computation_builder",
+ "//tensorflow/compiler/xla/client:padding",
+ "//tensorflow/compiler/xla/service/cpu:runtime_single_threaded_matmul",
+ "//tensorflow/core:lib",
+ ],
+)
+
+cc_test(
+ name = "reference_util_test",
+ srcs = ["reference_util_test.cc"],
+ deps = [
+ ":array2d",
+ ":array4d",
+ ":literal_util",
+ ":reference_util",
+ ":util",
+ ":xla_data_proto",
+ "//tensorflow/compiler/xla/client:padding",
+ "//tensorflow/compiler/xla/tests:literal_test_util",
+ "//tensorflow/core:test",
+ "//tensorflow/core:test_main",
+ ],
+)
+
+# -----------------------------------------------------------------------------
+
+filegroup(
+ name = "all_files",
+ srcs = glob(
+ ["**/*"],
+ exclude = [
+ "**/METADATA",
+ "**/OWNERS",
+ ],
+ ),
+ visibility = ["//tensorflow:__subpackages__"],
+)
generated by cgit v1.2.3 (git 2.39.1) at 2024-06-08 17:29:23 +0000