\;j<ddlZGddZedZdZdZy)Nc*eZdZdZddgZdZdZdZy)RegistryzA general registry object.namerulesc ||_i|_yN)rr)selfrs fG:\00. PROJECTS\API\Inventory\templateJSON\kerjaOCR\Lib\site-packages\paddle/decomposition/register.py__init__zRegistry.__init__s  ct|tsJtj|sJ||jvs Jd|d||j|<y)Nzname "z"" should not be registered before.) isinstancestrinspect isfunctionr)r op_typerules r registerzRegistry.registers\'3'''!!$''' 4:: % @ G9> ? @ %" 7r c8|jj|Sr)rget)r rs r lookupzRegistry.lookup"szz~~g&&r N)__name__ __module__ __qualname____doc__ __slots__r rrr r rrs$!I#'r r decompositioncbttstdtdfd}|S)ap Decorator for registering the lower function for an original op into sequence of primitive ops. Args: op_type(str): The op name Returns: wrapper: Inner wrapper function Examples: .. code-block:: python >>> from paddle.decomposition import register >>> @register.register_decomp('softmax') >>> def softmax(x, axis): ... molecular = exp(x) ... denominator = broadcast_to(sum(molecular, axis=axis, keepdim=True), x.shape) ... res = divide(molecular, denominator) ... return res zop_type must be str, but got .c4tj||Sr)_decomposition_opsr)frs r wrapperz register_decomp..wrapperAs##GQ/r )rr TypeErrortype)rr$s` r register_decompr')s4* gs #7W aHII Nr c0tj|}|Sr)r"r)r _lowerrules r get_decomp_ruler*Hs#**73J r )rrr"r'r*rr r r+s+''*o.>r