wehub-resource-sync
44 lines
1.2 KiB
Python
44 lines
1.2 KiB
Python
load("@xla//third_party/rules_python/python:py_library.bzl", "py_library")
|
|
load("//tensorflow:tensorflow.bzl", "tf_gen_op_wrapper_py")
|
|
load("//tensorflow/core/platform:rules_cc.bzl", "cc_library")
|
|
|
|
package(
|
|
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
|
|
default_visibility = ["//tensorflow/compiler/tf2xla:internal"],
|
|
licenses = ["notice"],
|
|
)
|
|
|
|
cc_library(
|
|
name = "xla_ops",
|
|
srcs = ["xla_ops.cc"],
|
|
deps = [
|
|
"//tensorflow/core:framework",
|
|
"@com_google_absl//absl/status",
|
|
],
|
|
alwayslink = 1,
|
|
)
|
|
|
|
tf_gen_op_wrapper_py(
|
|
name = "xla_ops_wrapper_py",
|
|
out = "xla_ops.py",
|
|
extra_py_deps = [
|
|
"//tensorflow/python:pywrap_tfe",
|
|
"//tensorflow/python/util:dispatch",
|
|
"//tensorflow/python/util:deprecation",
|
|
"//tensorflow/python/util:tf_export",
|
|
],
|
|
py_lib_rule = py_library,
|
|
deps = ["//tensorflow/compiler/jit/ops:xla_ops"],
|
|
)
|
|
|
|
py_library(
|
|
name = "xla_ops_grad",
|
|
srcs = ["xla_ops_grad.py"],
|
|
strict_deps = True,
|
|
visibility = [
|
|
"//tensorflow/compiler/tf2xla:internal",
|
|
"//tensorflow/python/ops:__pkg__",
|
|
],
|
|
deps = ["//tensorflow/python/framework:ops"],
|
|
)
|