aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/compiler/tf2xla/BUILD
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/compiler/tf2xla/BUILD')
-rw-r--r--tensorflow/compiler/tf2xla/BUILD193
1 files changed, 193 insertions, 0 deletions
diff --git a/tensorflow/compiler/tf2xla/BUILD b/tensorflow/compiler/tf2xla/BUILD
new file mode 100644
index 0000000000..3de9958cd6
--- /dev/null
+++ b/tensorflow/compiler/tf2xla/BUILD
@@ -0,0 +1,193 @@
+licenses(["notice"]) # Apache 2.0
+
+package_group(
+ name = "internal",
+ packages = [
+ "//tensorflow/compiler/aot/...",
+ "//tensorflow/compiler/jit/...",
+ "//tensorflow/compiler/tests/...",
+ "//tensorflow/compiler/tf2xla/...",
+ ],
+)
+
+package_group(
+ name = "friends",
+ includes = [":internal"],
+ packages = ["//tensorflow/..."],
+)
+
+package(
+ default_visibility = [":internal"],
+)
+
+cc_library(
+ name = "xla_compiler",
+ srcs = [
+ "op_registrations.cc",
+ "xla_compilation_device.cc",
+ "xla_compiler.cc",
+ "xla_context.cc",
+ "xla_helpers.cc",
+ "xla_op_kernel.cc",
+ ],
+ hdrs = [
+ "xla_compilation_device.h",
+ "xla_compiler.h",
+ "xla_context.h",
+ "xla_helpers.h",
+ "xla_op_kernel.h",
+ ],
+ deps = [
+ ":common",
+ ":dump_graph",
+ "//tensorflow/compiler/xla:literal_util",
+ "//tensorflow/compiler/xla:shape_util",
+ "//tensorflow/compiler/xla:statusor",
+ "//tensorflow/compiler/xla/client",
+ "//tensorflow/compiler/xla/client:client_library",
+ "//tensorflow/compiler/xla/client:computation_builder",
+ "//tensorflow/compiler/xla/client:local_client",
+ "//tensorflow/compiler/xla/service:cpu_plugin",
+ "//tensorflow/core:core_cpu",
+ "//tensorflow/core:core_cpu_internal",
+ "//tensorflow/core:framework",
+ "//tensorflow/core:lib",
+ "//tensorflow/core:lib_internal",
+ "//tensorflow/core:protos_all_cc",
+ "//tensorflow/core:stream_executor_no_cuda",
+ "//tensorflow/core:tensorflow_opensource",
+ "//tensorflow/core/kernels:cwise_op",
+ ],
+ alwayslink = 1,
+)
+
+cc_library(
+ name = "common",
+ srcs = [
+ "literal_util.cc",
+ "shape_util.cc",
+ "str_util.cc",
+ "type_util.cc",
+ ],
+ hdrs = [
+ "literal_util.h",
+ "shape_util.h",
+ "str_util.h",
+ "type_util.h",
+ ],
+ visibility = [":friends"],
+ deps = [
+ "//tensorflow/compiler/xla:literal_util",
+ "//tensorflow/compiler/xla:shape_util",
+ "//tensorflow/compiler/xla:xla_data_proto",
+ "//tensorflow/core:core_cpu_internal",
+ "//tensorflow/core:framework",
+ "//tensorflow/core:lib",
+ "//tensorflow/core:protos_all_cc",
+ ],
+)
+
+# Internal targets below this point.
+
+cc_test(
+ name = "str_util_test",
+ srcs = [
+ "str_util_test.cc",
+ ],
+ deps = [
+ ":common",
+ "//tensorflow/core:lib",
+ "//tensorflow/core:test",
+ "//tensorflow/core:test_main",
+ ],
+)
+
+cc_test(
+ name = "literal_util_test",
+ srcs = [
+ "literal_util_test.cc",
+ ],
+ deps = [
+ ":common",
+ "//tensorflow/compiler/xla:literal_util",
+ "//tensorflow/core:framework",
+ "//tensorflow/core:test",
+ "//tensorflow/core:test_main",
+ "//tensorflow/core:testlib",
+ ],
+)
+
+cc_library(
+ name = "const_analysis",
+ srcs = ["const_analysis.cc"],
+ hdrs = ["const_analysis.h"],
+ deps = [
+ "//tensorflow/core:core_cpu",
+ "//tensorflow/core:core_cpu_internal",
+ "//tensorflow/core:framework",
+ "//tensorflow/core:lib",
+ "//tensorflow/core:lib_internal",
+ "//tensorflow/core:protos_all_cc",
+ ],
+)
+
+cc_test(
+ name = "const_analysis_test",
+ size = "small",
+ srcs = ["const_analysis_test.cc"],
+ deps = [
+ ":const_analysis",
+ "//tensorflow/cc:cc_ops",
+ "//tensorflow/cc:function_ops",
+ "//tensorflow/cc:ops",
+ "//tensorflow/core:core_cpu",
+ "//tensorflow/core:core_cpu_internal",
+ "//tensorflow/core:framework",
+ "//tensorflow/core:ops",
+ "//tensorflow/core:test",
+ "//tensorflow/core:test_main",
+ "//tensorflow/core:testlib",
+ ],
+)
+
+cc_library(
+ name = "xla_local_runtime_context",
+ hdrs = ["xla_local_runtime_context.h"],
+ visibility = ["//visibility:public"],
+ deps = ["//tensorflow/core:framework_lite"],
+)
+
+cc_library(
+ name = "dump_graph",
+ srcs = [
+ "dump_graph.cc",
+ "dump_graph_flags.cc",
+ "dump_graph_flags.h",
+ ],
+ hdrs = [
+ "dump_graph.h",
+ ],
+ deps = [
+ "//tensorflow/compiler/xla/legacy_flags:parse_flags_from_env",
+ "//tensorflow/core:core_cpu",
+ "//tensorflow/core:core_cpu_internal",
+ "//tensorflow/core:framework",
+ "//tensorflow/core:framework_internal",
+ "//tensorflow/core:lib",
+ "//tensorflow/core:protos_all_cc",
+ ],
+)
+
+# -----------------------------------------------------------------------------
+
+filegroup(
+ name = "all_files",
+ srcs = glob(
+ ["**/*"],
+ exclude = [
+ "**/METADATA",
+ "**/OWNERS",
+ ],
+ ),
+ visibility = ["//tensorflow:__subpackages__"],
+)