Searched defs:wrapper_cls (Results 1 – 2 of 2) sorted by relevance
36 def register_loss_scale_wrapper(optimizer_cls, wrapper_fn, wrapper_cls=None): argument
576 wrapper_cls: Optional[Callable] = None # The wrapper class variable in _ConfigAutoWrap