Release tensor_encoding part of the project.

PiperOrigin-RevId: 247634243

Author: PraChetit

requirements.txt

@@ -1,3 +1,4 @@
 absl-py~=0.7
 numpy~=1.14
 six~=1.10
+scipy

tensorflow_model_optimization/python/core/internal/README.md

@@ -0,0 +1,10 @@
+# Internal
+
+The internal package contains a collection of modules that are documented,
+fully working, and there exists a plan to be graduated to the main `tf.mot`
+namespace -- either a literal swap of `tf.mot.internal.<module>` for
+`tf.mot.<module>`, or creation of separate public API that calls the internal
+API under the hood.
+
+The `tf.mot.internal` modules can change in backwards incompatible manner. Thus,
+depending on these modules is not recommended.

tensorflow_model_optimization/python/core/internal/tensor_encoding/BUILD

@@ -0,0 +1,14 @@
+package(default_visibility = ["//visibility:public"])
+
+licenses(["notice"])  # Apache 2.0
+
+py_library(
+    name = "tensor_encoding",
+    srcs = ["__init__.py"],
+    deps = [
+        "//tensorflow_model_optimization/python/core/internal/tensor_encoding/core",
+        "//tensorflow_model_optimization/python/core/internal/tensor_encoding/encoders",
+        "//tensorflow_model_optimization/python/core/internal/tensor_encoding/stages",
+        "//tensorflow_model_optimization/python/core/internal/tensor_encoding/utils",
+    ],
+)

tensorflow_model_optimization/python/core/internal/tensor_encoding/__init__.py

@@ -0,0 +1,23 @@
+# Copyright 2019, The TensorFlow Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""The Tensor Encoding package."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import print_function
+
+from tensorflow_model_optimization.python.core.internal.tensor_encoding import core
+from tensorflow_model_optimization.python.core.internal.tensor_encoding import encoders
+from tensorflow_model_optimization.python.core.internal.tensor_encoding import stages
+from tensorflow_model_optimization.python.core.internal.tensor_encoding import utils

tensorflow_model_optimization/python/core/internal/tensor_encoding/core/BUILD

@@ -0,0 +1,88 @@
+package(default_visibility = [
+    "//tensorflow_model_optimization/python/core/internal/tensor_encoding:__subpackages__",
+])
+
+licenses(["notice"])  # Apache 2.0
+
+py_library(
+    name = "core",
+    srcs = ["__init__.py"],
+    visibility = ["//visibility:public"],
+    deps = [
+        ":core_encoder",
+        ":encoding_stage",
+        ":simple_encoder",
+    ],
+)
+
+py_library(
+    name = "core_encoder",
+    srcs = ["core_encoder.py"],
+    deps = [
+        ":encoding_stage",
+        # six dep1,
+        # tensorflow dep1,
+        "//tensorflow_model_optimization/python/core/internal/tensor_encoding/utils:py_utils",
+    ],
+)
+
+py_test(
+    name = "core_encoder_test",
+    size = "small",
+    srcs = ["core_encoder_test.py"],
+    deps = [
+        ":core_encoder",
+        ":encoding_stage",
+        # numpy dep1,
+        # tensorflow dep1,
+        "//tensorflow_model_optimization/python/core/internal/tensor_encoding/testing:test_utils",
+    ],
+)
+
+py_library(
+    name = "encoding_stage",
+    srcs = ["encoding_stage.py"],
+    deps = [
+        # enum dep1,
+        # six dep1,
+        # tensorflow dep1,
+    ],
+)
+
+py_test(
+    name = "encoding_stage_test",
+    size = "small",
+    srcs = ["encoding_stage_test.py"],
+    deps = [
+        ":encoding_stage",
+        # absl/testing:parameterized dep1,
+        # mock dep1,
+        # numpy dep1,
+        # tensorflow dep1,
+        "//tensorflow_model_optimization/python/core/internal/tensor_encoding/testing:test_utils",
+    ],
+)
+
+py_library(
+    name = "simple_encoder",
+    srcs = ["simple_encoder.py"],
+    deps = [
+        ":core_encoder",
+        # tensorflow dep1,
+        "//tensorflow_model_optimization/python/core/internal/tensor_encoding/utils:py_utils",
+    ],
+)
+
+py_test(
+    name = "simple_encoder_test",
+    size = "small",
+    srcs = ["simple_encoder_test.py"],
+    deps = [
+        ":core_encoder",
+        ":simple_encoder",
+        # absl/testing:parameterized dep1,
+        # numpy dep1,
+        # tensorflow dep1,
+        "//tensorflow_model_optimization/python/core/internal/tensor_encoding/testing:test_utils",
+    ],
+)

tensorflow_model_optimization/python/core/internal/tensor_encoding/core/__init__.py

@@ -0,0 +1,30 @@
+# Copyright 2019, The TensorFlow Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""Core parts of the `tensor_encoding` package."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import print_function
+
+from tensorflow_model_optimization.python.core.internal.tensor_encoding.core.core_encoder import Encoder
+from tensorflow_model_optimization.python.core.internal.tensor_encoding.core.core_encoder import EncoderComposer
+
+from tensorflow_model_optimization.python.core.internal.tensor_encoding.core.encoding_stage import AdaptiveEncodingStageInterface
+from tensorflow_model_optimization.python.core.internal.tensor_encoding.core.encoding_stage import EncodingStageInterface
+from tensorflow_model_optimization.python.core.internal.tensor_encoding.core.encoding_stage import StateAggregationMode
+from tensorflow_model_optimization.python.core.internal.tensor_encoding.core.encoding_stage import tf_style_adaptive_encoding_stage
+from tensorflow_model_optimization.python.core.internal.tensor_encoding.core.encoding_stage import tf_style_encoding_stage
+
+from tensorflow_model_optimization.python.core.internal.tensor_encoding.core.simple_encoder import SimpleEncoder
+from tensorflow_model_optimization.python.core.internal.tensor_encoding.core.simple_encoder import StatefulSimpleEncoder