50 lines
1.1 KiB
Python
50 lines
1.1 KiB
Python
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):
|
|
REGISTERED_NAMES_TO_OBJS.get(name, None)
|
|
|
|
|
|
def get_name_from_symbol(symbol):
|
|
REGISTERED_OBJS_TO_NAMES.get(symbol, None)
|
|
|
|
|
|
if namex:
|
|
|
|
class keras_core_export(namex.export):
|
|
def __init__(self, path):
|
|
super().__init__(package="keras_core", path=path)
|
|
|
|
def __call__(self, symbol):
|
|
register_internal_serializable(self.path, symbol)
|
|
return super().__call__(symbol)
|
|
|
|
else:
|
|
|
|
class keras_core_export:
|
|
def __init__(self, path):
|
|
self.path = path
|
|
|
|
def __call__(self, symbol):
|
|
register_internal_serializable(self.path, symbol)
|
|
return symbol
|