__init__.py 1.5 KB
Newer Older
Q
Qingsheng Li 已提交
1
#   Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserve.
2
#   Copyright (c) 2021 NVIDIA Corporation.  All rights reserved.
Q
Qingsheng Li 已提交
3 4 5 6 7 8 9 10 11 12 13 14 15
#
# 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.

16 17
from __future__ import print_function

M
minqiyang 已提交
18 19
from . import decoder
from .decoder import *
20 21
from . import memory_usage_calc
from .memory_usage_calc import *
C
chengduo 已提交
22 23
from . import op_frequence
from .op_frequence import *
24
from . import quantize
D
Dang Qingqing 已提交
25
from .quantize import *
W
whs 已提交
26
from . import slim
C
chengduo 已提交
27 28
from . import extend_optimizer
from .extend_optimizer import *
29 30
from . import model_stat
from .model_stat import *
31 32
from . import mixed_precision
from .mixed_precision import *
33 34
from . import layers
from .layers import *
35
from . import optimizer
36 37
from . import sparsity
from .sparsity import *
Q
Qingsheng Li 已提交
38

C
chengduo 已提交
39 40 41 42
__all__ = []
__all__ += decoder.__all__
__all__ += memory_usage_calc.__all__
__all__ += op_frequence.__all__
D
test,  
Dang Qingqing 已提交
43
__all__ += quantize.__all__
C
chengduo 已提交
44
__all__ += extend_optimizer.__all__
45
__all__ += ['mixed_precision']
46
__all__ += layers.__all__
47
__all__ += optimizer.__all__
48
__all__ += sparsity.__all__