wehub-resource-sync
40 lines
1002 B
Python
40 lines
1002 B
Python
load("//tensorflow:tensorflow.default.bzl", "tf_custom_op_py_library")
|
|
load(
|
|
"//tensorflow/core/platform:build_config.bzl",
|
|
"tf_py_clif_cc",
|
|
)
|
|
|
|
package(
|
|
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
|
|
default_visibility = [
|
|
"//visibility:public",
|
|
],
|
|
licenses = ["notice"],
|
|
)
|
|
|
|
tf_py_clif_cc(
|
|
name = "xla_op_registry",
|
|
srcs = ["xla_op_registry.clif"],
|
|
pyclif_deps = [
|
|
"//tensorflow/core/framework:kernel_def_pyclif",
|
|
],
|
|
deps = [
|
|
"//tensorflow/compiler/tf2xla:xla_compiler",
|
|
],
|
|
)
|
|
|
|
tf_custom_op_py_library(
|
|
name = "xla",
|
|
srcs = ["xla.py"],
|
|
dso = ["//tensorflow/compiler/tf2xla/ops:_xla_ops.so"],
|
|
kernels = [
|
|
"//tensorflow/compiler/tf2xla/ops:xla_ops",
|
|
],
|
|
deps = [
|
|
"//tensorflow/compiler/tf2xla/ops:gen_xla_ops",
|
|
"//tensorflow/python/ops:random_ops_util",
|
|
"//tensorflow/python/ops/numpy_ops:np_utils",
|
|
"@xla//xla:xla_data_proto_py",
|
|
],
|
|
)
|