1load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") 2 3package( 4 default_visibility = [ 5 "//tensorflow/compiler/tf2xla:internal", 6 "//tensorflow/core/tpu:__subpackages__", 7 ], 8 licenses = ["notice"], 9) 10 11XLA_OPS_DEPS = [ 12 "@com_google_absl//absl/container:flat_hash_map", 13 "@com_google_absl//absl/memory", 14 "@com_google_absl//absl/synchronization", 15 "//tensorflow/compiler/jit:common", 16 "//tensorflow/compiler/jit:compilation_passes", 17 "//tensorflow/compiler/jit:flags", 18 "//tensorflow/compiler/jit:xla_activity_listener", 19 "//tensorflow/compiler/jit:xla_activity_proto_cc", 20 "//tensorflow/compiler/jit:xla_compilation_cache", 21 "//tensorflow/compiler/jit:xla_device_no_jit_rewrite_registration", 22 "//tensorflow/compiler/jit:xla_cluster_util", 23 "//tensorflow/compiler/jit:xla_launch_util", 24 "//tensorflow/compiler/tf2xla:common", 25 "//tensorflow/compiler/tf2xla:tf2xla_util", 26 "//tensorflow/compiler/tf2xla:xla_compiler", 27 "//tensorflow/compiler/tf2xla:xla_helpers", 28 "//tensorflow/compiler/tf2xla:xla_op_registry", 29 "//tensorflow/compiler/xla:executable_run_options", 30 "//tensorflow/compiler/xla:status_macros", 31 "//tensorflow/compiler/xla:statusor", 32 "//tensorflow/compiler/xla/client:client_library", 33 "//tensorflow/compiler/xla/client:local_client", 34 "//tensorflow/compiler/xla/service:compiler", 35 "//tensorflow/compiler/xla/service/gpu:gpu_executable_run_options", 36 "//tensorflow/core:core_cpu_internal", 37 "//tensorflow/core:framework", 38 "//tensorflow/core:lib", 39 "//tensorflow/core:state_ops_op_lib", 40 "//tensorflow/core/platform:stream_executor_no_cuda", 41 "//tensorflow/core/profiler/lib:traceme", 42 "//tensorflow/stream_executor:tf_allocator_adapter", 43 "@com_google_absl//absl/types:optional", 44] 45 46# Linked by tensorflow core, without registration of jit compilation passes. 47cc_library( 48 name = "xla_ops_no_jit_rewrite_registration", 49 srcs = ["xla_ops.cc"], 50 hdrs = ["xla_ops.h"], 51 deps = XLA_OPS_DEPS + ["//tensorflow/core/platform:refcount"], 52 alwayslink = 1, 53) 54 55cc_library( 56 name = "xla_ops", 57 hdrs = ["xla_ops.h"], 58 deps = XLA_OPS_DEPS + [ 59 ":xla_ops_no_jit_rewrite_registration", 60 "//tensorflow/compiler/jit:jit_compilation_passes", 61 ], 62 alwayslink = 1, 63) 64