Esempio n. 1
0
def register_results(cls, name=None):
    return register(cls, RESULTS, name, 'results')
Esempio n. 2
0
def register_arc_policy(cls, name=None):
    """Register a function as a plug-in"""
    return register(cls, BASELINE_SEQ2SEQ_ARC_POLICY, name, 'decoder')
Esempio n. 3
0
def register_reporting(cls, name=None):
    """Register a function as a plug-in"""
    return register(cls, BASELINE_REPORTING, name, 'reporting')
Esempio n. 4
0
def register_vectorizer(cls, name=None):
    """Register a function as a plug-in"""
    return register(cls, BASELINE_VECTORIZERS, name, 'vectorizer')
Esempio n. 5
0
def register_decoder(cls, name=None):
    """Register a function as a plug-in"""
    return register(cls, BASELINE_SEQ2SEQ_DECODERS, name, 'decoder')
Esempio n. 6
0
def register_reporting(cls, name=None):
    """Register a function as a plug-in"""
    return register(cls, BASELINE_REPORTING, name, 'reporting')
Esempio n. 7
0
def register_vectorizer(cls, name=None):
    """Register a function as a plug-in"""
    return register(cls, BASELINE_VECTORIZERS, name, 'vectorizer')
Esempio n. 8
0
def register_preprocessor(cls, name=None):
    """Register a function as a plug-in"""
    return register(cls, BASELINE_PREPROCESSORS, name, 'preprocessor')
Esempio n. 9
0
def register_sampler(cls, name=None):
    if cls.name is None:
        cls.name = name if name is not None else cls.__name__
    return register(cls, SAMPLERS, cls.name, "sampler")
Esempio n. 10
0
def register_arc_policy(cls, name=None):
    """Register a function as a plug-in"""
    return register(cls, BASELINE_SEQ2SEQ_ARC_POLICY, name, 'decoder')
Esempio n. 11
0
def register_progress(cls, name=None):
    return register(cls, BASELINE_PROGRESS, name, 'progress')
Esempio n. 12
0
def register_decoder(cls, name=None):
    """Register a function as a plug-in"""
    return register(cls, BASELINE_SEQ2SEQ_DECODERS, name, 'decoder')
Esempio n. 13
0
def register_backend(cls, name=None):
    return register(cls, BACKENDS, name, "backend")
Esempio n. 14
0
def register_backend(cls, name=None):
    return register(cls, BACKENDS, name, "backend")
Esempio n. 15
0
def register_lr_scheduler(cls, name=None):
    return register(cls, BASELINE_LR_SCHEDULERS, name, 'lr_scheduler')
Esempio n. 16
0
def register_frontend(cls, name=None):
    return register(cls, FRONTENDS, name, "frontend")
Esempio n. 17
0
def register_lr_scheduler(cls, name=None):
    return register(cls, BASELINE_LR_SCHEDULERS, name, 'lr_scheduler')
Esempio n. 18
0
def register_preprocessor(cls, name=None):
    """Register a function as a plug-in"""
    return register(cls, BASELINE_PREPROCESSORS, name, 'preprocessor')