6b715d7238
Now that we execute compression as a tensorflow op, a traceme will automatically be generated for the op, so we don't need a second traceme inside the compression util. PiperOrigin-RevId: 312776419 Change-Id: I709c89b6a7fafaf41dbc8a64c8c28025cd7cd287
80 lines
1.9 KiB
Python
80 lines
1.9 KiB
Python
load("//tensorflow:tensorflow.bzl", "tf_cc_test")
|
|
load(
|
|
"//tensorflow/core/platform:build_config.bzl",
|
|
"tf_additional_all_protos",
|
|
"tf_proto_library",
|
|
"tf_protos_all",
|
|
)
|
|
|
|
package(
|
|
default_visibility = [
|
|
"//tensorflow:internal",
|
|
],
|
|
licenses = ["notice"], # Apache 2.0
|
|
)
|
|
|
|
exports_files(["LICENSE"])
|
|
|
|
cc_library(
|
|
name = "compression_utils",
|
|
srcs = ["compression_utils.cc"],
|
|
hdrs = [
|
|
"compression_utils.h",
|
|
],
|
|
deps = [
|
|
":dataset_proto_cc",
|
|
"//tensorflow/core:core_cpu",
|
|
"//tensorflow/core:core_cpu_internal",
|
|
"//tensorflow/core:framework_internal",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:lib_internal",
|
|
"//tensorflow/core:protos_all_cc",
|
|
"@com_google_absl//absl/memory",
|
|
],
|
|
)
|
|
|
|
tf_cc_test(
|
|
name = "compression_utils_test",
|
|
srcs = ["compression_utils_test.cc"],
|
|
deps = [
|
|
":compression_utils",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:lib_internal",
|
|
"//tensorflow/core:test",
|
|
"//tensorflow/core:test_main",
|
|
"//tensorflow/core:testlib",
|
|
"//tensorflow/core/kernels/data:dataset_test_base",
|
|
],
|
|
)
|
|
|
|
tf_proto_library(
|
|
name = "dataset_proto",
|
|
srcs = ["dataset.proto"],
|
|
cc_api_version = 2,
|
|
protodeps = tf_additional_all_protos(),
|
|
)
|
|
|
|
cc_library(
|
|
name = "standalone",
|
|
srcs = ["standalone.cc"],
|
|
hdrs = ["standalone.h"],
|
|
deps = [
|
|
"//tensorflow/core:all_kernels",
|
|
"//tensorflow/core:core_cpu_internal",
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:session_options",
|
|
"@com_google_absl//absl/memory",
|
|
],
|
|
)
|
|
|
|
tf_cc_test(
|
|
name = "standalone_test",
|
|
srcs = ["standalone_test.cc"],
|
|
deps = [
|
|
":standalone",
|
|
"//tensorflow/core:test",
|
|
"//tensorflow/core:test_main",
|
|
] + tf_protos_all(),
|
|
)
|