• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1package(
2    default_visibility = ["//tensorflow/compiler/tf2xla:internal"],
3)
4
5licenses(["notice"])  # Apache 2.0
6
7load(
8    "//tensorflow:tensorflow.bzl",
9    "tf_custom_op_library",
10    "tf_gen_op_wrapper_py",
11)
12
13cc_library(
14    name = "xla_ops",
15    srcs = ["xla_ops.cc"],
16    deps = [
17        "//tensorflow/core:framework",
18        "//tensorflow/core:lib",
19        "@com_google_absl//absl/algorithm:container",
20        "@com_google_absl//absl/strings",
21    ],
22    alwayslink = 1,
23)
24
25tf_gen_op_wrapper_py(
26    name = "gen_xla_ops",
27    out = "gen_xla_ops.py",
28    deps = [
29        ":xla_ops",
30    ],
31)
32
33tf_custom_op_library(
34    name = "_xla_ops.so",
35    srcs = [
36        "xla_ops.cc",
37    ],
38    deps = [
39        "//tensorflow/core:lib",
40        "@com_google_absl//absl/algorithm:container",
41    ],
42)
43