File size: 1,173 Bytes
1f5470c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 |
try:
import namex
except ImportError:
namex = None
# These dicts reference "canonical names" only
# (i.e. the first name an object was registered with).
REGISTERED_NAMES_TO_OBJS = {}
REGISTERED_OBJS_TO_NAMES = {}
def register_internal_serializable(path, symbol):
global REGISTERED_NAMES_TO_OBJS
if isinstance(path, (list, tuple)):
name = path[0]
else:
name = path
REGISTERED_NAMES_TO_OBJS[name] = symbol
REGISTERED_OBJS_TO_NAMES[symbol] = name
def get_symbol_from_name(name):
return REGISTERED_NAMES_TO_OBJS.get(name, None)
def get_name_from_symbol(symbol):
return REGISTERED_OBJS_TO_NAMES.get(symbol, None)
if namex:
class keras_export(namex.export):
def __init__(self, path):
super().__init__(package="keras", path=path)
def __call__(self, symbol):
register_internal_serializable(self.path, symbol)
return super().__call__(symbol)
else:
class keras_export:
def __init__(self, path):
self.path = path
def __call__(self, symbol):
register_internal_serializable(self.path, symbol)
return symbol
|