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

View File

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

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