[Collaborative optimization] Pruning-Clustering-preserving Quantization Aware Training (PCQAT)

Author: Ruomei

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

@@ -35,6 +35,7 @@ py_strict_library(
     ],
     srcs_version = "PY3",
     deps = [
+        ":cluster_utils",
         # tensorflow dep1,
         "//tensorflow_model_optimization/python/core/clustering/keras:clustering_registry",
         "//tensorflow_model_optimization/python/core/quantization/keras:quant_ops",
@@ -61,6 +62,23 @@ py_test(
     ],
 )
 
+py_test(
+    name = "mnist_prune_cluster_preserve_qat_test",
+    srcs = [
+        "mnist_prune_cluster_preserve_qat_test.py",
+    ],
+    python_version = "PY3",
+    visibility = ["//visibility:private"],
+    deps = [
+        # tensorflow dep1,
+        ":default_8bit_cluster_preserve_quantize_scheme",
+        "//tensorflow_model_optimization/python/core/quantization/keras:quantize",
+        "//tensorflow_model_optimization/python/core/sparsity/keras:prune",
+        "//tensorflow_model_optimization/python/core/sparsity/keras:pruning_callbacks",
+        "//tensorflow_model_optimization/python/core/clustering/keras/experimental:cluster",
+    ]
+)
+
 py_strict_library(
     name = "default_8bit_cluster_preserve_quantize_scheme",
     srcs = [
@@ -72,3 +90,21 @@ py_strict_library(
         "//tensorflow_model_optimization/python/core/quantization/keras/default_8bit:default_8bit_quantize_scheme",
     ],
 )
+
+py_test(
+    name = "cluster_preserve_integration_test",
+    srcs = [
+        "cluster_preserve_integration_test.py",
+    ],
+    python_version = "PY3",
+    visibility = ["//visibility:private"],
+    deps = [
+        ":default_8bit_cluster_preserve_quantize_scheme",
+        # tensorflow dep1,
+        "//tensorflow_model_optimization/python/core/clustering/keras:clustering_registry",
+        "//tensorflow_model_optimization/python/core/clustering/keras:cluster",
+        "//tensorflow_model_optimization/python/core/clustering/keras/experimental:cluster",
+        "//tensorflow_model_optimization/python/core/quantization/keras:quantize_config",
+        "//tensorflow_model_optimization/python/core/quantization/keras:quantize"
+    ],
+)