52 lines
1.3 KiB
Python
52 lines
1.3 KiB
Python
load("//tensorflow:tensorflow.bzl", "tf_gen_op_wrapper_cc")
|
|
|
|
package(
|
|
default_visibility = ["//tensorflow/compiler/tf2xla:internal"],
|
|
licenses = ["notice"], # Apache 2.0
|
|
)
|
|
|
|
tf_gen_op_wrapper_cc(
|
|
name = "xla_ops_gen",
|
|
out_ops_file = "ops/xla_ops",
|
|
deps = ["//tensorflow/compiler/tf2xla/ops:xla_ops"],
|
|
)
|
|
|
|
cc_library(
|
|
name = "xla_ops",
|
|
srcs = ["ops/xla_ops.cc"],
|
|
hdrs = ["ops/xla_ops.h"],
|
|
deps = [
|
|
"//tensorflow/cc:const_op",
|
|
"//tensorflow/cc:ops",
|
|
"//tensorflow/cc:scope",
|
|
"//tensorflow/compiler/tf2xla/ops:xla_ops",
|
|
"//tensorflow/core:core_cpu",
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:protos_all_cc",
|
|
],
|
|
)
|
|
|
|
tf_gen_op_wrapper_cc(
|
|
name = "xla_jit_op_gen",
|
|
include_internal_ops = 1,
|
|
out_ops_file = "ops/xla_jit_ops",
|
|
deps = ["//tensorflow/compiler/jit/ops:xla_ops"],
|
|
)
|
|
|
|
cc_library(
|
|
name = "xla_jit_ops",
|
|
srcs = ["ops/xla_jit_ops.cc"],
|
|
hdrs = ["ops/xla_jit_ops.h"],
|
|
deps = [
|
|
"//tensorflow/cc:const_op",
|
|
"//tensorflow/cc:ops",
|
|
"//tensorflow/cc:scope",
|
|
"//tensorflow/compiler/jit/ops:xla_ops",
|
|
"//tensorflow/core:core_cpu",
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:protos_all_cc",
|
|
],
|
|
)
|