Skip to content

Commit 47cff11

Browse files
Johannes Ball?copybara-github
authored andcommitted
Creates PackedTensors class to share code between TFCI and BLS2017 model.
Also updates command line argument parsing and model creation in BLS2017 model. PiperOrigin-RevId: 259811485 Change-Id: I0e51a9b08d24697b12a28b66073e4068e74f024a
1 parent b758903 commit 47cff11

File tree

9 files changed

+544
-155
lines changed

9 files changed

+544
-155
lines changed

BUILD

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@ py_library(
1212
deps = [
1313
"//tensorflow_compression/python/layers",
1414
"//tensorflow_compression/python/ops",
15+
"//tensorflow_compression/python/util",
1516
],
1617
)
1718

@@ -34,6 +35,7 @@ py_binary(
3435
# The following targets are for Python test files.
3536
"//tensorflow_compression/python/layers:py_src",
3637
"//tensorflow_compression/python/ops:py_src",
38+
"//tensorflow_compression/python/util:py_src",
3739
],
3840
)
3941

@@ -43,6 +45,12 @@ py_binary(
4345
deps = [":tensorflow_compression"],
4446
)
4547

48+
py_binary(
49+
name = "bls2017",
50+
srcs = ["examples/bls2017.py"],
51+
deps = [":tensorflow_compression"],
52+
)
53+
4654
py_binary(
4755
name = "generate_docs",
4856
srcs = ["tools/generate_docs.py"],

0 commit comments

Comments
 (0)