Removes TF1 entropy models.

PiperOrigin-RevId: 354820788 Change-Id: I34f966be3d2b93957c8667fbf6b672510aad8acc
tensorflow · Jan 31, 2021 · 89eefc8 · 89eefc8
1 parent 1a9b176
commit 89eefc8
Show file tree

Hide file tree

Showing 6 changed files with 0 additions and 1,643 deletions.
diff --git a/BUILD b/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",

diff --git a/tensorflow_compression/__init__.py b/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 *

diff --git a/tensorflow_compression/all_tests.py b/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 *

diff --git a/tensorflow_compression/python/layers/BUILD b/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"],