o "j@s2ddlZGdddZedZddZddZdS) Nc@s0eZdZdZddgZddZddZdd Zd S) RegistryzA general registry object.namerulescCs||_i|_dSN)rr)selfrr^/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/decomposition/register.py__init__s zRegistry.__init__cCsDt|tsJt|sJ||jvsJd|d||j|<dS)Nzname "z"" should not be registered before.) isinstancestrinspect isfunctionr)rop_typerulerrrregisters   zRegistry.registercCs |j|Sr)rget)rrrrrlookup"s zRegistry.lookupN)__name__ __module__ __qualname____doc__ __slots__r rrrrrrrs  r decompositioncs.ttstdtdfdd}|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 .cst||Sr)_decomposition_opsr)frrrwrapperAs z register_decomp..wrapper)r r TypeErrortype)rrrrrregister_decomp)s  r cCst|}|Sr)rr)rZ _lowerrulerrrget_decomp_ruleHs r!)r rrr r!rrrrs