Move prune_preserve to collaborative_optimizations.

Update PQAT public API.

Change-Id: If99eabae1aef85aa7505e790522b8b04ee3c7b9d

Author: psunn

tensorflow_model_optimization/python/core/api/BUILD

@@ -12,7 +12,7 @@ py_library(
         "quantization/keras/__init__.py",
         "quantization/keras/default_8bit/__init__.py",
         "quantization/keras/quantizers/__init__.py",
-        "quantization/keras/experimental_scheme/__init__.py",
+        "quantization/keras/collaborative_optimization_scheme/__init__.py",
         "sparsity/__init__.py",
         "sparsity/keras/__init__.py",
     ],

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

@@ -18,7 +18,7 @@
 # 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 experimental_scheme
+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

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

@@ -14,5 +14,5 @@
 # ==============================================================================
 """Module containing experimental Quantization schemes."""
 
-from tensorflow_model_optimization.python.core.quantization.keras.prune_preserve.default_8bit_prune_preserve_quantize_scheme import (
+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/quantization/keras/BUILD

@@ -17,7 +17,7 @@ py_library(
         "//tensorflow_model_optimization/python/core/quantization/keras/graph_transformations",
         "//tensorflow_model_optimization/python/core/quantization/keras/layers",
         "//tensorflow_model_optimization/python/core/quantization/keras/default_8bit",
-        "//tensorflow_model_optimization/python/core/quantization/keras/prune_preserve",
+        "//tensorflow_model_optimization/python/core/quantization/keras/collaborative_optimizations",
     ],
 )
 

tensorflow_model_optimization/python/core/quantization/keras/collaborative_optimizations/BUILD

@@ -0,0 +1,10 @@
+package(default_visibility = ["//tensorflow_model_optimization:__subpackages__"])
+
+licenses(["notice"])  # Apache 2.0
+
+py_library(
+    name = "collaborative_optimizations",
+    srcs = ["__init__.py"],
+    srcs_version = "PY3",
+    deps = ["//tensorflow_model_optimization/python/core/quantization/keras/collaborative_optimizations/prune_preserve"],
+)

tensorflow_model_optimization/python/core/quantization/keras/prune_preserve/__init__.py renamed to tensorflow_model_optimization/python/core/quantization/keras/collaborative_optimizations/__init__.py

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

tensorflow_model_optimization/python/core/quantization/keras/prune_preserve/BUILD renamed to tensorflow_model_optimization/python/core/quantization/keras/collaborative_optimizations/prune_preserve/BUILD

@@ -16,7 +16,7 @@
 
 from tensorflow_model_optimization.python.core.quantization.keras.default_8bit import (
     default_8bit_quantize_scheme, )
-from tensorflow_model_optimization.python.core.quantization.keras.prune_preserve import (
+from tensorflow_model_optimization.python.core.quantization.keras.collaborative_optimizations.prune_preserve import (
     prune_preserve_quantize_registry, )