o #j@spddlZddlZddlmZddlmZeeejddZGdddeZ Gdd d Z Gd d d Z d d Z dS)N)Enum) get_loggerz&%(asctime)s-%(levelname)s: %(message)s)fmtc@seZdZdZdZdS) FunctionTyperN)__name__ __module__ __qualname__ FP16_ONLYCOMMONr r d/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/static/amp/function_overload.pyrsrc@s eZdZdZddZddZdS)Functiona; Function is a wrap over standard python function An instance of this Function class is also callable just like the python function that it wrapped. When the instance is "called" like a function it fetches the function to be invoked from the virtual namespace and then invokes the same. cCs ||_dSNfn)selfrr r r __init__,s zFunction.__init__cOs"tj|i|}||i|S)z] Overriding the __call__ function which makes the instance callable. ) Namespace get_instanceget)rargskwargsrr r r __call__/szFunction.__call__N)rrr __doc__rrr r r r r"s rc@s8eZdZdZdZddZeddZddZd d Z dS) rza Namespace is the singleton class that is responsible for holding all the functions. NcCs"|jdur i|_|t_dStd)Nz#cannot instantiate Namespace again.)_Namespace__instance function_mapr Exception)rr r r rCs  zNamespace.__init__cCstjdurttjSr)rrr r r r rJs zNamespace.get_instancecCs4t|tsJdt|dt|}||j|<|S)a Register the function in the virtual namespace and return an instance of callable Function that wraps the function fn. Args: fn (function): the native python function handle. key (FunctionType): the specified type. z>The type of key is expected to be FunctionType, but recieved .) isinstancertyperr)rrkeyfuncr r r registerPs  zNamespace.registerc Os:td|d|t|j}t|t|}|jD]V}||vr&q|j|}t|}t|t|t|krNtd|d|d||qt|dkru| D]\}} ||j vrttd|d|d||nqXqt|dkrt |d} nt|dkrt |d t rtj} ntj} |j| S) z Get the matching function from the virtual namespace according to the actual arguments. Return None if it did not find any matching function. zget function: args=z , kwargs=zfn=z (key=z) is not satisfied and removed.rr)_loggerdebugsetrkeysleninspectgetfullargspecremoveitemsrlistrfloatrr r r) rrrZsatisfied_function_keysZnum_actual_argsZfunc_keyrspecsZarg_namevaluer!r r r r`s<        z Namespace.get) rrr rrr staticmethodrr#rr r r r r;s  rcsfdd}|S)zjoverload is the decorator that wraps the function and returns a callable object of type Function. cst|Sr)rrr#rr!r r decoratorszoverload..decoratorr )r!r5r r4r overloads r6) r+loggingenumrZpaddle.base.log_helperrrINFOr&rrrr6r r r r s   K