Update PQAT public API.

Change-Id: Iedb5507868b2d6508ec0898de37127e8ca996823

Author: psunn

tensorflow_model_optimization/python/core/api/BUILD

@@ -12,9 +12,10 @@ py_library(
         "quantization/keras/__init__.py",
         "quantization/keras/default_8bit/__init__.py",
         "quantization/keras/quantizers/__init__.py",
-        "quantization/keras/collaborative_optimization_scheme/__init__.py",
         "sparsity/__init__.py",
         "sparsity/keras/__init__.py",
+        "experimental/__init__.py",
+        "experimental/combine/__init__.py",
     ],
     srcs_version = "PY3",
     deps = [

tensorflow_model_optimization/python/core/api/__init__.py

@@ -1,4 +1,4 @@
-# Copyright 2019 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2020 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.
@@ -16,3 +16,4 @@
 from tensorflow_model_optimization.python.core.api import clustering
 from tensorflow_model_optimization.python.core.api import quantization
 from tensorflow_model_optimization.python.core.api import sparsity
+from tensorflow_model_optimization.python.core.api import experimental

tensorflow_model_optimization/python/core/api/experimental/__init__.py

@@ -0,0 +1,16 @@
+# Copyright 2020 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.
+# ==============================================================================
+"""Module containing code for experimental features."""
+from tensorflow_model_optimization.python.core.api.experimental import combine

tensorflow_model_optimization/python/core/api/quantization/keras/collaborative_optimization_scheme/__init__.py renamed to tensorflow_model_optimization/python/core/api/experimental/combine/__init__.py

@@ -12,7 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # ==============================================================================
-"""Module containing experimental Quantization schemes."""
-
+"""Module containing collaborative optimization code."""
 from tensorflow_model_optimization.python.core.quantization.keras.collaborative_optimizations.prune_preserve.default_8bit_prune_preserve_quantize_scheme import (
     Default8BitPrunePreserveQuantizeScheme, )

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

@@ -18,7 +18,6 @@
 # submodules
 from tensorflow_model_optimization.python.core.api.quantization.keras import quantizers
 from tensorflow_model_optimization.python.core.api.quantization.keras import default_8bit
-from tensorflow_model_optimization.python.core.api.quantization.keras import collaborative_optimization_scheme
 
 # quantize all layers with default quantization implementation.
 from tensorflow_model_optimization.python.core.quantization.keras.quantize import quantize_model