Removes TF1 entropy models.

PiperOrigin-RevId: 354820788
Change-Id: I34f966be3d2b93957c8667fbf6b672510aad8acc

Author: Johannes Ballé

BUILD

@@ -15,7 +15,6 @@ py_library(
         "//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/__init__.py

@@ -32,7 +32,6 @@
 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 *

tensorflow_compression/all_tests.py

@@ -29,7 +29,6 @@
 from tensorflow_compression.python.entropy_models.continuous_indexed_test import *
 from tensorflow_compression.python.entropy_models.universal_test import *
 
-from tensorflow_compression.python.layers.entropy_models_test import *
 from tensorflow_compression.python.layers.gdn_test import *
 from tensorflow_compression.python.layers.initializers_test import *
 from tensorflow_compression.python.layers.parameterizers_test import *

tensorflow_compression/python/layers/BUILD

@@ -4,16 +4,6 @@ package(
 
 licenses(["notice"])
 
-py_library(
-    name = "entropy_models",
-    srcs = ["entropy_models.py"],
-    srcs_version = "PY3",
-    deps = [
-        "//tensorflow_compression/python/ops:math_ops",
-        "//tensorflow_compression/python/ops:range_coding_ops",
-    ],
-)
-
 py_library(
     name = "gdn",
     srcs = ["gdn.py"],
@@ -61,13 +51,6 @@ py_library(
     deps = ["//tensorflow_compression/python/ops:soft_round_ops"],
 )
 
-py_test(
-    name = "entropy_models_test",
-    srcs = ["entropy_models_test.py"],
-    python_version = "PY3",
-    deps = [":entropy_models"],
-)
-
 py_test(
     name = "initializers_test",
     srcs = ["initializers_test.py"],