o "j_& @sddlZddlZddlZddlmmZddlmZddlm Z m Z GdddZ e Z ej ddZej d d Zej d d Zd ddddddddddd Zd/ddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.ZdS)0N) framework)op_infoop_mapc@s$eZdZddZddZddZdS) PrimOptioncCs d|_dS)NF enable_primselfr _/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/incubate/autograd/utils.py__init__ zPrimOption.__init__cCs|jSNrr r r r get_statusszPrimOption.get_statuscCs ||_dSrr)r flagr r r set_statusrzPrimOption.set_statusN)__name__ __module__ __qualname__r rrr r r r rs rcCstS)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&srcCtddS)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 TNrrr r r r rHsrcCr)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_primbsrboolZint16Zint32Zint64Zfloat16Zfloat32Zfloat64Zuint8Zint8Z complex64Z complex128) rrcs^t|tsJt|dkrdSt|dkr&|r|dgS|dSfdd|DS)Nrrcsg|]}|qSr )var).0nameblockr r sz!get_var_block..) isinstancelistlenr%)r)namesis_tensor_listr r(r get_var_blocks  r0c&jdurgSfddtjDS)Ncg|] }tj|qSr )r0r)inputr&nopr r r*sz&get_input_var_list..) input_namessortedr6r r6r get_input_var_lists  r:cCs<d|vr|}n|dd}|\}}||fS)N=r)splitstrip)itemresarg_typearg_namer r r _solve_args rBcCsbt|j}d|vr||dvr|d|}||jvr dS|dkr,t||S||S)NattrsZDataType)rtypekeysZ attr_namesdtypesdtypeattr)r7r@rA op_contentr r r _get_attr_values    rJc CsHt|}|dd}t}g}|D]}t|\}}t|j} |dvrt|dkr*dnd} | | rt| | |} | sItd|d|d |jd d D](}| |} | d urs| |j vrst |j | | rstd |d| d|jd qK|dvrd| vr|| d vr||| d|<q|||<qt|||} || q||fS)z&get attrs' values for api args' valuesargs,)IntArrayZScalarrMZ 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 )ZTensorTensor[]inputs)rr< collections OrderedDictrBrrDget ValueErrorr8r0r)r3rErJappend)r7phi_namerKZ args_listrPrCr>r@rArIZ tensor_keyZtensor_contentZ arg_name_newZ attr_valuer r r _get_args_valuessB         rWcCs|jdurgS|jtvrt|jd}n|j}t||\}}g}|D].\}}||jvrM|dkr@|t|j||ddq$|t|j||q$|dq$|rZ| ||S)z Generate all args inputs of composite op. Because inputs of composite op is the same as phi op desribed in ops.yaml. So we need to map origin op to phi op and then push input data and attrs of origin op to correspondng phi op. NrVrOT)r/) r8rDrrWitemsrUr0r)r3extend)r7rVrPrCr?r>Z tensor_typer r r prepare_python_api_argumentss(     rZcr1)Ncr2r )r0r)outputr4r6r r r*sz'get_output_var_list..) output_namesr9r6r r6r get_output_var_list s  r]cCs|j}|dur gS|j}g}t|drBt|dD]"}t|d|}||vr1|dqt|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\rDrrSrErUr0r)r[r-rT)r7Zorigin_output_namesr'r?r>Zorigin_output_nameZorigin_output_varr r r map_output_for_composites2     r^cCs<|dus t|tjjjr|gSg}|D]}|t|7}q|Sr)r+paddlebaserVariableflatten)inp flattenedpartr r r rb7s rbcCst|}dd|DS)NcSsg|]}|dur|qSrr )r&r%r r r r*Bsz+flatten_and_remove_none..)rb)rcrdr r r flatten_and_remove_none@srfcCs*t|tjr |fSt|tjrt|S|Sr)r+rratypingSequencetuple)Zxsr r r as_tensorsEs  rjr)rQrgr_Zpaddle.framework.dtyperrGrFZ paddle.baseZ phi_ops_maprrrrZ static_onlyrrrZINT_DTYPE_2_STRINGr0r:rBrJrWrZr]r^rbrfrjr r r r sJ    !      /!