xref: /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/python/BUILD (revision b6fb3261f9314811a0f4371741dbb8839866f948)
1load(
2    "//tensorflow/core/platform:build_config.bzl",
3    "tf_py_clif_cc",
4)
5load("//tensorflow:tensorflow.bzl", "tf_custom_op_py_library")
6
7package(
8    default_visibility = [
9        "//visibility:public",
10    ],
11    licenses = ["notice"],
12)
13
14tf_py_clif_cc(
15    name = "xla_op_registry",
16    srcs = ["xla_op_registry.clif"],
17    pyclif_deps = [
18        "//tensorflow/core/framework:kernel_def_pyclif",
19    ],
20    deps = [
21        "//tensorflow/compiler/tf2xla:xla_compiler",
22    ],
23)
24
25tf_custom_op_py_library(
26    name = "xla",
27    srcs = ["xla.py"],
28    dso = ["//tensorflow/compiler/tf2xla/ops:_xla_ops.so"],
29    kernels = [
30        "//tensorflow/compiler/tf2xla/ops:xla_ops",
31    ],
32    deps = [
33        "//tensorflow/compiler/tf2xla/ops:gen_xla_ops",
34        "//tensorflow/compiler/xla:xla_data_proto_py",
35    ],
36)
37