Initial commit for compression api core part.

PiperOrigin-RevId: 335770102

Author: Xhark

tensorflow_model_optimization/python/core/internal/compression/keras/BUILD

@@ -0,0 +1,72 @@
+package(default_visibility = [
+    "//tensorflow_model_optimization:__subpackages__",
+    "//third_party/tensorflow:__subpackages__",
+])
+
+licenses(["notice"])
+
+py_library(
+    name = "keras",
+    srcs = [
+        "__init__.py",
+    ],
+    srcs_version = "PY2AND3",
+    deps = [
+        ":compress",
+    ],
+)
+
+py_library(
+    name = "compress",
+    srcs = ["compress.py"],
+    srcs_version = "PY2AND3",
+    deps = [
+        ":compress_wrapper",
+        # attr dep1,
+        # enum dep1,
+        # tensorflow dep1,
+        # python:util tensorflow dep2,
+        # python/keras:models tensorflow dep2,
+        # python/keras/engine tensorflow dep2,
+    ],
+)
+
+py_test(
+    name = "compress_test",
+    srcs = [
+        "compress_test.py",
+    ],
+    python_version = "PY3",
+    deps = [
+        ":compress",
+        # absl/testing:parameterized dep1,
+        # numpy dep1,
+        # tensorflow dep1,
+    ],
+)
+
+py_library(
+    name = "compress_wrapper",
+    srcs = [
+        "compress_wrapper.py",
+    ],
+    srcs_version = "PY2AND3",
+    deps = [
+        # tensorflow dep1,
+    ],
+)
+
+py_test(
+    name = "compress_wrapper_test",
+    srcs = [
+        "compress_wrapper_test.py",
+    ],
+    python_version = "PY3",
+    visibility = ["//visibility:public"],
+    deps = [
+        ":compress_wrapper",
+        # absl/testing:parameterized dep1,
+        # numpy dep1,
+        # tensorflow dep1,
+    ],
+)

tensorflow_model_optimization/python/core/internal/compression/keras/__init__.py

@@ -0,0 +1,14 @@
+# 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.
+# ==============================================================================