Simplifications to build system.

PiperOrigin-RevId: 348102676
Change-Id: Icd43dceb63297c2eeff07cbae2d7b973b33dae3a

Author: Johannes Ballé

BUILD

@@ -4,26 +4,37 @@ exports_files(["LICENSE"])
 
 py_library(
     name = "tensorflow_compression",
-    srcs = [
-        "tensorflow_compression/__init__.py",
-        "tensorflow_compression/python/__init__.py",
-    ],
+    srcs = ["tensorflow_compression/__init__.py"],
     srcs_version = "PY3",
     visibility = ["//visibility:public"],
     deps = [
-        "//tensorflow_compression/python/distributions",
-        "//tensorflow_compression/python/entropy_models",
-        "//tensorflow_compression/python/layers",
-        "//tensorflow_compression/python/ops",
-        "//tensorflow_compression/python/util",
+        "//tensorflow_compression/python/distributions:deep_factorized",
+        "//tensorflow_compression/python/distributions:helpers",
+        "//tensorflow_compression/python/distributions:round_adapters",
+        "//tensorflow_compression/python/distributions:uniform_noise",
+        "//tensorflow_compression/python/entropy_models:continuous_batched",
+        "//tensorflow_compression/python/entropy_models:continuous_indexed",
+        "//tensorflow_compression/python/entropy_models:universal",
+        "//tensorflow_compression/python/layers:entropy_models",
+        "//tensorflow_compression/python/layers:gdn",
+        "//tensorflow_compression/python/layers:initializers",
+        "//tensorflow_compression/python/layers:parameterizers",
+        "//tensorflow_compression/python/layers:signal_conv",
+        "//tensorflow_compression/python/layers:soft_round",
+        "//tensorflow_compression/python/ops:math_ops",
+        "//tensorflow_compression/python/ops:padding_ops",
+        "//tensorflow_compression/python/ops:range_coding_ops",
+        "//tensorflow_compression/python/ops:soft_round_ops",
+        "//tensorflow_compression/python/ops:spectral_ops",
+        "//tensorflow_compression/python/util:packed_tensors",
     ],
 )
 
 filegroup(
     name = "pip_src",
     srcs = [
         "MANIFEST.in",
-        "tensorflow_compression/python/all_test.py",
+        "tensorflow_compression/all_tests.py",
     ],
 )
 
@@ -44,38 +55,3 @@ py_binary(
     ],
     python_version = "PY3",
 )
-
-py_binary(
-    name = "generate_docs",
-    srcs = ["tools/generate_docs.py"],
-    python_version = "PY3",
-    deps = [":tensorflow_compression"],
-)
-
-py_binary(
-    name = "tfci",
-    srcs = ["models/tfci.py"],
-    python_version = "PY3",
-    deps = [":tensorflow_compression"],
-)
-
-py_binary(
-    name = "bls2017",
-    srcs = ["models/bls2017.py"],
-    python_version = "PY3",
-    deps = [":tensorflow_compression"],
-)
-
-py_binary(
-    name = "bmshj2018",
-    srcs = ["models/bmshj2018.py"],
-    python_version = "PY3",
-    deps = [":tensorflow_compression"],
-)
-
-py_binary(
-    name = "ms2020",
-    srcs = ["models/ms2020.py"],
-    python_version = "PY3",
-    deps = [":tensorflow_compression"],
-)

README.md

@@ -232,7 +232,7 @@ interpreter would attempt to import `tensorflow_compression` packages from the
 source tree, rather than from the installed package system directory):
 ```bash
 pushd /tmp
-python -m tensorflow_compression.python.all_test
+python -m tensorflow_compression.all_tests
 popd
 ```
 

tensorflow_compression/__init__.py

@@ -27,20 +27,23 @@
 from tensorflow_compression.python.distributions.helpers import *
 from tensorflow_compression.python.distributions.round_adapters import *
 from tensorflow_compression.python.distributions.uniform_noise import *
+
 from tensorflow_compression.python.entropy_models.continuous_batched import *
 from tensorflow_compression.python.entropy_models.continuous_indexed import *
 from tensorflow_compression.python.entropy_models.universal import *
+
 from tensorflow_compression.python.layers.entropy_models import *
 from tensorflow_compression.python.layers.gdn import *
 from tensorflow_compression.python.layers.initializers import *
 from tensorflow_compression.python.layers.parameterizers import *
 from tensorflow_compression.python.layers.signal_conv import *
 from tensorflow_compression.python.layers.soft_round import *
+
 from tensorflow_compression.python.ops.math_ops import *
 from tensorflow_compression.python.ops.padding_ops import *
 from tensorflow_compression.python.ops.range_coding_ops import *
 from tensorflow_compression.python.ops.soft_round_ops import *
 from tensorflow_compression.python.ops.spectral_ops import *
-from tensorflow_compression.python.util.packed_tensors import *
 
+from tensorflow_compression.python.util.packed_tensors import *
 # pylint: enable=wildcard-import

tensorflow_compression/python/all_test.py renamed to tensorflow_compression/all_tests.py

@@ -12,10 +12,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # ==============================================================================
-"""Tests for tensorflow_compression/python.
+"""All Python tests for tensorflow_compression.
 
-This is a convenience file to be included in PIP package.
-No BUILD entry exists for this file on purpose.
+This is a convenience file to be included in the pip package.
 """
 
 import tensorflow.compat.v1 as tf

tensorflow_compression/python/__init__.py

@@ -4,18 +4,6 @@ package(
 
 licenses(["notice"])  # Apache 2.0
 
-py_library(
-    name = "distributions",
-    srcs = ["__init__.py"],
-    srcs_version = "PY3",
-    deps = [
-        ":deep_factorized",
-        ":helpers",
-        ":round_adapters",
-        ":uniform_noise",
-    ],
-)
-
 py_library(
     name = "deep_factorized",
     srcs = ["deep_factorized.py"],

tensorflow_compression/python/distributions/BUILD

@@ -4,18 +4,6 @@ package(
 
 licenses(["notice"])  # Apache 2.0
 
-py_library(
-    name = "entropy_models",
-    srcs = ["__init__.py"],
-    srcs_version = "PY3",
-    deps = [
-        ":continuous_base",
-        ":continuous_batched",
-        ":continuous_indexed",
-        ":universal",
-    ],
-)
-
 py_library(
     name = "continuous_base",
     srcs = ["continuous_base.py"],

tensorflow_compression/python/distributions/__init__.py

@@ -4,20 +4,6 @@ package(
 
 licenses(["notice"])  # Apache 2.0
 
-py_library(
-    name = "layers",
-    srcs = ["__init__.py"],
-    srcs_version = "PY3",
-    deps = [
-        ":entropy_models",
-        ":gdn",
-        ":initializers",
-        ":parameterizers",
-        ":signal_conv",
-        ":soft_round",
-    ],
-)
-
 py_library(
     name = "entropy_models",
     srcs = ["entropy_models.py"],
@@ -46,7 +32,6 @@ py_library(
     name = "initializers",
     srcs = ["initializers.py"],
     srcs_version = "PY3",
-    visibility = ["//visibility:private"],
 )
 
 py_library(