ssk015
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add all for api

This commit is contained in:
zhang__sss 2021-07-20 17:12:35 +08:00
parent 66fe07a009
commit da2db8780e
2 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/compression/common/__init__.py
View File

@ -17,3 +17,5 @@ Compression common module.
"""
from .constant import QuantDtype
__all__ = ["QuantDtype"]

3
mindspore/compression/quant/__init__.py
View File

@ -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"]