o ñp iÔ&ã @sddlmZddlZddlZddlZddlmmZddl mZddl m Z m Z Gdd„dƒZ e ƒZejd2d d „ƒZejd3d d„ƒZejd3dd„ƒZd dddddddddddœ Zd4dd„Zdd„Zd d!„Zd"d#„Zd$d%„Zd&d'„Zd(d)„Zd*d+„Zd,d-„Zd.d/„Zd0d1„ZdS)5é)Ú annotationsN)Ú frameworké)Úop_infoÚop_mapc@s$eZdZdd„Zdd„Zdd„ZdS)Ú PrimOptioncCs d|_dS)NF©Ú enable_prim©Úself©r úe/home/app/PaddleOCR-VL/.venv_paddleocr/lib/python3.10/site-packages/paddle/incubate/autograd/utils.pyÚ__init__ó zPrimOption.__init__cCs|jS©Nrr r r r Ú get_statusszPrimOption.get_statuscCs ||_dSrr)r Úflagr r r Ú set_status"rzPrimOption.set_statusN)Ú__name__Ú __module__Ú __qualname__rrrr r r r rs rÚreturnÚboolcCst ¡S)a× Note: **ONLY available in the static graph mode.** Shows whether the automatic differentiation mechanism based on automatic differential basic operators is ON. Defaults to OFF. Returns: flag(bool): Whether the automatic differentiation mechanism based on automatic differential basic operators is ON. Examples: .. code-block:: python >>> import paddle >>> from paddle.incubate.autograd import enable_prim, disable_prim, prim_enabled >>> paddle.enable_static() >>> enable_prim() >>> print(prim_enabled()) True >>> disable_prim() >>> print(prim_enabled()) False )Ú prim_optionrr r r r Ú prim_enabled)srÚNonecCót d¡dS)aÇ Note: **ONLY available in the static graph mode.** Turns ON automatic differentiation mechanism based on automatic differential basic operators. Examples: .. code-block:: python >>> import paddle >>> from paddle.incubate.autograd import enable_prim, prim_enabled >>> paddle.enable_static() >>> enable_prim() >>> print(prim_enabled()) True TN©rrr r r r r Ksr cCr)a/ Note: **ONLY available in the static graph mode.** Turns OFF automatic differentiation mechanism based on automatic differential basic operators. Examples: .. code-block:: python >>> import paddle >>> from paddle.incubate.autograd import enable_prim, disable_prim, prim_enabled >>> paddle.enable_static() >>> enable_prim() >>> print(prim_enabled()) True >>> disable_prim() >>> print(prim_enabled()) False FNrr r r r Ú disable_primesrZint16Zint32Zint64Zfloat16Zfloat32Zfloat64Zuint8Zint8Z complex64Z complex128) rrééééééééécs^t|tƒsJ‚t|ƒdkrdSt|ƒdkr&|rˆ |d¡gSˆ |d¡S‡fdd„|DƒS)Nrrcsg|]}ˆ |¡‘qSr )Úvar)Ú.0Úname©Úblockr r Ú œsz!get_var_block..)Ú isinstanceÚlistÚlenr()r,ÚnamesÚis_tensor_listr r+r Ú get_var_block“s  r3có&ˆjdurgS‡fdd„tˆjƒDƒS)Ncóg|] }tˆjˆ |¡ƒ‘qSr )r3r,Úinput©r)Ún©Úopr r r-£sÿz&get_input_var_list..)Ú input_namesÚsortedr9r r9r Úget_input_var_listŸs  ÿr=cCs<d|vr|}n| d¡d}| ¡ ¡\}}| ¡| ¡fS)Nú=r)ÚsplitÚstrip)ÚitemÚresÚarg_typeÚarg_namer r r Ú _solve_arg¨s rEcCsbt|j}d| ¡vr||d ¡vr|d|}||jvr dS|dkr,t | |¡¡S| |¡S)NÚattrsZDataType)rÚtypeÚkeysZ attr_namesÚdtypesÚdtypeÚattr)r:rCrDÚ op_contentr r r Ú_get_attr_value±s    rMc CsHt|}|d d¡}t ¡}g}|D]Œ}t|ƒ\}}t|j} |dvrt|dkr*dnd} |  | ¡rt| |  |¡} | sItd|›d|›d |j›d ƒ‚d D](}|  |¡} | d urs| |j vrst |j |  | ¡ƒrstd |›d| ›d|j›d ƒ‚qK|dvr”d|   ¡vr|| d  ¡vr||| d|<q|||<qt|||ƒ} | | ¡q||fS)z&get attrs' values for api args' valuesÚargsú,)ÚIntArrayZScalarrPZ int_arrayZscalarzNo value found for z of z type for operator Ú.)Z tensor_nameZ tensors_nameNzTensor type of zL is not supported in composite op. Please set other type value of input arg z for operator )ZTensorúTensor[]Úinputs)rr?Ú collectionsÚ OrderedDictrErrGÚgetÚ ValueErrorr;r3r,r6rHrMÚappend)r:Úphi_namerNZ args_listrSrFrArCrDrLZ tensor_keyZtensor_contentZ arg_name_newÚ attr_valuer r r Ú_get_args_valuesÀsB   ÿ  ÿÿ€    r[cCs¸|jdurgS|jtvrt|jd}n|j}t||ƒ\}}g}| ¡D].\}}||jvrM|dkr@| t|j| |¡dd¡q$| t|j| |¡ƒ¡q$| d¡q$|rZ|  |¡|S)zö Generate all args inputs of composite op. Because inputs of composite op is the same as phi op described in ops.yaml. So we need to map origin op to phi op and then push input data and attrs of origin op to corresponding phi op. NrYrRT)r2) r;rGrr[ÚitemsrXr3r,r6Úextend)r:rYrSrFrBrAZ tensor_typer r r Úprepare_python_api_argumentsïs(   ÿÿ  r^cr4)Ncr5r )r3r,Úoutputr7r9r r r-sÿÿz'get_output_var_list..)Ú output_namesr<r9r r9r Úget_output_var_lists  þracCs¼|j}|dur gS|j}g}t| d¡rBt|d ¡D]"}t|d|}||vr1| d¡qt|j| |¡ƒ}| |¡q|St |ƒdkrZt|j| |d¡ƒ}| |¡|St dƒ‚)zlorigin op outputs must be mapped into outputs of composite rule. map info has been defined in op_compat.yamlNZoutputsrrzgWhen replace op with composite rule, there must exist output map info from origin op to composite rule.) r`rGrrVrHrXr3r,r_r0rW)r:Zorigin_output_namesr*rBrAZorigin_output_nameZorigin_output_varr r r Úmap_output_for_composites2  ÿ  öÿ ýÿrbcCsD|dust|tjjjtjjfƒr|gSg}|D]}|t|ƒ7}q|Sr)r.ÚpaddleÚbaserÚVariableÚpirÚValueÚflatten)ÚinpÚ flattenedÚpartr r r rh:s ÿrhcCst|ƒ}dd„|DƒS)NcSsg|]}|dur|‘qSrr )r)r(r r r r-Gsz+flatten_and_remove_none..)rh)rirjr r r Úflatten_and_remove_noneEsrlcCs2t|tjtjjfƒr |fSt|tjƒrt|ƒS|Sr) r.rrercrfrgÚtypingÚSequenceÚtuple)Úxsr r r Ú as_tensorsJs  rq)rr)rrr)Ú __future__rrTrmrcZpaddle.framework.dtyperrJrIZ paddle.baseZ phi_ops_maprrrrZ static_onlyrr rZINT_DTYPE_2_STRINGr3r=rErMr[r^rarbrhrlrqr r r r ÚsL    !  õ    /!