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