__init__.py 1.7 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14
# Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

15
from .quantization_pass import (
16 17
    QuantizationTransformPass,
)
18
from .quantization_pass import (
19 20
    QuantizationFreezePass,
)
21
from .quantization_pass import (
22 23
    ConvertToInt8Pass,
)
24
from .quantization_pass import (
25 26
    TransformForMobilePass,
)
27
from .quantization_pass import (
28 29
    OutScaleForTrainingPass,
)
30
from .quantization_pass import (
31 32
    OutScaleForInferencePass,
)
33
from .quantization_pass import (
34 35
    AddQuantDequantPass,
)
36
from .quantization_pass import (
37 38
    ReplaceFakeQuantDequantPass,
)
39 40 41 42
from .quantization_pass import (
    QuantWeightPass,
)
from .quantization_pass import (
43 44
    QuantizationTransformPassV2,
)
45
from .quantization_pass import (
46 47
    AddQuantDequantPassV2,
)
48 49 50 51
from .quantization_pass import (
    AddQuantDequantForInferencePass,
)
from .quant_int8_mkldnn_pass import (
52 53
    QuantInt8MkldnnPass,
)
54
from .quant2_int8_mkldnn_pass import (
55 56
    Quant2Int8MkldnnPass,
)
57

58
from .post_training_quantization import (
59 60
    PostTrainingQuantization,
)
61
from .post_training_quantization import (
62 63
    PostTrainingQuantizationProgram,
)
64
from .post_training_quantization import (
65 66
    WeightQuantization,
)
67 68 69 70
from .quanter import (
    quant_aware,
    convert,
)