diff --git a/mindspore/compression/common/__init__.py b/mindspore/compression/common/__init__.py
index 148bdbbf86..5ed90b1eef 100644
--- a/mindspore/compression/common/__init__.py
+++ b/mindspore/compression/common/__init__.py
@@ -17,3 +17,5 @@ Compression common module.
 """
 
 from .constant import QuantDtype
+
+__all__ = ["QuantDtype"]
diff --git a/mindspore/compression/quant/__init__.py b/mindspore/compression/quant/__init__.py
index 092a1c48a6..713970270b 100644
--- a/mindspore/compression/quant/__init__.py
+++ b/mindspore/compression/quant/__init__.py
@@ -19,3 +19,6 @@ Compression quant module.
 from .quantizer import OptimizeOption
 from .qat import QuantizationAwareTraining, create_quant_config
 from .quant_utils import load_nonquant_param_into_quant_net, query_quant_layers
+
+__all__ = ["load_nonquant_param_into_quant_net", "query_quant_layers", "QuantizationAwareTraining",
+           "create_quant_config", "OptimizeOption"]