o "j-@sddlmZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z m Z ddlZddlmZddlmZmZmZmZmZddlmZddlmZmZmZmZdd lmZm Z gZ!eZ"Gd d d Z#d d Z$e$a%dddZ&ej'dddZ(ddZ)ddZ*dS)) annotationsN)AnyCallable)Layer)convert_enumerate convert_len convert_print convert_range convert_zip)TranslatorLogger)CONVERSION_OPTIONSStaticFunctionconvert_to_staticunwrap_decorators) is_builtinis_paddle_funcc@s"eZdZdZdddZddZdS) ConversionOptionsa A container for conversion flags of a function in dynamic-to-static. Attributes: not_convert(bool): An attribute indicates that the function won't be converted in dynamic-to-static. NOTE(liym27): More attributes and methods can be added in this class. FcCs ||_dSN) not_convert)selfrrg/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/jit/dy2static/convert_call_func.py__init__@s zConversionOptions.__init__cCsBt|r|j}t|rt|t|dStdt|dS)NzLOnly support @not_to_static to type(function) or type(method), but received ) inspectismethod__func__ isfunctionsetattrr translator_loggerwarntype)rfuncrrrattachCs   zConversionOptions.attachN)F)__name__ __module__ __qualname____doc__rr#rrrrr6s rcCsBttttttttg}z ddl}| |W|St y Y|Sw)z! Return builtin modules. rN) copy collectionsrloggingnumpyospdbresixappend ImportError)modulesr/rrrbuiltin_modulesOs"   r3r2list[types.ModuleType]cCs |D] }|tvr t|qdS)z0 Adds modules that ignore transcription N)BUILTIN_LIKELY_MODULESr0)r2modulerrradd_ignore_modulejs  r7returnlist[Callable[..., Any]]cs(tdddfdd|S)Nr8list[tuple[str, Any]]cSs$zt|WStygYSwr)r getmembers Exceptionr6rrr_try_get_membersxs   z.get_module_functions.._try_get_memberscsZ|vrgS|g}|D]\}}t|r||t|r*||q|Sr)addcallabler0rismoduleextend)r6resultsZ _member_namemember_get_module_functionsr>visitedrrrF~s   z3get_module_functions.._get_module_functions)r8r:)setr=rrErget_module_functionsts  rIcCszddtD}|D]}||urtdd|ddSq ddlm}|g}t||vr.d St|r;tdd |dSd S) zJ Checks whether the func is supported by dygraph to static graph. cSsg|] }t|D]}|qqSr)rI).0r6r"rrr sz"is_unsupported..z Whitelist: z@ is part of built-in module and does not have to be transformed.Tr) SequentialFzKWhitelist: %s is part of Paddle module and does not have to be transformed.)r5rlog paddle.nnrMr!r)r"Zbuiltin_functionsZ builtin_fnrMZPADDLE_NEED_CONVERT_APISrrris_unsupporteds.   rPc Cs4tdd|d}d}t|\}}t|td}|dur'|jr'tdd||St|dr.tSt|dr5tSt|dr>> # doctest: +SKIP('`paddle.jit.to_static` can not run in xdoctest') >>> import paddle >>> from paddle.jit.dy2static import Call >>> paddle.enable_static() >>> def dyfunc(x): ... if paddle.mean(x) < 0: ... x_v = x - 1 ... else: ... x_v = x + 1 ... return x_v ... >>> new_func = Call(dyfunc) >>> x = paddle.tensor.manipulation.fill_constant(shape=[3, 3], value=0, dtype='float64') >>> x_v = new_func(x) >>> exe = paddle.static.Executor(paddle.CPUPlace()) >>> out = exe.run(fetch_list=[x_v]) >>> print(out[0]) [[1. 1. 1.] [1. 1. 1.] [1. 1. 1.]] rz$Convert callable object: convert %s.NrLzG%s is not converted when it is decorated by 'paddle.jit.not_to_static'.lenziprange enumerateprintz *z Your function:`zk` doesn't support to transform to static function because it is a generator function, it will be run as-is. z__self__zr%s doesn't have to be transformed to static function because it has been transformed before, it will be run as-is. __class__forwardz Callable z# can not be transformed at present.zO%s doesn't have to be transformed to static function, and it will be run as-is.).rrNrgetattrr rrrr r rr rPrisgeneratorfunctionr r$runwraprH __globals__valuesr? isinstancerisclass__dict__get staticmethodrAttributeErrorOSErrorrhasattrr@rZrr[rZ_original_funcs__get__ TypeError__call__NotImplementedError functoolspartial) r"Z func_selfZconverted_call_optionsZnumber_of_starsZ _origfuncZglobal_functionsfnZ forward_funcZ call_funcrrr convert_calls%                          rr)r2r4)r8r9)+ __future__rr)r(rmrr*r,r-r.typestypingrrr+rOrZconvert_operatorsrrr r r Z logging_utilsr Zprogram_translatorr rrrutilsrr__all__rrr3r5r7 lru_cacherIrPrrrrrrs6      '