49 строки
1.3 KiB
Python
49 строки
1.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
# File: __init__.py
|
|
|
|
# https://github.com/celery/kombu/blob/7d13f9b95d0b50c94393b962e6def928511bfda6/kombu/__init__.py#L34-L36
|
|
STATICA_HACK = True
|
|
globals()['kcah_acitats'[::-1].upper()] = False
|
|
if STATICA_HACK:
|
|
from .batch_norm import *
|
|
from .common import *
|
|
from .conv2d import *
|
|
from .fc import *
|
|
from .layer_norm import *
|
|
from .linearwrap import *
|
|
from .nonlin import *
|
|
from .pool import *
|
|
from .regularize import *
|
|
|
|
|
|
from pkgutil import iter_modules
|
|
import os
|
|
import os.path
|
|
# this line is necessary for _TFModuleFunc to work
|
|
import tensorflow as tf # noqa: F401
|
|
|
|
__all__ = []
|
|
|
|
|
|
def _global_import(name):
|
|
p = __import__(name, globals(), locals(), level=1)
|
|
lst = p.__all__ if '__all__' in dir(p) else dir(p)
|
|
del globals()[name]
|
|
for k in lst:
|
|
if not k.startswith('__'):
|
|
globals()[k] = p.__dict__[k]
|
|
__all__.append(k)
|
|
|
|
|
|
_CURR_DIR = os.path.dirname(__file__)
|
|
_SKIP = ['utils', 'registry', 'tflayer']
|
|
for _, module_name, _ in iter_modules(
|
|
[_CURR_DIR]):
|
|
srcpath = os.path.join(_CURR_DIR, module_name + '.py')
|
|
if not os.path.isfile(srcpath):
|
|
continue
|
|
if module_name.startswith('_'):
|
|
continue
|
|
if module_name not in _SKIP:
|
|
_global_import(module_name)
|