Merge pull request #839 from SaoirseARM:toupstream/tfmot_import

PiperOrigin-RevId: 419381204

Author: tensorflower-gardener

tensorflow_model_optimization/python/core/api/BUILD

@@ -49,6 +49,7 @@ py_strict_library(
         "//tensorflow_model_optimization/python/core/quantization/keras/default_8bit:default_8bit_quantize_registry",
         "//tensorflow_model_optimization/python/core/quantization/keras/default_8bit:default_8bit_quantize_scheme",
         "//tensorflow_model_optimization/python/core/quantization/keras/default_8bit:default_8bit_transforms",
+        "//tensorflow_model_optimization/python/core/quantization/keras/experimental",
         "//tensorflow_model_optimization/python/core/quantization/keras/experimental/default_n_bit:default_n_bit_quantize_layout_transform",
         "//tensorflow_model_optimization/python/core/quantization/keras/experimental/default_n_bit:default_n_bit_quantize_registry",
         "//tensorflow_model_optimization/python/core/quantization/keras/experimental/default_n_bit:default_n_bit_quantize_scheme",

tensorflow_model_optimization/python/core/quantization/keras/experimental/BUILD

@@ -0,0 +1,27 @@
+load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library")
+
+package(default_visibility = [
+    "//tensorflow_model_optimization:__subpackages__",
+])
+
+licenses(["notice"])
+
+py_strict_library(
+    name = "experimental",
+    srcs = [
+        "__init__.py",
+    ],
+    srcs_version = "PY3",
+    deps = [
+        ":quantization",  # buildcleaner: keep
+    ],
+)
+
+py_strict_library(
+    name = "quantization",
+    srcs = ["__init__.py"],
+    srcs_version = "PY3",
+    deps = [
+        "//tensorflow_model_optimization/python/core/quantization/keras",  # buildcleaner: keep
+    ],
+)

tensorflow_model_optimization/python/core/quantization/keras/experimental/__init__.py

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