Skip to content

Commit 892e137

Browse files
michaelreneercopybara-github
authored andcommitted
Update BUILD to expose run_length_gamma_kernels.
PiperOrigin-RevId: 487367842 Change-Id: Id18923b5e166ff4d3ceaf5a2ab48256556f9047f
1 parent dbc9a57 commit 892e137

File tree

1 file changed

+31
-0
lines changed
  • tensorflow_compression/cc

1 file changed

+31
-0
lines changed

tensorflow_compression/cc/BUILD

Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,3 +31,34 @@ cc_binary(
3131
"@tensorflow_pip//:libtensorflow_framework",
3232
],
3333
)
34+
35+
cc_library(
36+
name = "bit_coder",
37+
srcs = ["lib/bit_coder.cc"],
38+
hdrs = ["lib/bit_coder.h"],
39+
deps = [
40+
"@com_google_absl//absl/base:endian",
41+
"@com_google_absl//absl/status",
42+
"@com_google_absl//absl/status:statusor",
43+
"@com_google_absl//absl/strings",
44+
],
45+
)
46+
47+
cc_library(
48+
name = "run_length_gamma_kernels",
49+
srcs = [
50+
"kernels/run_length_gamma_kernels.cc",
51+
"ops/run_length_gamma_ops.cc",
52+
],
53+
linkstatic = 1,
54+
deps = [
55+
":bit_coder",
56+
"@com_google_absl//absl/types:span",
57+
"@org_tensorflow//tensorflow/core:framework",
58+
"@org_tensorflow//tensorflow/core:framework_lite",
59+
"@org_tensorflow//tensorflow/core:lib",
60+
"@org_tensorflow//tensorflow/core:tflite_portable_logging",
61+
"@org_tensorflow//tensorflow/core/platform:status",
62+
],
63+
alwayslink = 1,
64+
)

0 commit comments

Comments
 (0)