From da2db8780e9d368c17c5160b3809085464ae08cb Mon Sep 17 00:00:00 2001 From: zhang__sss Date: Tue, 20 Jul 2021 17:12:35 +0800 Subject: [PATCH] add all for api --- mindspore/compression/common/__init__.py | 2 ++ mindspore/compression/quant/__init__.py | 3 +++ 2 files changed, 5 insertions(+) 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"]